|
6 | 6 |
|
7 | 7 | bin_path = os.path.join(os.getcwd(), "bin") |
8 | 8 | sys.path.insert(0, bin_path) |
9 | | -import database |
| 9 | +import db_upgrade |
10 | 10 |
|
11 | 11 | from api import config |
12 | 12 |
|
13 | | -CDV = database.CURRENT_DATABASE_VERSION |
| 13 | +CDV = db_upgrade.CURRENT_DATABASE_VERSION |
14 | 14 |
|
15 | 15 | def test_all_upgrade_scripts_exist(): |
16 | 16 | for i in range(1, CDV): |
17 | 17 | script_name = 'upgrade_to_{}'.format(i) |
18 | | - assert hasattr(database, script_name) |
| 18 | + assert hasattr(db_upgrade, script_name) |
19 | 19 |
|
20 | 20 | def test_CDV_was_bumped(): |
21 | 21 | script_name = 'upgrade_to_{}'.format(CDV+1) |
22 | | - assert hasattr(database, script_name) is False |
| 22 | + assert hasattr(db_upgrade, script_name) is False |
23 | 23 |
|
24 | 24 |
|
25 | 25 | @patch('api.config.get_version', Mock(return_value={'database': 5})) |
26 | 26 | def test_get_db_version_from_config(): |
27 | | - assert database.get_db_version() == 5 |
| 27 | + assert db_upgrade.get_db_version() == 5 |
28 | 28 |
|
29 | 29 |
|
30 | 30 | @pytest.fixture(scope='function') |
31 | | -def database_mock_setup(): |
| 31 | +def db_upgrade_mock_setup(): |
32 | 32 | setattr(config.db.singletons, 'update_one', Mock()) |
33 | 33 | for i in range(1, CDV): |
34 | 34 | script_name = 'upgrade_to_{}'.format(i) |
35 | | - setattr(database, script_name, Mock()) |
| 35 | + setattr(db_upgrade, script_name, Mock()) |
36 | 36 |
|
37 | | -@patch('database.get_db_version', Mock(return_value=0)) |
38 | | -def test_all_upgrade_scripts_ran(database_mock_setup): |
| 37 | +@patch('db_upgrade.get_db_version', Mock(return_value=0)) |
| 38 | +def test_all_upgrade_scripts_ran(db_upgrade_mock_setup): |
39 | 39 | with pytest.raises(SystemExit): |
40 | | - database.upgrade_schema() |
| 40 | + db_upgrade.upgrade_schema() |
41 | 41 | for i in range(1, CDV): |
42 | 42 | script_name = 'upgrade_to_{}'.format(i) |
43 | | - assert getattr(database, script_name).called |
| 43 | + assert getattr(db_upgrade, script_name).called |
44 | 44 |
|
45 | | -@patch('database.get_db_version', Mock(return_value=CDV-4)) |
46 | | -def test_necessary_upgrade_scripts_ran(database_mock_setup): |
| 45 | +@patch('db_upgrade.get_db_version', Mock(return_value=CDV-4)) |
| 46 | +def test_necessary_upgrade_scripts_ran(db_upgrade_mock_setup): |
47 | 47 | with pytest.raises(SystemExit): |
48 | | - database.upgrade_schema() |
| 48 | + db_upgrade.upgrade_schema() |
49 | 49 | # Assert the necessary scripts were called |
50 | 50 | for i in range(CDV-3, CDV): |
51 | 51 | script_name = 'upgrade_to_{}'.format(i) |
52 | | - assert getattr(database, script_name).called |
| 52 | + assert getattr(db_upgrade, script_name).called |
53 | 53 | # But not the scripts before it |
54 | 54 | for i in range(1, CDV-4): |
55 | 55 | script_name = 'upgrade_to_{}'.format(i) |
56 | | - assert getattr(database, script_name).called is False |
| 56 | + assert getattr(db_upgrade, script_name).called is False |
0 commit comments