chore: Update Test file.

docqai · Sep 10, 2023 · c931352 · c931352
1 parent b29edd7
commit c931352
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/docq/support/store_test.py b/tests/docq/support/store_test.py
@@ -4,7 +4,7 @@
 
 import pytest
 from docq.config import FeatureType, SpaceType
-from docq.domain import SpaceKey
+from docq.domain import FeatureKey, SpaceKey
 from docq.support.store import (
     get_history_table_name,
     get_index_dir,
@@ -62,7 +62,7 @@ def test_get_index_dir(self: Self, type_: SpaceType, id_: int, expected: str):
         ],
     )
     def test_get_sqlite_usage_file(self: Self, id_: int, expected: str):
-        assert get_sqlite_usage_file(id_) == expected
+        assert get_sqlite_usage_file(FeatureKey(FeatureType.ASK_PERSONAL, id_)) == expected
 
     def test_get_sqlite_system_file(self: Self):
         assert get_sqlite_system_file() == DATA_DIR + "/sqlite/SHARED/system.db"