adding more test-cases

Author: kirankotari

netsim_wrapper/nso.py

@@ -223,11 +223,11 @@ def extract_neds(self, signed_bin=True, extract_tar=True):
         self.utils.change_dir(self.default['ppath'])
         for ned in files:
             if signed_bin:
-                self.utils.cmd.run(['bash', ned])
+                self.utils.cmd.run(['bash', ned, '--skip-verification'])
             if extract_tar:
                 tar_file = self.utils.get_tar(ned)
                 self.utils.cmd.run(['tar', '-xvf', tar_file])
-                self.utils.cmd.run(['rm', '-rf', tar_file, ned])
+                self.utils.cmd.run(['rm', '-rf', tar_file, ned, '*.signature', '*.py', '*.py3', 'tailf.cer'])
         self.log.debug('extraction done')
 
     def create_devices(self):

tests/.gitignore

@@ -0,0 +1,3 @@
+# custom file ignore from tests folder
+nso-run/
+*cisco*

tests/custom_authgroup.xml

@@ -0,0 +1,32 @@
+<devices xmlns="http://tail-f.com/ns/ncs">
+    <authgroups>
+        <group>
+            <name>default</name>
+            <default-map>
+                <remote-name>admin</remote-name>
+                <remote-password>admin</remote-password>
+                <remote-secondary-password>admin</remote-secondary-password>
+            </default-map>
+            <umap>
+                <local-user>admin</local-user>
+                <remote-name>admin</remote-name>
+                <remote-password>admin</remote-password>
+            </umap>
+            <umap>
+                <local-user>oper</local-user>
+                <remote-name>oper</remote-name>
+                <remote-password>admin</remote-password>
+            </umap>
+            <umap>
+                <local-user>cisco</local-user>
+                <remote-name>cisco</remote-name>
+                <remote-password>cisco</remote-password>
+            </umap>
+            <umap>
+                <local-user>lab</local-user>
+                <remote-name>lab</remote-name>
+                <remote-password>lab</remote-password>
+            </umap>
+        </group>
+    </authgroups>
+</devices>

tests/network.json

@@ -0,0 +1,28 @@
+{
+  "nso-packages-path": "./nso-run/packages",
+  "download-neds": false,
+  "neds": [
+    "https://ned-link"
+  ],
+  "compile-neds": true,
+  "start-devices": true,
+  "add-to-nso": true,
+  "add-authgroup-to-nso": true,
+  "authgroup": {
+    "type": "system"
+  },
+  "device-mode": {
+    "prefix-based": {
+      "cisco-ios": {
+        "count": 2,
+        "prefix": "ios"
+      }
+    }
+  },
+  "load-day0-config": true,
+  "config-path": "./day0/",
+  "config-files": [
+    "interface_loopback10.cfg",
+    "interface_loopback100.cfg"
+  ]
+}

tests/test_nwrap_common.py

@@ -142,4 +142,16 @@ def test_nwrap_long_verbose(self):
         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
+        # TODO: We logger level is getting set to INFO due to singleton objects
+
+    def test_nwrap_dummy(self):
+        self.args.append('--dummy')
+        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

tests/test_nwrap_delete_device.py

@@ -0,0 +1,70 @@
+"""
+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 os
+import sys
+import pytest
+import shutil
+import unittest
+
+from logging import DEBUG, INFO
+from netsim_wrapper import main
+from netsim_wrapper import utils
+
+from _pytest.monkeypatch import MonkeyPatch
+
+
+# @pytest.mark.run
+class TestNwrapDeleteDevice(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
+
+    @pytest.fixture(autouse=True)
+    def inject_fixtures_log(self, caplog):
+        self._caplog = caplog
+
+    def test_nwrap_delete_device(self):
+        # command = 'template create network yaml'.split()
+        # self.args.extend(command)
+        # self.applyPatch
+        # try:
+        #     main.nwrap()
+        # except SystemExit as e:
+        #     assert type(e) == SystemExit
+        # finally:
+        #     out, err = self._capfd.readouterr()
+        #     assert out.strip() == ''
+
+        #     messages = [ x.message for x in self._caplog.get_records('call') if x.levelno == INFO ]
+        #     msg = 'update the base YAML template file: template.yaml'
+        #     assert msg in messages
+            self._caplog.clear()
+

tests/test_nwrap_download.py

@@ -0,0 +1,72 @@
+"""
+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 os
+import sys
+import pytest
+import shutil
+import unittest
+
+from logging import DEBUG, INFO
+from netsim_wrapper import main
+from netsim_wrapper import utils
+
+from _pytest.monkeypatch import MonkeyPatch
+
+
+# @pytest.mark.run
+class TestNwrapDownload(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)
+        self.monkeypatch.setattr(utils.Command, 'run', lambda s, c, sensitive: 'download successful') # TODO: need to add mock for subprocessor
+
+    @pytest.fixture(autouse=True)
+    def inject_fixtures(self, capfd):
+        self._capfd = capfd
+
+    @pytest.fixture(autouse=True)
+    def inject_fixtures_log(self, caplog):
+        self._caplog = caplog
+
+    def test_nwrap_download(self):
+        command = 'template download user_test password_test https://test.com/download.txt'.split()
+        self.args.extend(command)
+        self.applyPatch
+        try:
+            main.nwrap()
+        except SystemExit as e:
+            assert type(e) == SystemExit
+        finally:
+            out, err = self._capfd.readouterr()
+            assert out.strip() == ''
+
+            messages = [ x.message for x in self._caplog.get_records('call') if x.levelno == INFO ]
+            msg = 'download: https://test.com/download.txt'
+            assert msg in messages
+            self._caplog.clear()
+
+

