diff options
Diffstat (limited to 'bitbake/lib/hashserv/tests.py')
-rw-r--r-- | bitbake/lib/hashserv/tests.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/bitbake/lib/hashserv/tests.py b/bitbake/lib/hashserv/tests.py index 4c98a280a5..268b27006f 100644 --- a/bitbake/lib/hashserv/tests.py +++ b/bitbake/lib/hashserv/tests.py | |||
@@ -33,7 +33,7 @@ class HashEquivalenceTestSetup(object): | |||
33 | def start_server(self, dbpath=None, upstream=None, read_only=False, prefunc=server_prefunc): | 33 | def start_server(self, dbpath=None, upstream=None, read_only=False, prefunc=server_prefunc): |
34 | self.server_index += 1 | 34 | self.server_index += 1 |
35 | if dbpath is None: | 35 | if dbpath is None: |
36 | dbpath = os.path.join(self.temp_dir.name, "db%d.sqlite" % self.server_index) | 36 | dbpath = self.make_dbpath() |
37 | 37 | ||
38 | def cleanup_server(server): | 38 | def cleanup_server(server): |
39 | if server.process.exitcode is not None: | 39 | if server.process.exitcode is not None: |
@@ -53,6 +53,9 @@ class HashEquivalenceTestSetup(object): | |||
53 | 53 | ||
54 | return server | 54 | return server |
55 | 55 | ||
56 | def make_dbpath(self): | ||
57 | return os.path.join(self.temp_dir.name, "db%d.sqlite" % self.server_index) | ||
58 | |||
56 | def start_client(self, server_address): | 59 | def start_client(self, server_address): |
57 | def cleanup_client(client): | 60 | def cleanup_client(client): |
58 | client.close() | 61 | client.close() |
@@ -517,6 +520,20 @@ class TestHashEquivalenceWebsocketServer(HashEquivalenceTestSetup, HashEquivalen | |||
517 | return "ws://%s:0" % host | 520 | return "ws://%s:0" % host |
518 | 521 | ||
519 | 522 | ||
523 | class TestHashEquivalenceWebsocketsSQLAlchemyServer(TestHashEquivalenceWebsocketServer): | ||
524 | def setUp(self): | ||
525 | try: | ||
526 | import sqlalchemy | ||
527 | import aiosqlite | ||
528 | except ImportError as e: | ||
529 | self.skipTest(str(e)) | ||
530 | |||
531 | super().setUp() | ||
532 | |||
533 | def make_dbpath(self): | ||
534 | return "sqlite+aiosqlite:///%s" % os.path.join(self.temp_dir.name, "db%d.sqlite" % self.server_index) | ||
535 | |||
536 | |||
520 | class TestHashEquivalenceExternalServer(HashEquivalenceTestSetup, HashEquivalenceCommonTests, unittest.TestCase): | 537 | class TestHashEquivalenceExternalServer(HashEquivalenceTestSetup, HashEquivalenceCommonTests, unittest.TestCase): |
521 | def start_test_server(self): | 538 | def start_test_server(self): |
522 | if 'BB_TEST_HASHSERV' not in os.environ: | 539 | if 'BB_TEST_HASHSERV' not in os.environ: |