add test for generating read key

buildtimetrend · Feb 11, 2015 · 800b143 · 800b143
1 parent 9353972
commit 800b143
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/buildtimetrend/test/keenio_tests.py b/buildtimetrend/test/keenio_tests.py
@@ -223,6 +223,16 @@ def test_keen_is_readable_envir_vars(self):
         os.environ["KEEN_READ_KEY"] = "4567abcd5678efgh"
         self.assertTrue(keen_is_readable())
 
+    def test_generate_read_key(self):
+        # should return None if master key is not set
+        self.assertEqual(None, keen_io_generate_read_key(None))
+        self.assertEqual(None, keen_io_generate_read_key("test/project"))
+
+        # set master_key
+        os.environ["KEEN_MASTER_KEY"] = "4567abcd5678efgh"
+        self.assertEqual(str, type(keen_io_generate_read_key(None)))
+        self.assertEqual(str, type(keen_io_generate_read_key("test/project")))
+
     def test_get_repo_filter(self):
         self.assertEqual(None, get_repo_filter())
         self.assertEqual(None, get_repo_filter(None))