init

Tests/JetStreamTests/Unit/MessageTests.swift

@@ -0,0 +1,119 @@
+// Copyright 2024 The NATS Authors
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+import XCTest
+
+@testable import JetStream
+@testable import Nats
+
+class JetStreamMessageTests: XCTestCase {
+
+    static var allTests = [
+        ("testValidOldFormatMessage", testValidOldFormatMessage),
+        ("testValidNewFormatMessage", testValidNewFormatMessage),
+        ("testMissingTokens", testMissingTokens),
+        ("testInvalidTokenValues", testInvalidTokenValues),
+        ("testInvalidPrefix", testInvalidPrefix),
+        ("testNoReplySubject", testNoReplySubject),
+    ]
+
+    func testValidOldFormatMessage() async throws {
+        let replySubject = "$JS.ACK.myStream.myConsumer.10.20.30.1234567890.5"
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: replySubject, length: 0, headers: nil,
+            status: nil, description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+
+        let metadata = try jetStreamMessage.metadata()
+
+        XCTAssertNil(metadata.domain)
+        XCTAssertNil(metadata.accountHash)
+        XCTAssertEqual(metadata.stream, "myStream")
+        XCTAssertEqual(metadata.consumer, "myConsumer")
+        XCTAssertEqual(metadata.delivered, 10)
+        XCTAssertEqual(metadata.streamSequence, 20)
+        XCTAssertEqual(metadata.consumerSequence, 30)
+        XCTAssertEqual(metadata.timestamp, "1234567890")
+        XCTAssertEqual(metadata.pending, 5)
+    }
+
+    func testValidNewFormatMessage() async throws {
+        let replySubject = "$JS.ACK.domain.accountHash123.myStream.myConsumer.10.20.30.1234567890.5"
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: replySubject, length: 0, headers: nil,
+            status: nil, description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+        let metadata = try jetStreamMessage.metadata()
+
+        XCTAssertEqual(metadata.domain, "domain")
+        XCTAssertEqual(metadata.accountHash, "accountHash123")
+        XCTAssertEqual(metadata.stream, "myStream")
+        XCTAssertEqual(metadata.consumer, "myConsumer")
+        XCTAssertEqual(metadata.delivered, 10)
+        XCTAssertEqual(metadata.streamSequence, 20)
+        XCTAssertEqual(metadata.consumerSequence, 30)
+        XCTAssertEqual(metadata.timestamp, "1234567890")
+        XCTAssertEqual(metadata.pending, 5)
+    }
+
+    func testMissingTokens() async throws {
+        let replySubject = "$JS.ACK.myStream.myConsumer"
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: replySubject, length: 0, headers: nil,
+            status: nil, description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+        do {
+            _ = try jetStreamMessage.metadata()
+        } catch JetStreamError.MessageMetadataError.invalidTokenNum {
+            return
+        }
+    }
+
+    func testInvalidTokenValues() async throws {
+        let replySubject = "$JS.ACK.myStream.myConsumer.invalid.20.30.1234567890.5"
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: replySubject, length: 0, headers: nil,
+            status: nil, description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+        do {
+            _ = try jetStreamMessage.metadata()
+        } catch JetStreamError.MessageMetadataError.invalidTokenValue {
+            return
+        }
+    }
+
+    func testInvalidPrefix() async throws {
+        let replySubject = "$JS.WRONG.myStream.myConsumer.10.20.30.1234567890.5"
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: replySubject, length: 0, headers: nil,
+            status: nil, description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+        do {
+            _ = try jetStreamMessage.metadata()
+        } catch JetStreamError.MessageMetadataError.invalidPrefix {
+            return
+        }
+    }
+
+    func testNoReplySubject() async throws {
+        let natsMessage = NatsMessage(
+            payload: nil, subject: "", replySubject: nil, length: 0, headers: nil, status: nil,
+            description: nil)
+        let jetStreamMessage = JetStreamMessage(message: natsMessage, client: NatsClient())
+        do {
+            _ = try jetStreamMessage.metadata()
+        } catch JetStreamError.MessageMetadataError.noReplyInMessage {
+            return
+        }
+    }
+}