Эх сурвалжийг харах

HUE-3065 [desktop] Fix test_redact_statements

Jenny Kim 9 жил өмнө
parent
commit
aa94ca2

+ 4 - 0
desktop/core/src/desktop/tests.py

@@ -967,6 +967,7 @@ class TestDocument(object):
     try:
       self.document2.type = 'notebook'
       self.document2.update_data({'snippets': snippets})
+      self.document2.search = sensitive_query
       self.document2.save()
       saved_snippets = self.document2.data_dict['snippets']
 
@@ -979,6 +980,9 @@ class TestDocument(object):
       assert_equal(redacted_query, saved_snippets[1]['statement_raw'])
       assert_equal(True, saved_snippets[1]['is_redacted'])
 
+      document = Document2.objects.get(pk=self.document2.pk)
+      assert_equal(redacted_query, document.search)
+
       # Make sure unredacted queries are not redacted.
       assert_equal(nonsensitive_query, saved_snippets[2]['statement'])
       assert_equal(nonsensitive_query, saved_snippets[2]['statement_raw'])