1/* 2 * Copyright (C) 2024 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17import {UserNotifier} from 'common/user_notifier'; 18import {UserNotification} from 'messaging/user_notification'; 19 20export class UserNotifierChecker { 21 private userNotifierAdd: jasmine.Spy<jasmine.Func>; 22 private userNotifierNotify: jasmine.Spy<jasmine.Func>; 23 24 constructor() { 25 this.userNotifierAdd = spyOn(UserNotifier, 'add').and.callThrough(); 26 this.userNotifierNotify = spyOn(UserNotifier, 'notify'); 27 } 28 29 expectNone() { 30 expect(this.userNotifierAdd).not.toHaveBeenCalled(); 31 } 32 33 expectAdded(notifications: UserNotification[]) { 34 this.checkAdded(notifications); 35 expect(this.userNotifierNotify).not.toHaveBeenCalled(); 36 } 37 38 expectNotified(notifications: UserNotification[]) { 39 this.checkAdded(notifications); 40 expect(this.userNotifierNotify).toHaveBeenCalled(); 41 } 42 43 reset() { 44 this.userNotifierAdd.calls.reset(); 45 this.userNotifierNotify.calls.reset(); 46 } 47 48 private checkAdded(notifications: UserNotification[]) { 49 expect(this.userNotifierAdd).toHaveBeenCalledTimes(notifications.length); 50 notifications.forEach((n) => 51 expect(this.userNotifierAdd).toHaveBeenCalledWith(n), 52 ); 53 } 54} 55