mirror of https://github.com/oxen-io/session-ios
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.1 KiB
Objective-C
45 lines
1.1 KiB
Objective-C
//
|
|
// 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
|