Test version info printing

Author: MrMino

tests/test_find_extra_reqs.py

@@ -166,3 +166,13 @@ def parse_args(self):
         assert messages[1:] == result
     else:
         assert messages == result
+
+
+def test_main_version(monkeypatch, capsys, fake_opts):
+    fake_opts.options.version = True
+    monkeypatch.setattr(optparse, 'OptionParser', fake_opts)
+
+    with pytest.raises(SystemExit):
+        find_extra_reqs.main()
+
+    assert capsys.readouterr().out == common.version_info() + "\n"

tests/test_find_missing_reqs.py

@@ -176,3 +176,13 @@ def parse_args(self):
         assert messages[1:] == result
     else:
         assert messages == result
+
+
+def test_main_version(monkeypatch, capsys, fake_opts):
+    fake_opts.options.version = True
+    monkeypatch.setattr(optparse, 'OptionParser', fake_opts)
+
+    with pytest.raises(SystemExit):
+        find_missing_reqs.main()
+
+    assert capsys.readouterr().out == common.version_info() + "\n"