tests/test_nwrap_template.py

@@ -17,11 +17,12 @@
 import os
 import sys
 import pytest
+import shutil
 import unittest
 
 from logging import DEBUG, INFO
 from netsim_wrapper import main
-from netsim_wrapper.nwrap import NWrap
+from netsim_wrapper import utils
 
 from _pytest.monkeypatch import MonkeyPatch
 
@@ -188,25 +189,88 @@ def test_nwrap_create_device_json_with_file(self):
             assert msg in messages
             self._caplog.clear()
 
+class TestNwrapTemplateData:
+    @classmethod
+    def network(cls):
+        return {
+            "nso-packages-path": "./nso-run/packages",
+            "download-neds": True,
+            "neds": [
+                "https://earth.tail-f.com:8443/ncs-pkgs/cisco-ios/5.7.1/ncs-5.7.1-cisco-ios-6.79.signed.bin"
+            ],
+            "compile-neds": True,
+            "start-devices": True,
+            "add-to-nso": True,
+            "add-authgroup-to-nso": True,
+            "authgroup": {
+                "type": "system"
+            },
+            "device-mode": {
+                "prefix-based": {
+                "cisco-ios-6.56": {
+                    "count": 2,
+                    "prefix": "network-ios"
+                }
+                }
+            },
+            "load-day0-config": True,
+            "config-path": "./day0/",
+            "config-files": [
+                "interface_loopback10.cfg",
+                "interface_loopback100.cfg",
+            ]
+        }
+
+    @classmethod
+    def device(cls):
+        return {
+            "nso-packages-path": "./nso-run/packages",
+            "download-neds": True,
+            "neds": [
+                "https://earth.tail-f.com:8443/ncs-pkgs/cisco-ios/5.7.1/ncs-5.7.1-cisco-ios-6.79.signed.bin"
+            ],
+            "compile-neds": True,
+            "start-devices": True,
+            "add-to-nso": True,
+            "add-authgroup-to-nso": True,
+            "authgroup": {
+                "type": "custom",
+                "path": "./custom_authgroup.xml"
+            },
+            "device-mode": {
+                "name-based": {
+                "cisco-ios-6.56": [
+                    "device-ios0",
+                    "device-ios1"
+                ]
+                }
+            },
+            "load-day0-config": True,
+            "config-path": "./day0/",
+            "config-files": [
+                "interface_loopback10.cfg",
+                "interface_loopback100.cfg",
+            ]
+        }
 
 # @pytest.mark.run
-class TestNwrapTemplateLoadNetwork(unittest.TestCase):
+class TestNwrapTemplateLoad(unittest.TestCase):
+    yaml = utils.Yaml()
+    json = utils.Json()
 
     def setUp(self):
         self.monkeypatch = MonkeyPatch()
         self.args = ['nwrap']
-        files = ['template.json', 'template.yaml', 'basic_template.json', 'basic_template.yaml']
 
     def tearDown(self) -> None:
         gc.collect()
-        files = ['template.json', 'template.yaml', 'basic_template.json', 'basic_template.yaml']
-        [os.unlink(each) for each in files if os.path.exists(each)]
         self.monkeypatch.undo()
         return super().tearDown()
 
     @property
     def applyPatch(self):
         self.monkeypatch.setattr(sys, 'argv', self.args)
+        self.monkeypatch.setattr(utils.Download, 'get', lambda s, u, p, l: None)
 
     @pytest.fixture(autouse=True)
     def inject_fixtures(self, capfd):
@@ -216,3 +280,20 @@ def inject_fixtures(self, capfd):
     def inject_fixtures_log(self, caplog):
         self._caplog = caplog
 
+    def test_nwrap_create_network_json(self):
+        self.json.dump(TestNwrapTemplateData.network(), 'network.json')
+        command = 'template load network.json'.split()
+        self.args.extend(command)
+        self.applyPatch
+        # try:
+        #     main.nwrap()
+        # except SystemExit as e:
+        #     assert type(e) == SystemExit
+        # finally:
+        #     out, err = self._capfd.readouterr()
+        #     assert out.strip() == ''
+        #     # TODO: need to add some validations
+        #     messages = [ x.message for x in self._caplog.get_records('call') if x.levelno == INFO ]
+        #     msg = 'update the base YAML template file: template.yaml'
+        #     assert msg in messages
+        #     self._caplog.clear()