|
1
|
from unittest.mock import Mock
|
|
2
|
|
|
3
|
from client.src.usb_detector import detector
|
|
4
|
|
|
5
|
|
|
6
|
def test_notify_listeners_1():
|
|
7
|
mocks = [Mock() for _ in range(10)]
|
|
8
|
detector._notify_listeners([mock.listener for mock in mocks], [1, 2, 3])
|
|
9
|
for mock in mocks:
|
|
10
|
mock.listener.assert_called()
|
|
11
|
|
|
12
|
|
|
13
|
def test_notify_listeners_2():
|
|
14
|
mocks = [Mock() for _ in range(10)]
|
|
15
|
detector._notify_listeners([mock.listener for mock in mocks], [])
|
|
16
|
for mock in mocks:
|
|
17
|
mock.listener.assert_not_called()
|
|
18
|
|
|
19
|
|
|
20
|
def test_notify_listeners_3():
|
|
21
|
mocks = [Mock() for _ in range(10)]
|
|
22
|
detector._notify_listeners([mock.listener for mock in mocks], None)
|
|
23
|
for mock in mocks:
|
|
24
|
mock.listener.assert_not_called()
|