fix tests

Author: Andrii Soldatenko

src/sentry/dynamic_sampling/prioritise_projects.py

@@ -23,7 +23,7 @@ def fetch_projects_with_total_volumes() -> Mapping[int, Sequence[int]]:
     while (time.time() - start_time) < MAX_SECONDS:
         query = (
             Query(
-                match=Entity(EntityKey.OrgMetricsCounters.value),
+                match=Entity(EntityKey.GenericMetricsCounters.value),
                 select=[
                     Column("org_id"),
                     Column("project_id"),
@@ -47,10 +47,12 @@ def fetch_projects_with_total_volumes() -> Mapping[int, Sequence[int]]:
             .set_limit(CHUNK_SIZE + 1)
             .set_offset(offset)
         )
-        request = Request(dataset=Dataset.Metrics.value, app_id="dynamic_sampling", query=query)
+        request = Request(
+            dataset=Dataset.PerformanceMetrics.value, app_id="dynamic_sampling", query=query
+        )
         data = raw_snql_query(
             request,
-            referrer="dynamic_sampling.fetch_projects_with_total_volumes",
+            referrer="dynamic_sampling.fetch_projects_with_count_per_root_total_volumes",
         )["data"]
         count = len(data)
         more_results = count > CHUNK_SIZE

src/sentry/dynamic_sampling/tasks.py

@@ -34,4 +34,4 @@ def process_projects_sample_rates(org_id, project_ids) -> None:
 
     # Get adjusted sample rate via adjustment model
     #
-    # prioritize_projects.delay(org_id, project_ids)
+    # prioritise_projects.delay(org_id, project_ids)

src/sentry/snuba/dataset.py

@@ -28,4 +28,5 @@ class EntityKey(Enum):
     MetricsDistributions = "metrics_distributions"
     GenericMetricsDistributions = "generic_metrics_distributions"
     GenericMetricsSets = "generic_metrics_sets"
+    GenericMetricsCounters = "generic_metrics_counters"
     IssuePlatform = "search_issues"

tests/sentry/dynamic_sampling/test_prioritise_projects.py

@@ -1,6 +1,11 @@
+import pytest
+
 from sentry.dynamic_sampling.prioritise_projects import fetch_projects_with_total_volumes
+from sentry.testutils.factories import Factories
 
 
+@pytest.mark.django_db
 def test_prioritize_projects():
-    results = fetch_projects_with_total_volumes()
-    1 == 1
+    organization = Factories.create_organization(name="test-org")
+    Factories.create_project(organization=organization)
+    fetch_projects_with_total_volumes()