quota: fixing registry total worker failing to start (PROJQUAY-6010) (q…

…uay#2217)

Fixes import in the registry total worker.

workers/quotaregistrysizeworker.py

@@ -2,7 +2,8 @@
 import time
 
 import features
-from app import app, registry_model
+from app import app
+from data.model.quota import calculate_registry_size
 from util.locking import GlobalLock
 from util.log import logfile_path
 from workers.gunicorn_worker import GunicornWorker
@@ -18,7 +19,7 @@ def __init__(self):
         self.add_operation(self._calculate_registry_size, POLL_PERIOD)
 
     def _calculate_registry_size(self):
-        registry_model.quota.calculate_registry_size()
+        calculate_registry_size()
 
 
 def create_gunicorn_worker():

workers/test/test_quotaregistrysizeworker.py

@@ -0,0 +1,13 @@
+from test.fixtures import *
+from unittest.mock import MagicMock, patch
+
+from workers.quotaregistrysizeworker import QuotaRegistrySizeWorker
+
+
+def test_registrysizeworker(initialized_db):
+    with patch(
+        "workers.quotaregistrysizeworker.calculate_registry_size", MagicMock()
+    ) as mock_calculate_registry_size:
+        worker = QuotaRegistrySizeWorker()
+        worker._calculate_registry_size()
+        assert mock_calculate_registry_size.called