Browse Source

HUE-3287 [core] Django 1.11 upgrade
- Fixing Beeswax test

Prakash Ranade 7 năm trước cách đây
mục cha
commit
1a64e6a62d
1 tập tin đã thay đổi với 5 bổ sung5 xóa
  1. 5 5
      apps/beeswax/src/beeswax/tests.py

+ 5 - 5
apps/beeswax/src/beeswax/tests.py

@@ -198,7 +198,7 @@ for x in sys.stdin:
       SELECT MIN(foo), MAX(foo), SUM(foo) FROM test;
     """
     wait_for_query_to_finish(self.client, _make_query(self.client, QUERY, local=False, database=self.db_name), max=180.0)
-    self._verify_query_state(beeswax.models.QueryHistory.STATE.available)
+    self._verify_query_state(beeswax.models.QueryHistory.STATE.available.value)
 
     # Make sure expired query states are lazily updated.
     resp = self.client.get('/beeswax/query_history')
@@ -209,7 +209,7 @@ for x in sys.stdin:
 
     resp = self.client.get('/beeswax/query_history')
     history = resp.context[0]['page'].object_list[0]
-    assert_equal(history.last_state, beeswax.models.QueryHistory.STATE.expired)
+    assert_equal(history.last_state, beeswax.models.QueryHistory.STATE.expired.value)
 
 
   def test_basic_flow(self):
@@ -229,14 +229,14 @@ for x in sys.stdin:
 
     # Check that we report this query as "running" (this query should take a little while).
     if not is_hive_on_spark():
-      self._verify_query_state(beeswax.models.QueryHistory.STATE.running, beeswax.models.QueryHistory.STATE.available)
+      self._verify_query_state(beeswax.models.QueryHistory.STATE.running.value, beeswax.models.QueryHistory.STATE.available.value)
 
     response = wait_for_query_to_finish(self.client, response, max=180.0)
     content = fetch_query_result_data(self.client, response)
 
     assert_equal([0, 255, 32640], content["results"][0], content["results"][0])
     assert_equal(['INT_TYPE', 'INT_TYPE', 'BIGINT_TYPE'], [col['type'] for col in content["columns"]])
-    self._verify_query_state(beeswax.models.QueryHistory.STATE.available)
+    self._verify_query_state(beeswax.models.QueryHistory.STATE.available.value)
 
     # Query multi-page request
     QUERY = """
@@ -413,7 +413,7 @@ for x in sys.stdin:
     assert_true("ParseException line" in json.loads(resp.content)["message"])
 
     # Watch page will fail as operationHandle=None
-    self._verify_query_state(beeswax.models.QueryHistory.STATE.failed)
+    self._verify_query_state(beeswax.models.QueryHistory.STATE.failed.value)
 
 
   def test_sync_query_exec(self):