bit-bots · jaagut · Dec 19, 2024 · Dec 16, 2024
diff --git a/package.xml b/package.xml
@@ -18,7 +18,7 @@
   <depend>bitbots_utils</depend>
   <depend>python3-cryptography</depend>
 
-  <test_depend>python3-hypothesis</test_depend>
+  <test_depend>python3-pytest</test_depend>
 
   <export>
     <bitbots_documentation>

diff --git a/setup.py b/setup.py
@@ -17,6 +17,7 @@
         "launch",
         "setuptools",
     ],
+    tests_require=["pytest"],
     zip_safe=True,
     keywords=["ROS"],
     license="MIT",

diff --git a/test/rostests/test_sender.launch b/test/rostests/test_sender.launch
diff --git a/test/rostests/test_sender.py b/test/rostests/test_sender.py
diff --git a/test/rostests/test_sender_receiver.launch b/test/rostests/test_sender_receiver.launch
diff --git a/test/rostests/test_sender_receiver.py b/test/rostests/test_sender_receiver.py
diff --git a/test/udp_bridge/test_aes_helper.py b/test/udp_bridge/test_aes_helper.py
@@ -0,0 +1,9 @@
+from udp_bridge import aes_helper
+
+
+def test_decrypt_inverts_encrypt():
+    message = b"Hello, World!"
+    encrypted = aes_helper.AESCipher("key").encrypt(message)
+    decrypted = aes_helper.AESCipher("key").decrypt(encrypted)
+
+    assert message == decrypted
diff --git a/test/unit_tests/test_aes_helper.py b/test/unit_tests/test_aes_helper.py