diff options
Diffstat (limited to 'wrappers/python/test_discovery.py')
| -rwxr-xr-x | wrappers/python/test_discovery.py | 78 |
1 files changed, 0 insertions, 78 deletions
diff --git a/wrappers/python/test_discovery.py b/wrappers/python/test_discovery.py deleted file mode 100755 index a1a70af..0000000 --- a/wrappers/python/test_discovery.py +++ /dev/null @@ -1,78 +0,0 @@ -#!/usr/bin/env python3 - -import unittest -import eduvpncommon.discovery as discovery - -test_data_dir = "../../src/test_data" - - -def read_bytes(path: str) -> bytes: - with open(path, "rb") as f: - return f.read() - - -class VerifyTests(unittest.TestCase): - @classmethod - def setUpClass(cls) -> None: - with open(f"{test_data_dir}/public.key") as f: - discovery._insecure_testing_set_extra_key(f.readlines()[-1][:-1]) - - def testValid(self): - discovery.verify( - read_bytes(f"{test_data_dir}/server_list.json.minisig"), - read_bytes(f"{test_data_dir}/server_list.json"), - "server_list.json", - 10 - ) - - def testValidMemoryView(self): - discovery.verify( - read_bytes(f"{test_data_dir}/server_list.json.minisig"), - memoryview(b"abc" + read_bytes(f"{test_data_dir}/server_list.json") + b"abc")[3:-3], - "server_list.json", - 0 - ) - - def testInvalidSignature(self): - with self.assertRaises(discovery.VerifyError) as ctx: - discovery.verify( - read_bytes(f"{test_data_dir}/random.txt"), - read_bytes(f"{test_data_dir}/server_list.json"), - "server_list.json", - 0 - ) - self.assertEqual(ctx.exception.code, discovery.VerifyErrorCode.ErrInvalidSignature) - - def testWrongKey(self): - with self.assertRaises(discovery.VerifyError) as ctx: - discovery.verify( - read_bytes(f"{test_data_dir}/server_list.json.wrong_key.minisig"), - read_bytes(f"{test_data_dir}/server_list.json"), - "server_list.json", - 0 - ) - self.assertEqual(ctx.exception.code, discovery.VerifyErrorCode.ErrInvalidSignatureUnknownKey) - - def testOldSignature(self): - with self.assertRaises(discovery.VerifyError) as ctx: - discovery.verify( - read_bytes(f"{test_data_dir}/server_list.json.minisig"), - read_bytes(f"{test_data_dir}/server_list.json"), - "server_list.json", - 11 - ) - self.assertEqual(ctx.exception.code, discovery.VerifyErrorCode.ErrTooOld) - - def TestUnknownExpectedFile(self): - with self.assertRaises(discovery.VerifyError) as ctx: - discovery.verify( - read_bytes(f"{test_data_dir}/other_list.json.minisig"), - read_bytes(f"{test_data_dir}/other_list.json"), - "other_list.json", - 0 - ) - self.assertEqual(ctx.exception.code, discovery.VerifyErrorCode.ErrUnknownExpectedFileName) - - -if __name__ == "__main__": - unittest.main() |
