Add tests for start_date and end_date Report filtering

Alex Schworer · Alex Schworer · commit d8856a6f473f · 2013-05-22T14:17:48.000-07:00
diff --git a/test/fixtures/report_all_date.json b/test/fixtures/report_all_date.json
@@ -0,0 +1,38 @@
+{
+    "statistics": {
+        "vod": [
+            {
+                "encoded_minutes": 5,
+                "billable_minutes": 0,
+                "grouping": null,
+                "collected_on": "2013-05-13"
+            }
+        ],
+        "live": [
+            {
+                "total_billable_hours": 2,
+                "stream_hours": 1,
+                "billable_stream_hours": 0,
+                "encoded_hours": 2,
+                "grouping": null,
+                "collected_on": "2013-05-13",
+                "billable_encoded_hours": 0,
+                "total_hours": 2
+            }
+        ]
+    },
+    "total": {
+        "vod": {
+            "encoded_minutes": 5,
+            "billable_minutes": 0
+        },
+        "live": {
+            "total_billable_hours": 2,
+            "stream_hours": 1,
+            "billable_stream_hours": 0,
+            "encoded_hours": 2,
+            "billable_encoded_hours": 0,
+            "total_hours": 2
+        }
+    }
+}
diff --git a/test/test_reports.py b/test/test_reports.py
@@ -4,6 +4,8 @@
 from test_util import TEST_API_KEY, load_response
 from zencoder import Zencoder
 
+import datetime
+
 class TestReports(unittest.TestCase):
     def setUp(self):
         self.zen = Zencoder(api_key=TEST_API_KEY)
@@ -33,7 +35,7 @@ def test_reports_live(self, get):
     def test_reports_all(self, get):
         get.return_value = load_response(200, 'fixtures/report_all.json')
 
-        resp = self.zen.report.live()
+        resp = self.zen.report.all()
 
         self.assertEquals(resp.code, 200)
 
@@ -43,6 +45,26 @@ def test_reports_all(self, get):
         self.assertEquals(resp.body['total']['vod']['billable_minutes'], 8)
         self.assertEquals(resp.body['statistics']['live']['length'], 2)
 
+    @patch("requests.Session.get")
+    def test_reports_all_date_filter(self, get):
+        get.return_value = load_response(200, 'fixtures/report_all_date.json')
+
+        start = datetime.date(2013, 5, 13)
+        end = datetime.date(2013, 5, 13)
+        resp = self.zen.report.all(start_date=start, end_date=end)
+
+        self.assertEquals(resp.code, 200)
+
+        self.assertEquals(resp.body['statistics']['vod'][0]['encoded_minutes'], 5)
+        self.assertEquals(resp.body['statistics']['vod'][0]['billable_minutes'], 0)
+        self.assertEquals(resp.body['statistics']['live'][0]['stream_hours'], 1)
+        self.assertEquals(resp.body['statistics']['live'][0]['total_hours'], 2)
+
+        self.assertEquals(resp.body['total']['vod']['encoded_minutes'], 5)
+        self.assertEquals(resp.body['total']['vod']['billable_minutes'], 0)
+        self.assertEquals(resp.body['total']['live']['stream_hours'], 1)
+        self.assertEquals(resp.body['total']['live']['total_hours'], 2)
+
 if __name__ == "__main__":
     unittest.main()
 
