mirror of
https://github.com/google/santa.git
synced 2026-01-14 08:47:57 -05:00
70 lines
2.2 KiB
Objective-C
70 lines
2.2 KiB
Objective-C
/// Copyright 2021 Google Inc. All rights reserved.
|
|
///
|
|
/// Licensed under the Apache License, Version 2.0 (the "License");
|
|
/// you may not use this file except in compliance with the License.
|
|
/// You may obtain a copy of the License at
|
|
///
|
|
/// http://www.apache.org/licenses/LICENSE-2.0
|
|
///
|
|
/// Unless required by applicable law or agreed to in writing, software
|
|
/// distributed under the License is distributed on an "AS IS" BASIS,
|
|
/// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
/// See the License for the specific language governing permissions and
|
|
/// limitations under the License.
|
|
#import <Foundation/Foundation.h>
|
|
#import <XCTest/XCTest.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
@interface SNTExecTest : XCTestCase
|
|
@end
|
|
|
|
@implementation SNTExecTest : XCTestCase
|
|
- (void)setUp {
|
|
[super setUp];
|
|
fclose(stdout);
|
|
}
|
|
|
|
- (void)checkExecution:(NSString *)path shouldExec:(BOOL)shouldExec {
|
|
__block int status;
|
|
|
|
XCTestExpectation *expectation =
|
|
[self expectationWithDescription:@"Wait for test binary to execute"];
|
|
|
|
__block NSTask *task = [[NSTask alloc] init];
|
|
dispatch_async(dispatch_get_global_queue(QOS_CLASS_BACKGROUND, 0), ^{
|
|
task.launchPath = path;
|
|
[task launch];
|
|
[task waitUntilExit];
|
|
status = [task terminationStatus];
|
|
[expectation fulfill];
|
|
});
|
|
|
|
[self waitForExpectationsWithTimeout:20.0
|
|
handler:^(NSError *error) {
|
|
XCTAssertFalse([task isRunning], @"Test timed out: %@", error);
|
|
[task terminate];
|
|
}];
|
|
|
|
BOOL didExec = (status == 0);
|
|
XCTAssertEqual(didExec, shouldExec);
|
|
}
|
|
|
|
- (void)testShouldExecute {
|
|
NSString *testPath = @"santa/Source/santad/testdata/binaryrules";
|
|
NSString *fullTestPath = [NSString pathWithComponents:@[
|
|
[[[NSProcessInfo processInfo] environment] objectForKey:@"TEST_SRCDIR"], testPath
|
|
]];
|
|
NSDictionary *testCases = @{
|
|
@"goodbinary" : @YES,
|
|
@"badbinary" : @NO,
|
|
@"noop" : @YES,
|
|
};
|
|
for (NSString *binary in testCases) {
|
|
[self checkExecution:[NSString pathWithComponents:@[ fullTestPath, binary ]]
|
|
shouldExec:[testCases[binary] boolValue]];
|
|
}
|
|
}
|
|
|
|
@end
|