72 lines
2.9 KiB
Objective-C
72 lines
2.9 KiB
Objective-C
//
|
|
// Copyright 2018 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
#import "SSKBaseTestObjC.h"
|
|
#import "TSAttachmentStream.h"
|
|
#import "TSContactThread.h"
|
|
#import "TSIncomingMessage.h"
|
|
#import "TSOutgoingMessage.h"
|
|
#import "TestAppContext.h"
|
|
#import <SignalServiceKit/SignalServiceKit-Swift.h>
|
|
|
|
@interface TSThreadTest : SSKBaseTestObjC
|
|
|
|
@end
|
|
|
|
#pragma mark -
|
|
|
|
@implementation TSThreadTest
|
|
|
|
- (NSUInteger)numberOfInteractionsInThread:(TSThread *)thread transaction:(SDSAnyReadTransaction *)transaction
|
|
{
|
|
InteractionFinder *finder = [[InteractionFinder alloc] initWithThreadUniqueId:thread.uniqueId];
|
|
__block NSUInteger result = 0;
|
|
[finder enumerateInteractionIdsWithTransaction:transaction
|
|
error:NULL
|
|
block:^(NSString *uniqueId, BOOL *stop) { result += 1; }];
|
|
return result;
|
|
}
|
|
|
|
- (void)testDeletingThreadDeletesInteractions
|
|
{
|
|
AciObjC *aci = [[AciObjC alloc] initWithAciString:@"00000000-0000-4000-8000-000000000000"];
|
|
TSContactThread *thread =
|
|
[[TSContactThread alloc] initWithContactAddress:[[SignalServiceAddress alloc] initWithServiceIdObjC:aci]];
|
|
[self writeWithBlock:^(SDSAnyWriteTransaction *transaction) { [thread anyInsertWithTransaction:transaction]; }];
|
|
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
XCTAssertEqual(0, [self numberOfInteractionsInThread:thread transaction:transaction]);
|
|
}];
|
|
|
|
TSIncomingMessageBuilder *incomingMessageBuilder =
|
|
[TSIncomingMessageBuilder withDefaultsObjcWithThread:thread
|
|
messageBody:@"Incoming message body"
|
|
timestamp:10000
|
|
authorAci:aci];
|
|
TSIncomingMessage *incomingMessage = [incomingMessageBuilder build];
|
|
[self writeWithBlock:^(
|
|
SDSAnyWriteTransaction *transaction) { [incomingMessage anyInsertWithTransaction:transaction]; }];
|
|
|
|
TSOutgoingMessageBuilder *messageBuilder =
|
|
[TSOutgoingMessageBuilder outgoingMessageBuilderWithThread:thread messageBody:@"outgoing message body"];
|
|
messageBuilder.timestamp = 20000;
|
|
TSOutgoingMessage *outgoingMessage = [messageBuilder buildWithSneakyTransaction];
|
|
[self writeWithBlock:^(
|
|
SDSAnyWriteTransaction *transaction) { [outgoingMessage anyInsertWithTransaction:transaction]; }];
|
|
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
XCTAssertEqual(2, [self numberOfInteractionsInThread:thread transaction:transaction]);
|
|
}];
|
|
|
|
[self writeWithBlock:^(
|
|
SDSAnyWriteTransaction *transaction) { [thread softDeleteThreadWithTransaction:transaction]; }];
|
|
[self readWithBlock:^(SDSAnyReadTransaction *transaction) {
|
|
XCTAssertEqual(0, [self numberOfInteractionsInThread:thread transaction:transaction]);
|
|
XCTAssertEqual(0, [TSInteraction anyCountWithTransaction:transaction]);
|
|
}];
|
|
}
|
|
|
|
@end
|