adding pytest for commons

Author: kkotari

tests/_test_netsim.py

@@ -0,0 +1,145 @@
+"""
+Usage nwrap template create [network | device] [yaml | json] [<fileName>]
+            template load <fileName> [<username> <password>]
+            download <username> <password> link
+            delete-devices <DeviceNames>
+            -v  | --version
+            -vv | --verbose
+            -h  | --help
+
+nwrap is an alias of `netsim-wrapper` and you can pass
+multiple devnames instead of single devname, eg. start, stop, ...
+
+Additionally we support all `ncs-netsim` commands
+"""
+
+import gc
+import sys
+import pytest
+import unittest
+
+from logging import DEBUG, INFO
+from netsim_wrapper import main
+from netsim_wrapper.nwrap import NWrap
+
+from _pytest.monkeypatch import MonkeyPatch
+
+
+# @pytest.mark.run
+class TestNwrapCommon(unittest.TestCase):
+
+    def setUp(self):
+        self.monkeypatch = MonkeyPatch()
+        self.args = ['nwrap']
+
+    def tearDown(self) -> None:
+        gc.collect()
+        self.monkeypatch.undo()
+        return super().tearDown()
+
+    @property
+    def applyPatch(self):
+        self.monkeypatch.setattr(sys, 'argv', self.args)
+
+    @pytest.fixture(autouse=True)
+    def inject_fixtures(self, capfd):
+        self._capfd = capfd
+
+    def test_nwrap_version(self):
+        self.args.append('-v')
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            assert out.strip() == 'netsim-wrapper version 4.0.0'
+
+    def test_nwrap_long_version(self):
+        self.args.append('--version')
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            assert out.strip() == 'netsim-wrapper version 4.0.0'
+
+    def test_nwrap(self):
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            out = out.strip()
+            assert 'Usage nwrap template create' in out
+            assert 'template create [network | device] [yaml | json] [<fileName>]' in out
+            assert 'template load <fileName> [<username> <password>]' in out
+            assert 'download <username> <password> link' in out
+            assert 'delete-devices <DeviceNames>' in out
+            assert '-v  | --version' in out
+            assert '-vv | --verbose' in out
+            assert '-h  | --help' in out
+    
+    def test_nwrap_help(self):
+        self.args.append('-h')
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            out = out.strip()
+            assert 'Usage nwrap template create' in out
+    
+    def test_nwrap_long_help(self):
+        self.args.append('--help')
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            out = out.strip()
+            assert 'Usage nwrap template create' in out
+
+
+# @pytest.mark.run
+class TestNwrapVerbose(unittest.TestCase):
+
+    def setUp(self):
+        gc.collect()
+        self.monkeypatch = MonkeyPatch()
+        self.args = ['nwrap']
+
+    def tearDown(self) -> None:
+        self.monkeypatch.undo()
+        return super().tearDown()
+
+    @property
+    def applyPatch(self):
+        self.monkeypatch.setattr(sys, 'argv', self.args)
+
+    @pytest.fixture(autouse=True)
+    def inject_fixtures(self, capfd):
+        self._capfd = capfd
+
+    def test_nwrap_verbose(self):
+        self.args.append('-vv')
+        self.applyPatch
+        nwrap, p = main.check_verbose(self.args)
+        assert nwrap.log.level == INFO # DEBUG
+        # TODO: We logger level is getting set to INFO due to singleton objects
+
+    def test_nwrap_long_verbose(self):
+        self.args.append('--verbose')
+        self.applyPatch
+        nwrap, p = main.check_verbose(self.args)
+        assert nwrap.log.level == INFO # DEBUG
+        # TODO: We logger level is getting set to INFO due to singleton objects