AMP-36170 add more unit test

amplitude · Feb 7, 2024 · 06b5eaf · 06b5eaf
1 parent 9b20ade
commit 06b5eaf
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/Tests/AMPUtilsTests.m b/Tests/AMPUtilsTests.m
@@ -13,10 +13,19 @@ @interface AMPUtilTests : XCTestCase
 
 @end
 
-@implementation AMPUtilTests {
-
+@interface FakeAMPUtils: AMPUtils
+@end
+
+@implementation FakeAMPUtils
+
++ (NSDictionary *)getEnvironment {
+    return @{@"APP_SANDBOX_CONTAINER_ID": @"test-container-id"};
 }
 
+@end
+
+@implementation AMPUtilTests {}
+
 - (void) testIsSandboxEnabled {
     BOOL isSandboxEnabled = [AMPUtils isSandboxEnabled];
     #if TARGET_OS_OSX
@@ -26,4 +35,11 @@ - (void) testIsSandboxEnabled {
     #endif
 }
 
+#if TARGET_OS_OSX
+- (void) testIsSandboxEnabledWhenMacOSIsSandboxed {
+    BOOL isSandboxEnabled = [FakeAMPUtils isSandboxEnabled];
+    XCTAssertEqual(isSandboxEnabled, YES);
+}
+#endif
+
 @end