commit 6431d3a8dd5f5ce7e7129a44d7a9843d605fd259
author Alex Klein <saklein@chromium.org> Thu Aug 29 10:16:08 2019 -0600
committer Commit Bot <commit-bot@chromium.org> Thu Aug 29 21:05:06 2019 +0000
tree 638084ffc77792bc49b9bf5ef7d1a4cf461f37f5
parent 6ad091737d1bf4f7ddda4452389f390b7997447f

api/router.py: Cleanup unused methods, add tests.

BUG=None
TEST=run_tests

Change-Id: I387df293b33eb334d6248e5e8b8849baa82af0a6
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/1776346
Tested-by: Alex Klein <saklein@chromium.org>
Auto-Submit: Alex Klein <saklein@chromium.org>
Reviewed-by: David Burger <dburger@chromium.org>
Commit-Queue: Alex Klein <saklein@chromium.org>

api/router_unittest.py

diff --git a/api/router_unittest.py b/api/router_unittest.py
index 95a2ae0..742ed69 100644
--- a/api/router_unittest.py
+++ b/api/router_unittest.py
@@ -211,3 +211,21 @@
     self.assertCommandContains(['build_api', service_method], enter_chroot=True)
     # It should be writing the empty message out.
     self.assertFileContents(self.output_file, '{}')
+
+  def testInvalidService(self):
+    """Test invalid service call."""
+    service = 'chromite.api.DoesNotExist'
+    method = 'OutsideServiceInsideMethod'
+
+    with self.assertRaises(router.UnknownServiceError):
+      self.router.Route(service, method, self.input_file, self.output_file,
+                        self.api_config)
+
+  def testInvalidMethod(self):
+    """Test invalid method call."""
+    service = 'chromite.api.OutsideChrootApiService'
+    method = 'DoesNotExist'
+
+    with self.assertRaises(router.UnknownMethodError):
+      self.router.Route(service, method, self.input_file, self.output_file,
+                        self.api_config)