Adjusted enable/disable processor tests for new output from API

GoogleCloudPlatform · Jan 5, 2023 · 455bdac · 455bdac
1 parent d1e144d
commit 455bdac
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/documentai/snippets/disable_processor_sample_test.py b/documentai/snippets/disable_processor_sample_test.py
@@ -29,7 +29,7 @@ def test_disable_processor(capsys):
     )
     out, _ = capsys.readouterr()
 
-    assert "projects" in out or "DISABLED" in out
+    assert "projects" in out or "Processor" in out
 
     # Re-Enable Processor
     enable_processor_sample.enable_processor_sample(

diff --git a/documentai/snippets/enable_processor_sample_test.py b/documentai/snippets/enable_processor_sample_test.py
@@ -29,7 +29,7 @@ def test_enable_processor(capsys):
     )
     out, _ = capsys.readouterr()
 
-    assert "projects" in out or "ENABLED" in out
+    assert "projects" in out or "Processor" in out
 
     # Re-Disable Processor
     disable_processor_sample.disable_processor_sample(