Fix mistake in test code

apache · Apr 30, 2020 · c85fb20 · c85fb20
1 parent 26c9d64
commit c85fb20
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/www/test_views.py b/tests/www/test_views.py
@@ -521,7 +521,7 @@ def test_dag_stats(self):
     def test_task_stats(self):
         resp = self.client.post('task_stats', follow_redirects=True)
         self.assertEqual(resp.status_code, 200)
-        self.assertEqual(set(resp.json().items()[0][1][0].keys()),
+        self.assertEqual(set(list(resp.json.items())[0][1][0].keys()),
                          {'state', 'count', 'color'})
 
     def test_task_stats_only_noncompleted(self):
@@ -1618,7 +1618,7 @@ def test_dag_stats_success(self):
         self.login()
         resp = self.client.post('dag_stats', follow_redirects=True)
         self.check_content_in_response('example_bash_operator', resp)
-        self.assertEqual(set(resp.json().items()[0][1][0].keys()),
+        self.assertEqual(set(list(resp.json.items())[0][1][0].keys()),
                          {'state', 'count', 'color'})
 
     def test_dag_stats_failure(self):