initial code repo

[stor4nfv.git] / src / ceph / src / ceph-volume / ceph_volume / tests / devices / test_zap.py

diff --git a/src/ceph/src/ceph-volume/ceph_volume/tests/devices/test_zap.py b/src/ceph/src/ceph-volume/ceph_volume/tests/devices/test_zap.py
new file mode 100644 (file)
index 0000000..bc26e33
--- /dev/null
+++ b/src/ceph/src/ceph-volume/ceph_volume/tests/devices/test_zap.py
@@ -0,0 +1,17 @@
+import pytest
+from ceph_volume.devices import lvm
+
+
+class TestZap(object):
+
+    def test_main_spits_help_with_no_arguments(self, capsys):
+        lvm.zap.Zap([]).main()
+        stdout, stderr = capsys.readouterr()
+        assert 'Zaps the given logical volume or partition' in stdout
+
+    def test_main_shows_full_help(self, capsys):
+        with pytest.raises(SystemExit):
+            lvm.zap.Zap(argv=['--help']).main()
+        stdout, stderr = capsys.readouterr()
+        assert 'optional arguments' in stdout
+        assert 'positional arguments' in stdout