pytest: Add tests for pd_id and pd_cap API

Related-to: #181
Signed-off-by: Siddharth Chandrasekaran <[email protected]>

tests/pytest/test_data.py

@@ -0,0 +1,62 @@
+#
+#  Copyright (c) 2024 Siddharth Chandrasekaran <[email protected]>
+#
+#  SPDX-License-Identifier: Apache-2.0
+#
+
+import pytest
+
+from osdp import *
+from conftest import make_fifo_pair, cleanup_fifo_pair
+
+pd_cap = PDCapabilities([
+    (Capability.OutputControl, 1, 1),
+    (Capability.LEDControl, 1, 2),
+    (Capability.AudibleControl, 1, 3),
+    (Capability.TextOutput, 1, 4),
+])
+
+key = KeyStore.gen_key()
+f1, f2 = make_fifo_pair("data")
+
+pd_id = PdId(1, 1, 314, 512, 443)
+
+# TODO remove this.
+pd_addr = 101
+pd = PeripheralDevice(PDInfo(pd_addr, f1, scbk=key, id=pd_id), pd_cap, log_level=LogLevel.Debug)
+cp = ControlPanel([PDInfo(pd_addr, f2, scbk=key, id=pd_id)])
+
+@pytest.fixture(scope='module', autouse=True)
+def setup_test():
+    pd.start()
+    cp.start()
+    cp.sc_wait_all()
+    yield
+    teardown_test()
+
+def teardown_test():
+    cp.teardown()
+    pd.teardown()
+    cleanup_fifo_pair("data")
+
+def test_cp_pd_id():
+    assert cp.online_wait(pd.address)
+    pd_id_recv = cp.get_pd_id(pd.address)
+    assert pd_id_recv.version == pd_id.version
+    assert pd_id_recv.model == pd_id.model
+    assert pd_id_recv.vendor_code == pd_id.vendor_code
+    assert pd_id_recv.serial_number == pd_id.serial_number
+    assert pd_id_recv.firmware_version == pd_id.firmware_version
+
+def test_cp_check_capability():
+    assert cp.online_wait(pd.address)
+    # check libosdp default capabilities
+    assert cp.check_capability(pd.address, Capability.CheckCharacter) == (1, 0)
+    assert cp.check_capability(pd.address, Capability.CommunicationSecurity) == (1, 0)
+    assert cp.check_capability(pd.address, Capability.ReceiveBufferSize) == (0, 1)
+
+    # check local capabilities
+    assert cp.check_capability(pd.address, Capability.OutputControl) == (1, 1)
+    assert cp.check_capability(pd.address, Capability.LEDControl) == (1, 2)
+    assert cp.check_capability(pd.address, Capability.AudibleControl) == (1, 3)
+    assert cp.check_capability(pd.address, Capability.TextOutput) == (1, 4)