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: |
