summaryrefslogtreecommitdiff
path: root/wrappers/python/test_discovery.py
diff options
context:
space:
mode:
Diffstat (limited to 'wrappers/python/test_discovery.py')
-rwxr-xr-xwrappers/python/test_discovery.py78
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()