1
|
import platform
|
2
|
import getpass
|
3
|
from unittest import mock
|
4
|
from datetime import datetime
|
5
|
|
6
|
import client.src.usb_detector.event_listener
|
7
|
|
8
|
|
9
|
@mock.patch('client.src.usb_detector.event_listener.send_data')
|
10
|
def test_send_payload_to_server_1(send_data_mock):
|
11
|
device_mock = {
|
12
|
"vendor_id": 1,
|
13
|
"product_id": 2
|
14
|
}
|
15
|
metadata_mock = {
|
16
|
"username": getpass.getuser(),
|
17
|
"hostname": platform.uname().node,
|
18
|
"timestamp": str(datetime.now()).split('.')[0]
|
19
|
}
|
20
|
status_mock = "connected"
|
21
|
|
22
|
metadata_mock["device"] = device_mock
|
23
|
metadata_mock["status"] = status_mock
|
24
|
|
25
|
client.src.usb_detector.event_listener._send_payload_to_server(device_mock, status_mock)
|
26
|
|
27
|
args = send_data_mock.call_args.args
|
28
|
send_data_mock.assert_called()
|
29
|
assert args[0] == metadata_mock
|