170 lines
6.7 KiB
Objective-C
170 lines
6.7 KiB
Objective-C
//
|
|
// Copyright 2018 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
#import "OWSMessageManager.h"
|
|
#import "ContactsManagerProtocol.h"
|
|
#import "HTTPUtils.h"
|
|
#import "MessageSender.h"
|
|
#import "MockSSKEnvironment.h"
|
|
#import "OWSIdentityManager.h"
|
|
#import "OWSSyncGroupsMessage.h"
|
|
#import "SSKBaseTestObjC.h"
|
|
#import "TSAccountManager.h"
|
|
#import "TSGroupThread.h"
|
|
#import <SignalCoreKit/Cryptography.h>
|
|
#import <SignalCoreKit/OWSAsserts.h>
|
|
#import <SignalCoreKit/OWSLogs.h>
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
|
|
NS_ASSUME_NONNULL_BEGIN
|
|
|
|
NSString *const kLocalE164 = @"+13215550198";
|
|
NSString *const kLocalUuidString = @"B0D19730-950B-462C-84E7-60421F879EEF";
|
|
|
|
#pragma mark -
|
|
|
|
@interface OWSMessageManagerTest : SSKBaseTestObjC
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation OWSMessageManagerTest
|
|
|
|
- (void)setUp
|
|
{
|
|
[super setUp];
|
|
[self.tsAccountManager registerForTestsWithLocalNumber:kLocalE164
|
|
uuid:[[NSUUID alloc] initWithUUIDString:kLocalUuidString]];
|
|
[self.sskJobQueues.messageSenderJobQueue setup];
|
|
}
|
|
|
|
#pragma mark -
|
|
|
|
- (void)test_IncomingSyncGroupsMessage
|
|
{
|
|
XCTestExpectation *messageWasSent = [self expectationWithDescription:@"message was sent"];
|
|
|
|
OWSAssertDebug([((NSObject *) SSKEnvironment.shared.syncManager) isKindOfClass:[OWSMockSyncManager class]]);
|
|
OWSMockSyncManager *mockSyncManager = (OWSMockSyncManager *)SSKEnvironment.shared.syncManager;
|
|
mockSyncManager.syncGroupsHook = ^{
|
|
[messageWasSent fulfill];
|
|
};
|
|
|
|
SSKProtoSyncMessageRequestBuilder *requestBuilder =
|
|
[SSKProtoSyncMessageRequest builder];
|
|
[requestBuilder setType:SSKProtoSyncMessageRequestTypeGroups];
|
|
|
|
SSKProtoSyncMessageBuilder *messageBuilder = [SSKProtoSyncMessage builder];
|
|
[messageBuilder setRequest:[requestBuilder buildIgnoringErrors]];
|
|
|
|
SSKProtoEnvelopeBuilder *envelopeBuilder =
|
|
[SSKProtoEnvelope builderWithTimestamp:12345];
|
|
[envelopeBuilder setType:SSKProtoEnvelopeTypeCiphertext];
|
|
[envelopeBuilder setSourceUuid:kLocalUuidString];
|
|
[envelopeBuilder setSourceDevice:1];
|
|
|
|
[self writeWithBlock:^(SDSAnyWriteTransaction *transaction) {
|
|
[self.messageManager handleIncomingEnvelope:[envelopeBuilder buildIgnoringErrors]
|
|
withSyncMessage:[messageBuilder buildIgnoringErrors]
|
|
plaintextData:[NSData data]
|
|
wasReceivedByUD:NO
|
|
serverDeliveryTimestamp:0
|
|
transaction:transaction];
|
|
}];
|
|
|
|
[self waitForExpectationsWithTimeout:5
|
|
handler:^(NSError *error) {
|
|
OWSLogInfo(@"No message submitted.");
|
|
}];
|
|
}
|
|
|
|
- (void)test_GroupUpdate
|
|
{
|
|
// GroupsV2 TODO: Handle v2 groups.
|
|
NSData *groupId = [TSGroupModel generateRandomV1GroupId];
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
TSGroupThread *_Nullable thread = [TSGroupThread fetchWithGroupId:groupId transaction:transaction];
|
|
XCTAssertNil(thread);
|
|
}];
|
|
|
|
SSKProtoEnvelopeBuilder *envelopeBuilder = [SSKProtoEnvelope builderWithTimestamp:12345];
|
|
[envelopeBuilder setSourceUuid:NSUUID.UUID.UUIDString];
|
|
[envelopeBuilder setType:SSKProtoEnvelopeTypeCiphertext];
|
|
|
|
SSKProtoGroupContextBuilder *groupContextBuilder =
|
|
[SSKProtoGroupContext builderWithId:groupId];
|
|
[groupContextBuilder setType:SSKProtoGroupContextTypeUpdate];
|
|
[groupContextBuilder setName:@"Newly created Group Name"];
|
|
|
|
SSKProtoDataMessageBuilder *messageBuilder = [SSKProtoDataMessage builder];
|
|
messageBuilder.group = [groupContextBuilder buildIgnoringErrors];
|
|
|
|
[self writeWithBlock:^(SDSAnyWriteTransaction *transaction) {
|
|
[self.messageManager handleIncomingEnvelope:[envelopeBuilder buildIgnoringErrors]
|
|
withDataMessage:[messageBuilder buildIgnoringErrors]
|
|
plaintextData:[NSData data]
|
|
wasReceivedByUD:NO
|
|
serverDeliveryTimestamp:0
|
|
shouldDiscardVisibleMessages:NO
|
|
transaction:transaction];
|
|
}];
|
|
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
TSGroupThread *_Nullable thread = [TSGroupThread fetchWithGroupId:groupId transaction:transaction];
|
|
XCTAssertNotNil(thread);
|
|
XCTAssertEqualObjects(@"Newly created Group Name", thread.groupNameOrDefault);
|
|
}];
|
|
}
|
|
|
|
- (void)test_GroupUpdateWithAvatar
|
|
{
|
|
// GroupsV2 TODO: Handle v2 groups.
|
|
NSData *groupId = [TSGroupModel generateRandomV1GroupId];
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
TSGroupThread *_Nullable thread = [TSGroupThread fetchWithGroupId:groupId transaction:transaction];
|
|
XCTAssertNil(thread);
|
|
}];
|
|
|
|
SSKProtoEnvelopeBuilder *envelopeBuilder = [SSKProtoEnvelope builderWithTimestamp:12345];
|
|
[envelopeBuilder setSourceUuid:NSUUID.UUID.UUIDString];
|
|
[envelopeBuilder setType:SSKProtoEnvelopeTypeCiphertext];
|
|
|
|
SSKProtoGroupContextBuilder *groupContextBuilder =
|
|
[SSKProtoGroupContext builderWithId:groupId];
|
|
[groupContextBuilder setType:SSKProtoGroupContextTypeUpdate];
|
|
[groupContextBuilder setName:@"Newly created Group Name"];
|
|
|
|
SSKProtoAttachmentPointerBuilder *attachmentBuilder = [SSKProtoAttachmentPointer builder];
|
|
attachmentBuilder.cdnID = 1234;
|
|
[attachmentBuilder setContentType:@"image/png"];
|
|
[attachmentBuilder setKey:[Cryptography generateRandomBytes:32]];
|
|
[attachmentBuilder setSize:123];
|
|
[groupContextBuilder setAvatar:[attachmentBuilder buildIgnoringErrors]];
|
|
|
|
SSKProtoDataMessageBuilder *messageBuilder = [SSKProtoDataMessage builder];
|
|
messageBuilder.group = [groupContextBuilder buildIgnoringErrors];
|
|
|
|
[self writeWithBlock:^(SDSAnyWriteTransaction *transaction) {
|
|
[self.messageManager handleIncomingEnvelope:[envelopeBuilder buildIgnoringErrors]
|
|
withDataMessage:[messageBuilder buildIgnoringErrors]
|
|
plaintextData:[NSData data]
|
|
wasReceivedByUD:NO
|
|
serverDeliveryTimestamp:0
|
|
shouldDiscardVisibleMessages:NO
|
|
transaction:transaction];
|
|
}];
|
|
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
TSGroupThread *_Nullable thread = [TSGroupThread fetchWithGroupId:groupId transaction:transaction];
|
|
XCTAssertNotNil(thread);
|
|
XCTAssertEqualObjects(@"Newly created Group Name", thread.groupNameOrDefault);
|
|
}];
|
|
}
|
|
|
|
@end
|
|
|
|
NS_ASSUME_NONNULL_END
|