Add test coverage for syncing USB mounting options (#711)

google · Jan 11, 2022 · fbd0de3 · fbd0de3
1 parent 6f2ae62
commit fbd0de3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/Source/santasyncservice/SNTSyncTest.m b/Source/santasyncservice/SNTSyncTest.m
@@ -198,6 +198,18 @@ - (void)testPreflightBasicResponse {
   XCTAssertNil(self.syncState.blocklistRegex);
 }
 
+- (void)testPreflightBlockUSBMount {
+  SNTSyncPreflight *sut = [[SNTSyncPreflight alloc] initWithState:self.syncState];
+
+  NSData *respData = [self dataFromFixture:@"sync_preflight_toggle_blockusb.json"];
+  [self stubRequestBody:respData response:nil error:nil validateBlock:nil];
+
+  XCTAssertTrue([sut sync]);
+  XCTAssertEqual(self.syncState.blockUSBMount, true);
+  NSArray<NSString *> *wantRemountUSBMode = @[ @"rdonly", @"noexec" ];
+  XCTAssertEqualObjects(self.syncState.remountUSBMode, wantRemountUSBMode);
+}
+
 - (void)testPreflightDatabaseCounts {
   SNTSyncPreflight *sut = [[SNTSyncPreflight alloc] initWithState:self.syncState];
 

diff --git a/Source/santasyncservice/testdata/sync_preflight_toggle_blockusb.json b/Source/santasyncservice/testdata/sync_preflight_toggle_blockusb.json
@@ -0,0 +1 @@
+{"allowed_path_regex": null, "client_mode": "LOCKDOWN", "blocked_path_regex": null, "batch_size": 100, "block_usb_mount":true, "remount_usb_mode": ["rdonly", "noexec"]}
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		{"allowed_path_regex": null, "client_mode": "LOCKDOWN", "blocked_path_regex": null, "batch_size": 100, "block_usb_mount":true, "remount_usb_mode": ["rdonly", "noexec"]}