session-ios/SignalServiceKit/tests/Messages/Interactions/TSMessageTest.m

//
//  Copyright (c) 2019 Open Whisper Systems. All rights reserved.
//

#import "TSMessage.h"
#import "SSKBaseTestObjC.h"
#import "TSAttachmentStream.h"
#import "TSContactThread.h"
#import <SignalCoreKit/NSDate+OWS.h>

NS_ASSUME_NONNULL_BEGIN

@interface TSMessageTest : SSKBaseTestObjC

@property TSThread *thread;

@end

@implementation TSMessageTest

- (void)setUp {
    [super setUp];
    self.thread = [TSContactThread getOrCreateThreadWithContactId:@"fake-thread-id"];
}

- (void)tearDown {
    // Put teardown code here. This method is called after the invocation of each test method in the class.
    [super tearDown];
}

- (void)testExpiresAtWithoutStartedTimer
{
    TSMessage *message = [[TSMessage alloc] initMessageWithTimestamp:1
                                                            inThread:self.thread
                                                         messageBody:@"foo"
                                                       attachmentIds:@[]
                                                    expiresInSeconds:100
                                                     expireStartedAt:0
                                                       quotedMessage:nil
                                                        contactShare:nil
                                                         linkPreview:nil];

    XCTAssertEqual(0, message.expiresAt);
}

- (void)testExpiresAtWithStartedTimer
{
    uint64_t now = [NSDate ows_millisecondTimeStamp];
    const uint32_t expirationSeconds = 10;
    const uint32_t expirationMs = expirationSeconds * 1000;
    TSMessage *message = [[TSMessage alloc] initMessageWithTimestamp:1
                                                            inThread:self.thread
                                                         messageBody:@"foo"
                                                       attachmentIds:@[]
                                                    expiresInSeconds:expirationSeconds
                                                     expireStartedAt:now
                                                       quotedMessage:nil
                                                        contactShare:nil
                                                         linkPreview:nil];
    XCTAssertEqual(now + expirationMs, message.expiresAt);
}

@end

NS_ASSUME_NONNULL_END