commit 85d38402bb73053b863ce19c2528608d841908cc
author Michael Mortensen <mmortensen@google.com> Thu Dec 12 09:50:29 2019 -0700
committer Commit Bot <commit-bot@chromium.org> Thu Dec 12 20:13:32 2019 +0000
tree d3fbb8f2731c791cff790d50c2625ca41f6cd42a
parent 150164937c6b936d40d538e3ca4c3844fd4fb397

BuildAPI: Use faux.empty_completed_unsuccessfully_error for tests.

BUG=chromium:1029872
TEST=run_tests

Change-Id: Id4be0f617e1a9ed1df545c2c7c2970dfbfd796cc
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1964694
Reviewed-by: Alex Klein <saklein@chromium.org>
Commit-Queue: Michael Mortensen <mmortensen@google.com>
Tested-by: Michael Mortensen <mmortensen@google.com>

api/controller/payload_unittest.py

diff --git a/api/controller/payload_unittest.py b/api/controller/payload_unittest.py
index 16cfe47..d161b99 100644
--- a/api/controller/payload_unittest.py
+++ b/api/controller/payload_unittest.py
@@ -55,3 +55,21 @@
     self.PatchObject(paygen_payload_lib, 'PaygenPayload')
     res = payload.GeneratePayload(self.req, self.result, self.api_config)
     self.assertEqual(res, controller.RETURN_CODE_SUCCESS)
+
+  def testMockError(self):
+    """Test mock error call does not execute any logic, returns error."""
+    patch = self.PatchObject(paygen_payload_lib, 'PaygenPayload')
+
+    res = payload.GeneratePayload(self.req, self.result,
+                                  self.mock_error_config)
+    patch.assert_not_called()
+    self.assertEqual(controller.RETURN_CODE_COMPLETED_UNSUCCESSFULLY, res)
+
+  def testMockCall(self):
+    """Test mock call does not execute any logic, returns success."""
+    patch = self.PatchObject(paygen_payload_lib, 'PaygenPayload')
+
+    res = payload.GeneratePayload(self.req, self.result,
+                                  self.mock_call_config)
+    patch.assert_not_called()
+    self.assertEqual(controller.RETURN_CODE_SUCCESS, res)