瀏覽代碼

HUE-6083 [metadata] Update test to look for content in json

Romain Rigaux 8 年之前
父節點
當前提交
888103362a
共有 1 個文件被更改,包括 6 次插入4 次删除
  1. 6 4
      apps/metastore/src/metastore/tests.py

+ 6 - 4
apps/metastore/src/metastore/tests.py

@@ -224,15 +224,17 @@ class TestMetastoreWithHadoop(BeeswaxSampleProvider):
     wait_for_query_to_finish(self.client, resp, max=30.0)
 
     # Assert partition exists
-    response = self.client.get("/metastore/table/%s/test_partitions/partitions" % self.db_name)
-    assert_true("baz_drop" in response.content)
+    response = self.client.get("/metastore/table/%s/test_partitions/partitions" % self.db_name, {'format': 'json'})
+    data = json.loads(response.content)
+    assert_true("baz_drop" in [part['columns'][0] for part in data['partition_values_json']], data)
 
     # Drop partition
     self.client.post("/metastore/table/%s/test_partitions/partitions/drop" % self.db_name, {'partition_selection': [partition_spec]}, follow=True)
     query = QueryHistory.objects.latest('id')
     assert_equal_mod_whitespace("ALTER TABLE `%s`.`test_partitions` DROP IF EXISTS PARTITION (%s) PURGE" % (self.db_name, partition_spec), query.query)
-    response = self.client.get("/metastore/table/%s/test_partitions/partitions" % self.db_name)
-    assert_false("baz_drop" in response.content)
+    response = self.client.get("/metastore/table/%s/test_partitions/partitions" % self.db_name, {'format': 'json'})
+    data = json.loads(response.content)
+    assert_false("baz_drop" in [part['columns'][0] for part in data['partition_values_json']], data)
 
   def test_drop_multi_tables(self):
     hql = """