tests: tests for coverage endpoint

Author: sgratzl

integrations/server/test_covidcast_endpoints.py

@@ -84,7 +84,7 @@ def from_json(json: Dict[str, Any]) -> "CovidcastRow":
             sample_size=json["sample_size"],
             missing_value=json["missing_value"],
             missing_stderr=json["missing_stderr"],
-            missing_sample_size=json["missing_sample_size"]
+            missing_sample_size=json["missing_sample_size"],
         )
 
     @property
@@ -361,3 +361,26 @@ def test_meta(self):
             self.assertEqual(out[0]["source"], first.source)
             out = self._fetch("/meta", signal=f"{first.source}:X")
             self.assertEqual(len(out), 0)
+
+    def test_coverage(self):
+        """Request a signal the /meta endpoint."""
+
+        num_geos_per_date = [10, 20, 30, 40, 44]
+        dates = [20200401 + i for i in range(num_geos_per_date)]
+        rows = [CovidcastRow(time_value=dates[i], value=i, geo_value=str(geo_value)) for i, num_geo in enumerate(num_geos_per_date) for geo_value in range(num_geo)]
+        self._insert_rows(rows)
+
+        with self.subTest("default"):
+            out = self._fetch("/coverage")
+            self.assertEqual(len(out), len(num_geos_per_date))
+            self.assertEqual([o["time_value"] for o in out], dates)
+            self.assertEqual([o["count"] for o in out], num_geos_per_date)
+
+        with self.subTest("specify window"):
+            out = self._fetch("/coverage", window="20200401-20200402")
+            self.assertEqual(len(out), 2)
+            self.assertEqual([o["time_value"] for o in out], dates[:2])
+            self.assertEqual([o["count"] for o in out], num_geos_per_date[:2])
+        with self.subTest("invalid geo_type"):
+            out = self._fetch("/coverage", geo_type="state")
+            self.assertEqual(len(out), 0)