commit ae5d78b838fd910d042bdddd65ff96fbc8a1103f Author: Arturo Filastò arturo@filasto.net Date: Tue Jul 26 12:39:40 2016 +0200
Add two basic unittests --- ooni/tests/test_resources.py | 41 +++++++++++++++++++++++++++++++++++++++++ ooni/tests/test_socks.py | 10 ++++++++++ 2 files changed, 51 insertions(+)
diff --git a/ooni/tests/test_resources.py b/ooni/tests/test_resources.py new file mode 100644 index 0000000..92961ea --- /dev/null +++ b/ooni/tests/test_resources.py @@ -0,0 +1,41 @@ +from ooni.resources import get_out_of_date_resources, check_for_update +from ooni.tests.bases import ConfigTestCase + +SAMPLE_CURRENT_MANIFEST = { + "resources": [ + { + "version": 0, + "path": "some/file-to-update.txt" + }, + { + "version": 0, + "path": "some/file-stays-stame.txt" + }, + { + "version": 0, + "path": "some/file-to-delete.txt" + } + ] +} + +SAMPLE_NEW_MANIFEST = { + "resources": [ + { + "version": 1, + "path": "some/file-to-update.txt" + }, + { + "version": 0, + "path": "some/file-stays-stame.txt" + } + ] +} +class TestResourceUpdate(ConfigTestCase): + def test_check_for_updates(self): + return check_for_update() + + def test_resources_out_of_date(self): + paths_to_update, paths_to_delete = get_out_of_date_resources( + SAMPLE_CURRENT_MANIFEST, SAMPLE_NEW_MANIFEST) + self.assertEqual(paths_to_update[0]["path"], "some/file-to-update.txt") + self.assertEqual(paths_to_delete[0]["path"], "some/file-to-delete.txt") diff --git a/ooni/tests/test_socks.py b/ooni/tests/test_socks.py new file mode 100644 index 0000000..2d2cdd8 --- /dev/null +++ b/ooni/tests/test_socks.py @@ -0,0 +1,10 @@ +from twisted.trial import unittest + +from twisted.internet.endpoints import TCP4ClientEndpoint +from twisted.internet import reactor +from ooni.utils.socks import TrueHeadersSOCKS5Agent + +class TestSocks(unittest.TestCase): + def test_create_agent(self): + proxyEndpoint = TCP4ClientEndpoint(reactor, '127.0.0.1', 9050) + agent = TrueHeadersSOCKS5Agent(reactor, proxyEndpoint=proxyEndpoint)