mirror of https://github.com/oxen-io/session-ios
CR: test malformed protos
Michael Kirk
8 years ago
parent
e5856b2ac5
commit
d39906f606
@ -0,0 +1,68 @@
|
||||
//
|
||||
// Copyright (c) 2018 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
import XCTest
|
||||
|
||||
import SignalServiceKit
|
||||
import SwiftProtobuf
|
||||
|
||||
class SSKEnvelopeTest: XCTestCase {
|
||||
|
||||
override func setUp() {
|
||||
super.setUp()
|
||||
// Put setup code here. This method is called before the invocation of each test method in the class.
|
||||
}
|
||||
|
||||
override func tearDown() {
|
||||
// Put teardown code here. This method is called after the invocation of each test method in the class.
|
||||
super.tearDown()
|
||||
}
|
||||
|
||||
func testParse_EmptyData() {
|
||||
let data = Data()
|
||||
XCTAssertThrowsError(try SSKEnvelope(serializedData: data))
|
||||
}
|
||||
|
||||
func testParse_UnparseableData() {
|
||||
let data = "asdf".data(using: .utf8)!
|
||||
XCTAssertThrowsError(try SSKEnvelope(serializedData: data)) { error in
|
||||
XCTAssert(error is SwiftProtobuf.BinaryDecodingError)
|
||||
}
|
||||
}
|
||||
|
||||
func testParse_ValidData() {
|
||||
// `encodedData` was derived thus:
|
||||
// let builder = OWSSignalServiceProtosEnvelopeBuilder()
|
||||
// builder.setTimestamp(NSDate.ows_millisecondTimeStamp())
|
||||
// builder.setSource("+15551231234")
|
||||
// builder.setSourceDevice(1)
|
||||
// builder.setType(OWSSignalServiceProtosEnvelopeType.ciphertext)
|
||||
// let encodedData = builder.build().data()!.base64EncodedString()
|
||||
let encodedData = "CAESDCsxNTU1MTIzMTIzNCjKm4WazSw4AQ=="
|
||||
let data = Data(base64Encoded: encodedData)!
|
||||
|
||||
XCTAssertNoThrow(try SSKEnvelope(serializedData: data))
|
||||
}
|
||||
|
||||
func testParse_invalidData() {
|
||||
// `encodedData` was derived thus:
|
||||
// let builder = OWSSignalServiceProtosEnvelopeBuilder()
|
||||
// builder.setTimestamp(NSDate.ows_millisecondTimeStamp())
|
||||
// builder.setSource("+15551231234")
|
||||
// builder.setSourceDevice(1)
|
||||
// // MISSING TYPE!
|
||||
// let encodedData = builder.build().data()!.base64EncodedString()
|
||||
let encodedData = "EgwrMTU1NTEyMzEyMzQojdmOms0sOAE="
|
||||
let data = Data(base64Encoded: encodedData)!
|
||||
|
||||
XCTAssertThrowsError(try SSKEnvelope(serializedData: data)) { (error) -> Void in
|
||||
switch error {
|
||||
case SSKEnvelope.EnvelopeError.invalidProtobuf:
|
||||
break
|
||||
default:
|
||||
XCTFail("unexpected error: \(error)")
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue