From 455bdaca517da2152f1730406ebec289075c5aa3 Mon Sep 17 00:00:00 2001 From: Holt Skinner Date: Thu, 5 Jan 2023 13:26:17 -0600 Subject: [PATCH] Adjusted enable/disable processor tests for new output from API --- documentai/snippets/disable_processor_sample_test.py | 2 +- documentai/snippets/enable_processor_sample_test.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/documentai/snippets/disable_processor_sample_test.py b/documentai/snippets/disable_processor_sample_test.py index 6cd98479d164..8f74e3976396 100644 --- 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 index 0374a6423561..764550cb6c69 100644 --- 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(