diff options
Diffstat (limited to 'tests/test_leap_backend.py')
| -rw-r--r-- | tests/test_leap_backend.py | 168 |
1 files changed, 127 insertions, 41 deletions
diff --git a/tests/test_leap_backend.py b/tests/test_leap_backend.py index 72106aef..97de101f 100644 --- a/tests/test_leap_backend.py +++ b/tests/test_leap_backend.py @@ -4,14 +4,20 @@ For these tests to run, a leap server has to be running on (default) port 5984. """ -import sys +from shutil import rmtree +import os import copy +import unittest2 as unittest import testtools import testscenarios -from u1db import errors, Document +import u1db +from leap.soledad import Soledad from leap.soledad.backends import leap_backend from leap.soledad.tests import u1db_tests as tests -from leap.soledad.tests.u1db_tests.test_remote_sync_target import make_http_app +from leap.soledad.tests.u1db_tests.test_remote_sync_target import ( + make_http_app, + make_oauth_http_app, +) from leap.soledad.tests.u1db_tests.test_backends import AllDatabaseTests from leap.soledad.tests.u1db_tests.test_http_database import ( TestHTTPDatabaseSimpleOperations, @@ -29,7 +35,53 @@ from leap.soledad.tests.u1db_tests.test_remote_sync_target import ( TestHTTPSyncTargetBasics, TestParsingSyncStream, ) +from leap.soledad.tests.u1db_tests.test_sync import ( + _make_local_db_and_target, + DatabaseSyncTargetTests, +) +from leap.soledad.tests.u1db_tests.test_https import ( + TestHttpSyncTargetHttpsSupport, + https_server_def, +) +from leap.soledad.tests.test_encrypted import ( + PUBLIC_KEY, + PRIVATE_KEY, + KEY_FINGERPRINT, +) + +try: + import simplejson as json +except ImportError: + import json # noqa + +#----------------------------------------------------------------------------- +# The EncryptedSyncTest is used with multiple inheritance to guarantee that we +# have a working Soledad instance in each test. +#----------------------------------------------------------------------------- + +class SoledadTest(unittest.TestCase): + + PREFIX = "/var/tmp" + GNUPG_HOME = "%s/gnupg" % PREFIX + DB1_FILE = "%s/db1.u1db" % PREFIX + DB2_FILE = "%s/db2.u1db" % PREFIX + EMAIL = 'leap@leap.se' + def setUp(self): + super(SoledadTest, self).setUp() + self._db1 = u1db.open(self.DB1_FILE, create=True, + document_factory=leap_backend.LeapDocument) + self._db2 = u1db.open(self.DB2_FILE, create=True, + document_factory=leap_backend.LeapDocument) + self._soledad = Soledad(self.EMAIL, gpghome=self.GNUPG_HOME) + self._soledad._gpg.import_keys(PUBLIC_KEY) + self._soledad._gpg.import_keys(PRIVATE_KEY) + + def tearDown(self): + super(SoledadTest, self).tearDown() + os.unlink(self.DB1_FILE) + os.unlink(self.DB2_FILE) + #rmtree(self.GNUPG_HOME) #----------------------------------------------------------------------------- @@ -73,21 +125,40 @@ def make_oauth_leap_database_for_test(test, replica_uid): return http_db +def make_document_for_test(test, doc_id, rev, content, has_conflicts=False): + return leap_backend.LeapDocument( + doc_id, rev, content, has_conflicts=has_conflicts) + + +def make_leap_document_for_test(test, doc_id, rev, content, has_conflicts=False): + return leap_backend.LeapDocument( + doc_id, rev, content, has_conflicts=has_conflicts, + soledad=test._soledad) + + +def make_leap_encrypted_document_for_test(test, doc_id, rev, encrypted_content, + has_conflicts=False): + return leap_backend.LeapDocument( + doc_id, rev, encrypted_json=encrypted_content, + has_conflicts=has_conflicts, + soledad=test._soledad) + + LEAP_SCENARIOS = [ ('http', {'make_database_for_test': make_leap_database_for_test, 'copy_database_for_test': copy_leap_database_for_test, - 'make_document_for_test': tests.make_document_for_test, + 'make_document_for_test': make_leap_document_for_test, 'make_app_with_state': make_http_app}), ] -class LeapTests(AllDatabaseTests): +class LeapTests(AllDatabaseTests, SoledadTest): scenarios = LEAP_SCENARIOS #----------------------------------------------------------------------------- -# The following tests come from `u1db.tests.test_leap_backend`. +# The following tests come from `u1db.tests.test_http_database`. #----------------------------------------------------------------------------- class TestLeapDatabaseSimpleOperations(TestHTTPDatabaseSimpleOperations): @@ -116,6 +187,11 @@ class TestLeapDatabaseSimpleOperations(TestHTTPDatabaseSimpleOperations): self.db._request = _request self.db._request_json = _request_json + def test_get_sync_target(self): + st = self.db.get_sync_target() + self.assertIsInstance(st, leap_backend.LeapSyncTarget) + self.assertEqual(st._url, self.db._url) + class TestLeapDatabaseCtrWithCreds(TestHTTPDatabaseCtrWithCreds): pass @@ -125,7 +201,7 @@ class TestLeapDatabaseIntegration(TestHTTPDatabaseIntegration): def test_non_existing_db(self): db = leap_backend.LeapDatabase(self.getURL('not-there')) - self.assertRaises(errors.DatabaseDoesNotExist, db.get_doc, 'doc1') + self.assertRaises(u1db.errors.DatabaseDoesNotExist, db.get_doc, 'doc1') def test__ensure(self): db = leap_backend.LeapDatabase(self.getURL('new')) @@ -136,7 +212,7 @@ class TestLeapDatabaseIntegration(TestHTTPDatabaseIntegration): self.request_state._create_database('db0') db = leap_backend.LeapDatabase(self.getURL('db0')) db._delete() - self.assertRaises(errors.DatabaseDoesNotExist, + self.assertRaises(u1db.errors.DatabaseDoesNotExist, self.request_state.check_database, 'db0') def test_open_database_existing(self): @@ -146,7 +222,7 @@ class TestLeapDatabaseIntegration(TestHTTPDatabaseIntegration): self.assertIs(None, db.get_doc('doc1')) def test_open_database_non_existing(self): - self.assertRaises(errors.DatabaseDoesNotExist, + self.assertRaises(u1db.errors.DatabaseDoesNotExist, leap_backend.LeapDatabase.open_database, self.getURL('not-there'), create=False) @@ -159,14 +235,14 @@ class TestLeapDatabaseIntegration(TestHTTPDatabaseIntegration): def test_delete_database_existing(self): self.request_state._create_database('db0') leap_backend.LeapDatabase.delete_database(self.getURL('db0')) - self.assertRaises(errors.DatabaseDoesNotExist, + self.assertRaises(u1db.errors.DatabaseDoesNotExist, self.request_state.check_database, 'db0') def test_doc_ids_needing_quoting(self): db0 = self.request_state._create_database('db0') db = leap_backend.LeapDatabase.open_database(self.getURL('db0'), create=False) - doc = Document('%fff', None, '{}') + doc = leap_backend.LeapDocument('%fff', None, '{}') db.put_doc(doc) self.assertGetDoc(db0, '%fff', doc.rev, '{}', False) self.assertGetDoc(db, '%fff', doc.rev, '{}', False) @@ -184,25 +260,20 @@ class TestLeapClientBase(TestHTTPClientBase): # The following tests come from `u1db.tests.test_document`. #----------------------------------------------------------------------------- -def make_document_for_test(test, doc_id, rev, content, has_conflicts=False): - return leap_backend.LeapDocument( - doc_id, rev, content, has_conflicts=has_conflicts) - - -class TestLeapDocument(TestDocument): +class TestLeapDocument(TestDocument, SoledadTest): scenarios = ([( - 'leap', {'make_document_for_test': make_document_for_test})]) + 'leap', {'make_document_for_test': make_leap_document_for_test})]) -class TestLeapPyDocument(TestPyDocument): +class TestLeapPyDocument(TestPyDocument, SoledadTest): scenarios = ([( - 'leap', {'make_document_for_test': make_document_for_test})]) + 'leap', {'make_document_for_test': make_leap_document_for_test})]) #----------------------------------------------------------------------------- -# The following tests come from `u1db.tests.test_sync_target`. +# The following tests come from `u1db.tests.test_remote_sync_target`. #----------------------------------------------------------------------------- class TestLeapSyncTargetBasics(TestHTTPSyncTargetBasics): @@ -219,28 +290,28 @@ class TestLeapParsingSyncStream(TestParsingSyncStream): def test_wrong_start(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "{}\r\n]", None) - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "\r\n{}\r\n]", None) - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "", None) def test_wrong_end(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "[\r\n{}", None) - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "[\r\n", None) def test_missing_comma(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, '[\r\n{}\r\n{"id": "i", "rev": "r", ' '"content": "c", "gen": 3}\r\n]', None) @@ -248,13 +319,13 @@ class TestLeapParsingSyncStream(TestParsingSyncStream): def test_no_entries(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "[\r\n]", None) def test_extra_comma(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, "[\r\n{},\r\n]", None) self.assertRaises(leap_backend.NoSoledadInstance, @@ -267,16 +338,16 @@ class TestLeapParsingSyncStream(TestParsingSyncStream): def test_error_in_stream(self): tgt = leap_backend.LeapSyncTarget("http://foo/foo") - self.assertRaises(errors.Unavailable, + self.assertRaises(u1db.errors.Unavailable, tgt._parse_sync_stream, '[\r\n{"new_generation": 0},' '\r\n{"error": "unavailable"}\r\n', None) - self.assertRaises(errors.Unavailable, + self.assertRaises(u1db.errors.Unavailable, tgt._parse_sync_stream, '[\r\n{"error": "unavailable"}\r\n', None) - self.assertRaises(errors.BrokenSyncStream, + self.assertRaises(u1db.errors.BrokenSyncStream, tgt._parse_sync_stream, '[\r\n{"error": "?"}\r\n', None) @@ -285,13 +356,6 @@ def leap_sync_target(test, path): return leap_backend.LeapSyncTarget(test.getURL(path)) -def make_oauth_http_app(state): - app = http_app.HTTPApp(state) - application = oauth_middleware.OAuthMiddleware(app, None, prefix='/~/') - application.get_oauth_data_store = lambda: tests.testingOAuthStore - return application - - def oauth_leap_sync_target(test, path): st = leap_sync_target(test, '~/' + path) st.set_oauth_credentials(tests.consumer1.key, tests.consumer1.secret, @@ -303,11 +367,33 @@ class TestRemoteSyncTargets(tests.TestCaseWithServer): scenarios = [ ('http', {'make_app_with_state': make_http_app, - 'make_document_for_test': tests.make_document_for_test, + 'make_document_for_test': make_leap_document_for_test, 'sync_target': leap_sync_target}), ('oauth_http', {'make_app_with_state': make_oauth_http_app, - 'make_document_for_test': tests.make_document_for_test, + 'make_document_for_test': make_leap_document_for_test, 'sync_target': oauth_leap_sync_target}), ] + +#----------------------------------------------------------------------------- +# The following tests come from `u1db.tests.test_https`. +#----------------------------------------------------------------------------- + +def oauth_https_sync_target(test, host, path): + _, port = test.server.server_address + st = leap_backend.LeapSyncTarget('https://%s:%d/~/%s' % (host, port, path)) + st.set_oauth_credentials(tests.consumer1.key, tests.consumer1.secret, + tests.token1.key, tests.token1.secret) + return st + +class TestLeapSyncTargetHttpsSupport(TestHttpSyncTargetHttpsSupport, SoledadTest): + + scenarios = [ + ('oauth_https', {'server_def': https_server_def, + 'make_app_with_state': make_oauth_http_app, + 'make_document_for_test': make_leap_document_for_test, + 'sync_target': oauth_https_sync_target + }), + ] + load_tests = tests.load_with_scenarios |