add tests to both new utilities

Signed-off-by: Alfredo Deza <alfredo.deza@inktank.com>

diff --git a/ceph_deploy/tests/unit/test_mon.py b/ceph_deploy/tests/unit/test_mon.py
index 5ba4136a9be9ffd2e189027424e5dbe8f62d7a48..871d7e04a8b0b5a17babdae70873ea755a08c49b 100644 (file)
--- a/ceph_deploy/tests/unit/test_mon.py
+++ b/ceph_deploy/tests/unit/test_mon.py
@@ -165,3 +165,72 @@ class TestIsRunning(object):
         with patch('ceph_deploy.mon.subprocess.Popen', self.fake_popen):
             result = mon.is_running(['ceph', 'status'])
         assert result is False
+
+
+class TestKeyringParser(object):
+
+    def test_line_ends_with_newline_char(self, tmpdir):
+        keyring = tmpdir.join('foo.mon.keyring')
+        keyring.write('[section]\nasdfasdf\nkey = value')
+        result = mon.keyring_parser(keyring.strpath)
+
+        assert result == ['section']
+
+    def test_line_does_not_end_with_newline_char(self, tmpdir):
+        keyring = tmpdir.join('foo.mon.keyring')
+        keyring.write('[section]asdfasdf\nkey = value')
+        result = mon.keyring_parser(keyring.strpath)
+
+        assert result == []
+
+
+class TestConcatenateKeyrings(object):
+
+    def setup(self):
+        self.args = Mock()
+
+    def make_keyring(self, tmpdir, name, contents):
+        keyring = tmpdir.join(name)
+        keyring.write(contents)
+        return keyring
+
+    def test_multiple_keyrings_work(self, tmpdir):
+        self.make_keyring(tmpdir, 'foo.keyring', '[mon.1]\nkey = value\n')
+        self.make_keyring(tmpdir, 'bar.keyring', '[mon.2]\nkey = value\n')
+        self.make_keyring(tmpdir, 'fez.keyring', '[mon.3]\nkey = value\n')
+        self.args.keyrings = tmpdir.strpath
+        expected = '\n'.join([
+            '[mon.2]',
+            'key = value',
+            '[mon.3]',
+            'key = value',
+            '[mon.1]',
+            'key = value',
+            '',
+        ])
+
+        assert mon.concatenate_keyrings(self.args) == expected
+
+    def test_skips_duplicate_content(self, tmpdir):
+        self.make_keyring(tmpdir, 'foo.keyring', '[mon.1]\nkey = value\n')
+        self.make_keyring(tmpdir, 'bar.keyring', '[mon.2]\nkey = value\n')
+        self.make_keyring(tmpdir, 'fez.keyring', '[mon.3]\nkey = value\n')
+        self.make_keyring(tmpdir, 'dupe.keyring', '[mon.3]\nkey = value\n')
+        self.args.keyrings = tmpdir.strpath
+        expected = '\n'.join([
+            '[mon.2]',
+            'key = value',
+            '[mon.3]',
+            'key = value',
+            '[mon.1]',
+            'key = value',
+            '',
+        ])
+
+        assert mon.concatenate_keyrings(self.args) == expected
+
+    def test_errors_when_no_keyrings(self, tmpdir):
+        self.args.keyrings = tmpdir.strpath
+
+        with py.test.raises(RuntimeError):
+            mon.concatenate_keyrings(self.args)