Kubernetes container handler

src/commissaire/containermgr/kubernetes/__init__.py

@@ -93,6 +93,22 @@ def check_config(cls, config):
                     'Server URL scheme must be "https" when using client '
                     'side certificates (got "{}")'.format(url.scheme))
 
+    def _fix_part(self, part):
+        """
+        Fixes part if it doesn't start with a slash.
+
+        :param part: The URI part. EG: /nodes
+        :type part: str
+        :returns: The part in the proper format.
+        :rtype: str
+        """
+        if not part.startswith('/'):
+            self.logger.debug(
+                'Part given without starting slash. Adding...')
+            part = '/{}'.format(part)
+
+        return part
+
     def _get(self, part, *args, **kwargs):
         """
         Get information from the Kubernetes apiserver.
@@ -105,12 +121,7 @@ def _get(self, part, *args, **kwargs):
         :type kwargs: dict
         :returns: requests.Response
         """
-        # Fix part if it doesn't start with a slash
-        if not part.startswith('/'):
-            self.logger.debug(
-                'Part given without starting slash. Adding...')
-            part = '/{}'.format(part)
-
+        part = self._fix_part(part)
         self.logger.debug('Executing GET for {}'.format(part))
         resp = self.con.get(
             '{}{}'.format(self.base_uri, part), *args, **kwargs)

test/test_containermgr_kubernetes.py

@@ -98,3 +98,19 @@ def test_get_host_status(self):
             self.instance.con.get.assert_called_with(
                 CONTAINER_MGR_CONFIG['server_url'] + 'api/v1/nodes/test')
 
+    def test__fix_part_with_valid_part(self):
+        """
+        Verify that when a valid part is given it is returned without modification.
+        """
+        expected = '/test'
+        self.assertEquals(
+            expected,
+            self.instance._fix_part(expected))
+
+    def test__fix_part_with_invalid_part(self):
+        """
+        Verify that when an invalid part it is fixed.
+        """
+        self.assertEquals(
+            '/test',
+            self.instance._fix_part('test'))