120 lines
4.9 KiB
Swift
120 lines
4.9 KiB
Swift
// 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
|
|
}
|
|
}
|
|
}
|