Unit test for migrating from OSInAppMessage to OSInAppMessageInternal

Author: emawby

iOS_SDK/OneSignalSDK/UnitTests/MigrationTests.m

@@ -288,6 +288,40 @@ - (void)testIAMNilCacheToNilMigration {
     XCTAssertNil(retrievedDict);
 }
 
+- (void)testIAMToInternalMigration {
+    let limit = 5;
+    let delay = 60;
+    let message = [OSInAppMessageTestHelper testMessageWithRedisplayLimit:limit delay:@(delay)];
+    message.isDisplayedInSession = true;
+    
+    // Cached Messages
+    NSArray<OSInAppMessageInternal *> *messages = [[NSArray alloc] initWithObjects:message, nil];
+    
+    [NSKeyedArchiver setClassName:@"OSInAppMessage" forClass:[OSInAppMessageInternal class]];
+    
+    [OneSignalUserDefaults.initStandard saveCodeableDataForKey:OS_IAM_MESSAGES_ARRAY withValue:messages];
+    
+    [migrationController migrate];
+    
+    NSArray<OSInAppMessageInternal *>*retrievedArray = [OneSignalUserDefaults.initStandard
+                                                                getSavedCodeableDataForKey:OS_IAM_MESSAGES_ARRAY defaultValue:nil];
+    XCTAssertEqualObjects(messages, retrievedArray);
+    
+    // Cached Redisplay Messages
+    NSMutableDictionary <NSString *, OSInAppMessageInternal *> *redisplayedInAppMessages = [NSMutableDictionary new];
+    [redisplayedInAppMessages setObject:message forKey:message.messageId];
+    
+    [NSKeyedArchiver setClassName:@"OSInAppMessage" forClass:[OSInAppMessageInternal class]];
+    
+    [OneSignalUserDefaults.initStandard saveCodeableDataForKey:OS_IAM_REDISPLAY_DICTIONARY withValue:redisplayedInAppMessages];
+    
+    [migrationController migrate];
+    
+    NSDictionary<NSString *, OSInAppMessageInternal *>*retrievedDict = [OneSignalUserDefaults.initStandard
+                                                                getSavedCodeableDataForKey:OS_IAM_REDISPLAY_DICTIONARY defaultValue:nil];
+    XCTAssertEqualObjects(redisplayedInAppMessages, retrievedDict);
+}
+