Initial unit tests for CLI

Change-Id: Ie18bda7edd09cf95bc2c0066ddaf73e1c4c8a7fa
Signed-off-by: Paul Belanger <pabelanger@redhat.com>

grafana_dashboards/grafana.py

@@ -14,12 +14,15 @@
 
 import json
 import requests
-import urlparse
+try:
+    from urllib.parse import urljoin
+except ImportError:
+    from urlparse import urljoin
 
 
 class Grafana(object):
     def __init__(self, url, key):
-        self.url = urlparse.urljoin(url, 'api/dashboards/db')
+        self.url = urljoin(url, 'api/dashboards/db')
         self.session = requests.Session()
         self.session.headers.update({
             'Authorization': 'Bearer %s' % key,

tests/test_cmd.py

@@ -0,0 +1,63 @@
+# Copyright 2015 Red Hat, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import re
+import sys
+
+import fixtures
+import six
+from testtools import matchers
+from testtools import TestCase
+
+from grafana_dashboards import cmd
+
+
+class TestCaseCmd(TestCase):
+
+    def setUp(self):
+        super(TestCaseCmd, self).setUp()
+
+    def shell(self, argstr, exitcodes=(0,)):
+        orig = sys.stdout
+        orig_stderr = sys.stderr
+        try:
+            sys.stdout = six.StringIO()
+            sys.stderr = six.StringIO()
+            argv = ['grafana-dashboards']
+            argv += argstr.split()
+            self.useFixture(fixtures.MonkeyPatch('sys.argv', argv))
+            cmd.main()
+        except SystemExit:
+            exc_type, exc_value, exc_trackback = sys.exc_info()
+            self.assertIn(exc_value.code, exitcodes)
+        finally:
+            stdout = sys.stdout.getvalue()
+            sys.stdout.close()
+            sys.stdout = orig
+            stderr = sys.stderr.getvalue()
+            sys.stderr.close()
+            sys.stderr = orig_stderr
+        return (stdout, stderr)
+
+    def test_update_without_path(self):
+        required = [
+            '.*?^usage: grafana-dashboards update \[-h\] path',
+            '.*?^grafana-dashboards update: error: (too few arguments|the '
+            'following arguments are required: path)',
+        ]
+        stdout, stderr = self.shell('update', exitcodes=[2])
+        for r in required:
+            self.assertThat(
+                (stdout + stderr),
+                matchers.MatchesRegex(r, re.DOTALL | re.MULTILINE))