Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 18 additions & 10 deletions tests/mock_vws/test_flask_app_usage.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,10 @@
from requests_mock_flask import add_flask_app_to_mock
from vws import VWS, CloudRecoService

from mock_vws._flask_server.target_manager import TARGET_MANAGER_FLASK_APP
from mock_vws._flask_server.target_manager import (
TARGET_MANAGER,
TARGET_MANAGER_FLASK_APP,
)
from mock_vws._flask_server.vwq import CLOUDRECO_FLASK_APP
from mock_vws._flask_server.vws import VWS_FLASK_APP
from mock_vws.database import CloudDatabase, VuMarkDatabase
Expand Down Expand Up @@ -57,6 +60,11 @@ def _(monkeypatch: pytest.MonkeyPatch) -> Iterator[None]:

yield

for cloud_database in TARGET_MANAGER.cloud_databases:
TARGET_MANAGER.remove_cloud_database(cloud_database=cloud_database)
for vumark_database in TARGET_MANAGER.vumark_databases:
TARGET_MANAGER.remove_vumark_database(vumark_database=vumark_database)


class TestProcessingTime:
"""Tests for the time taken to process targets in the mock."""
Expand Down Expand Up @@ -176,26 +184,26 @@ def test_duplicate_vumark_keys() -> None:
keys, including VuMark databases.
"""
database = VuMarkDatabase(
server_access_key="v1",
server_secret_key="v2",
database_name="v3",
server_access_key="1",
server_secret_key="2",
database_name="3",
)

bad_server_access_key_db = VuMarkDatabase(server_access_key="v1")
bad_server_secret_key_db = VuMarkDatabase(server_secret_key="v2")
bad_database_name_db = VuMarkDatabase(database_name="v3")
bad_server_access_key_db = VuMarkDatabase(server_access_key="1")
bad_server_secret_key_db = VuMarkDatabase(server_secret_key="2")
bad_database_name_db = VuMarkDatabase(database_name="3")

server_access_key_conflict_error = (
"All server access keys must be unique. "
'There is already a database with the server access key "v1".'
'There is already a database with the server access key "1".'
)
server_secret_key_conflict_error = (
"All server secret keys must be unique. "
'There is already a database with the server secret key "v2".'
'There is already a database with the server secret key "2".'
)
database_name_conflict_error = (
"All names must be unique. "
'There is already a database with the name "v3".'
'There is already a database with the name "3".'
)

databases_url = _EXAMPLE_URL_FOR_TARGET_MANAGER + "/vumark_databases"
Expand Down