test: add nil check and fix/update tests

amplitude · Apr 4, 2024 · 2fc7f3e · 2fc7f3e
1 parent eca6cc2
commit 2fc7f3e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/Sources/Amplitude/Amplitude.m b/Sources/Amplitude/Amplitude.m
@@ -1669,7 +1669,10 @@ - (NSString *)getAdSupportID {
 }
 
 - (NSString *)getDeviceId {
-    return [self initializeDeviceId];
+    if (self.deviceId == nil) {
+        return [self initializeDeviceId];
+    }
+    return self.deviceId;
 }
 
 - (long long)getSessionId {

diff --git a/Tests/AmplitudeTests.m b/Tests/AmplitudeTests.m
@@ -1705,4 +1705,12 @@ - (void)testOpenURLFiresDeepLinkEvent {
     XCTAssertEqualObjects([[event objectForKey:@"event_properties"] objectForKey:kAMPEventPropLinkUrl], @"https://test-app.com");
 }
 
+- (void)testGetDeviceIdAfterInit {
+    NSString *instanceName = @"testGetDeviceIdAfterInit";
+    Amplitude *client = [Amplitude instanceWithName:instanceName];
+    [client initializeApiKey:@"test"];
+    NSString *deviceId = [client getDeviceId];
+    XCTAssertNotEqual(deviceId, @"");
+}
+
 @end
diff --git a/Tests/SetupTests.m b/Tests/SetupTests.m
@@ -115,15 +115,12 @@ - (void)testUserPropertiesSet {
 
 - (void)testSetDeviceId {
     AMPDatabaseHelper *dbHelper = [AMPDatabaseHelper getDatabaseHelper];
-    NSString *initialDeviceId = [self.amplitude getDeviceId];
-    XCTAssertNil(initialDeviceId);  // device id not initialized yet
 
     [self.amplitude initializeApiKey:apiKey];
     [self.amplitude flushQueueWithQueue:self.amplitude.initializerQueue];
     [self.amplitude flushQueue];
     NSString *generatedDeviceId = [self.amplitude getDeviceId];
     XCTAssertNotNil(generatedDeviceId);
-    XCTAssertNotEqualObjects(initialDeviceId, generatedDeviceId);
 #if !TARGET_OS_OSX
     XCTAssertEqual(generatedDeviceId.length, 36);
 #else