Revize 2f2eab4c

re #9421 Added unit tests for get_metadata() and send_payload_to_server()

client/src/usb_detector/event_listener.py

@@ -3 +3 @@
 import getpass
 from datetime import datetime
 
-from api_client import send_data
+from ..api_client import send_data
 
 
 def _get_metadata() -> dict:
@@ -11 +11 @@
     return {
         "username": getpass.getuser(),
         "hostname": platform.uname().node,
-        "timestamp": str(datetime.now())
+        "timestamp": str(datetime.now()).split('.')[0]
     }
 
 

client/tests/event_listener/test_get_metadata.py

@@ +1 @@
+import getpass
+import platform
+from datetime import datetime
+
+from client.src.usb_detector import event_listener
+
+
+def test_get_metadata_1():
+    expected_metadata = {
+        "username": getpass.getuser(),
+        "hostname": platform.uname().node,
+        "timestamp": str(datetime.now()).split('.')[0]
+    }
+
+    actual_metadata = event_listener._get_metadata()
+
+    assert actual_metadata["username"] == expected_metadata["username"]
+    assert actual_metadata["hostname"] == expected_metadata["hostname"]
+    assert actual_metadata["timestamp"] == expected_metadata["timestamp"]

client/tests/event_listener/test_send_payload_to_server.py

@@ +1 @@
+import platform
+import getpass
+from unittest import mock
+from datetime import datetime
+
+import client.src.usb_detector.event_listener
+
+
[e-mail]('client.src.usb_detector.event_listener.send_data')
+def test_send_payload_to_server_1(send_data_mock):
+    device_mock = {
+        "vendor_id": 1,
+        "product_id": 2
+    }
+    metadata_mock = {
+        "username": getpass.getuser(),
+        "hostname": platform.uname().node,
+        "timestamp": str(datetime.now()).split('.')[0]
+    }
+    status_mock = "connected"
+
+    metadata_mock["device"] = device_mock
+    metadata_mock["status"] = status_mock
+
+    client.src.usb_detector.event_listener._send_payload_to_server(device_mock, status_mock)
+
+    args = send_data_mock.call_args.args
+    send_data_mock.assert_called()
+    assert args[0] == metadata_mock