mirror of
https://github.com/quay/quay.git
synced 2026-01-26 06:21:37 +03:00
22290c9812
* Updated raven dependency to be compatible with python 3.12 --------- Co-authored-by: shudeshp <shudeshp@redhat.com>
43 lines
1.6 KiB
Python
43 lines
1.6 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from data.model.organization import create_organization
|
|
from data.model.quota import update_namespacesize
|
|
from data.model.user import create_robot, get_user
|
|
from test.fixtures import *
|
|
from workers.quotatotalworker import QuotaTotalWorker
|
|
|
|
ORG_NAME = "orgdoesnotexist"
|
|
|
|
|
|
def test_namespace_discovery(initialized_db):
|
|
user = get_user("devtable")
|
|
orgdoesnotexist = create_organization("orgdoesnotexist", "orgdoesnotexist@devtable.com", user)
|
|
create_robot("testrobot", orgdoesnotexist) # Robot accounts should not have total calculated
|
|
orgbackfillreset = create_organization(
|
|
"orgbackfillreset", "orgbackfillreset@devtable.com", user
|
|
)
|
|
update_namespacesize(
|
|
orgbackfillreset.id,
|
|
{"size_bytes": 0, "backfill_start_ms": None, "backfill_complete": False},
|
|
)
|
|
orgalreadycounted = create_organization(
|
|
"orgalreadycounted", "orgalreadycounted@devtable.com", user
|
|
)
|
|
update_namespacesize(
|
|
orgalreadycounted.id,
|
|
{"size_bytes": 0, "backfill_start_ms": 0, "backfill_complete": True},
|
|
)
|
|
|
|
expected_calls = [orgdoesnotexist.id, orgbackfillreset.id]
|
|
with patch("workers.quotatotalworker.run_backfill", MagicMock()) as mock_run_backfill:
|
|
|
|
def assert_mock_run_backfill(namespace_id):
|
|
assert namespace_id != orgalreadycounted.id
|
|
if namespace_id in expected_calls:
|
|
expected_calls.remove(namespace_id)
|
|
|
|
mock_run_backfill.side_effect = assert_mock_run_backfill
|
|
worker = QuotaTotalWorker()
|
|
worker.backfill()
|
|
assert len(expected_calls) == 0
|