Coverage for mlos_bench/mlos_bench/tests/storage/conftest.py: 100%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5"""Export test fixtures for mlos_bench storage.""" 

6 

7import mlos_bench.tests.docker_fixtures_util as docker_fixtures_util 

8import mlos_bench.tests.storage.sql.fixtures as sql_storage_fixtures 

9 

10# NOTE: For future storage implementation additions, we can refactor this to use 

11# lazy_fixture and parameterize the tests across fixtures but keep the test code the 

12# same. 

13 

14# Expose some of those as local names so they can be picked up as fixtures by pytest. 

15docker_compose_file = sql_storage_fixtures.docker_compose_file 

16docker_compose_project_name = sql_storage_fixtures.docker_compose_project_name 

17 

18docker_setup = docker_fixtures_util.docker_setup 

19docker_services_lock = docker_fixtures_util.docker_services_lock 

20docker_setup_teardown_lock = docker_fixtures_util.docker_setup_teardown_lock 

21locked_docker_services = docker_fixtures_util.locked_docker_services 

22 

23mysql_storage_info = sql_storage_fixtures.mysql_storage_info 

24mysql_storage = sql_storage_fixtures.mysql_storage 

25postgres_storage_info = sql_storage_fixtures.postgres_storage_info 

26postgres_storage = sql_storage_fixtures.postgres_storage 

27sqlite_storage = sql_storage_fixtures.sqlite_storage 

28storage = sql_storage_fixtures.storage 

29exp_storage = sql_storage_fixtures.exp_storage 

30exp_no_tunables_storage = sql_storage_fixtures.exp_no_tunables_storage 

31mixed_numerics_exp_storage = sql_storage_fixtures.mixed_numerics_exp_storage 

32exp_data = sql_storage_fixtures.exp_data 

33exp_no_tunables_data = sql_storage_fixtures.exp_no_tunables_data 

34mixed_numerics_exp_data = sql_storage_fixtures.mixed_numerics_exp_data