mirror of https://github.com/oxen-io/session-ios
Fixes #530
parent
212f0d4350
commit
456d1c479a
@ -0,0 +1,44 @@
|
||||
//
|
||||
// MessagePaddingTests.m
|
||||
// Signal
|
||||
//
|
||||
// Created by Frederic Jacobs on 26/02/15.
|
||||
// Copyright (c) 2015 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
#import "Cryptography.h"
|
||||
#import "NSData+messagePadding.h"
|
||||
#import <XCTest/XCTest.h>
|
||||
|
||||
@interface MessagePaddingTests : XCTestCase
|
||||
|
||||
@end
|
||||
|
||||
@implementation MessagePaddingTests
|
||||
|
||||
- (void)testV3Padding {
|
||||
for (NSUInteger i=0;i<159;i++) {
|
||||
NSData *data = [NSMutableData dataWithLength:i];
|
||||
XCTAssert([data paddedMessageBody].length == 159);
|
||||
}
|
||||
|
||||
for (NSUInteger i=159;i<319;i++) {
|
||||
NSData *data = [NSMutableData dataWithLength:i];
|
||||
XCTAssert([data paddedMessageBody].length == 319);
|
||||
}
|
||||
|
||||
for (NSUInteger i=319;i<479;i++) {
|
||||
NSData *data = [NSMutableData dataWithLength:i];
|
||||
XCTAssert([data paddedMessageBody].length == 479);
|
||||
}
|
||||
}
|
||||
|
||||
- (void)testV3RandomPadding{
|
||||
for (int i = 0; i < 1000; i++) {
|
||||
NSData *randomMessage = [Cryptography generateRandomBytes:501];
|
||||
NSData *paddedMessage = [randomMessage paddedMessageBody];
|
||||
XCTAssert([[paddedMessage removePadding] isEqualToData:randomMessage]);
|
||||
}
|
||||
}
|
||||
|
||||
@end
|
Loading…
Reference in New Issue