mirror of
https://github.com/google/santa.git
synced 2026-01-14 08:47:57 -05:00
c1189493e8
This allows a sync server to send a `custom_url` field along with a rule blocking execution and this will be used as the URL for the "open" button in place of the normally generated URL.
76 lines
2.9 KiB
Objective-C
76 lines
2.9 KiB
Objective-C
/// Copyright 2022 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 <OCMock/OCMock.h>
|
|
#import <XCTest/XCTest.h>
|
|
|
|
#import "Source/gui/SNTNotificationManager.h"
|
|
|
|
#import "Source/common/SNTStoredEvent.h"
|
|
|
|
@class SNTBinaryMessageWindowController;
|
|
|
|
@interface SNTNotificationManager (Testing)
|
|
- (void)hashBundleBinariesForEvent:(SNTStoredEvent *)event
|
|
withController:(SNTBinaryMessageWindowController *)controller;
|
|
@end
|
|
|
|
@interface SNTNotificationManagerTest : XCTestCase
|
|
@end
|
|
|
|
@implementation SNTNotificationManagerTest
|
|
|
|
- (void)setUp {
|
|
[super setUp];
|
|
fclose(stdout);
|
|
}
|
|
|
|
- (void)testPostBlockNotificationSendsDistributedNotification {
|
|
SNTStoredEvent *ev = [[SNTStoredEvent alloc] init];
|
|
ev.fileSHA256 = @"the-sha256";
|
|
ev.filePath = @"/Applications/Safari.app/Contents/MacOS/Safari";
|
|
ev.fileBundleName = @"Safari";
|
|
ev.fileBundlePath = @"/Applications/Safari.app";
|
|
ev.fileBundleID = @"com.apple.Safari";
|
|
ev.fileBundleVersion = @"18614.1.14.1.15";
|
|
ev.fileBundleVersionString = @"16.0";
|
|
ev.executingUser = @"rah";
|
|
ev.occurrenceDate = [NSDate dateWithTimeIntervalSince1970:1660221048];
|
|
ev.decision = SNTEventStateBlockBinary;
|
|
ev.pid = @84156;
|
|
ev.ppid = @1;
|
|
ev.parentName = @"launchd";
|
|
|
|
SNTNotificationManager *sut = OCMPartialMock([[SNTNotificationManager alloc] init]);
|
|
OCMStub([sut hashBundleBinariesForEvent:OCMOCK_ANY withController:OCMOCK_ANY]).andDo(nil);
|
|
|
|
id dncMock = OCMClassMock([NSDistributedNotificationCenter class]);
|
|
OCMStub([dncMock defaultCenter]).andReturn(dncMock);
|
|
|
|
[sut postBlockNotification:ev withCustomMessage:@"" andCustomURL:@""];
|
|
|
|
OCMVerify([dncMock postNotificationName:@"com.google.santa.notification.blockedeexecution"
|
|
object:@"com.google.santa"
|
|
userInfo:[OCMArg checkWithBlock:^BOOL(NSDictionary *userInfo) {
|
|
XCTAssertEqualObjects(userInfo[@"file_sha256"], @"the-sha256");
|
|
XCTAssertEqualObjects(userInfo[@"pid"], @84156);
|
|
XCTAssertEqualObjects(userInfo[@"ppid"], @1);
|
|
XCTAssertEqualObjects(userInfo[@"execution_time"], @1660221048);
|
|
return YES;
|
|
}]
|
|
deliverImmediately:YES]);
|
|
}
|
|
|
|
@end
|