Add new unit test for sanitized filtered configs

Author: bewing

napalm/base/test/getters.py

@@ -541,6 +541,38 @@ def test_get_config_sanitized(self, test_case):
 
         return get_config
 
+    @wrap_test_cases
+    def test_get_config_sanitized_filtered(self, test_case):
+        """Test get_config with both sanitized=True and retrieve parameter."""
+        return_config = {}
+        get_config = self.device.get_config(retrieve="running", sanitized=True)
+        assert isinstance(get_config, dict)
+        assert helpers.test_model(models.ConfigDict, get_config)
+        assert get_config["startup"] == ""
+        assert get_config["candidate"] == ""
+        assert get_config["running"] != ""
+        assert (
+            "<removed>" in get_config["running"]
+        ), "Running config should be sanitized"
+
+        return_config["running"] = get_config["running"]
+
+        get_config = self.device.get_config(retrieve="startup", sanitized=True)
+        assert isinstance(get_config, dict)
+        assert helpers.test_model(models.ConfigDict, get_config)
+        assert get_config["running"] == ""
+        assert get_config["candidate"] == ""
+        assert get_config["startup"] != ""
+        assert (
+            "<removed>" in get_config["startup"]
+        ), "Startup config should be sanitized"
+
+        return_config["startup"] = get_config["startup"]
+
+        return_config["candidate"] = ""
+
+        return return_config
+
     @wrap_test_cases
     def test_get_network_instances(self, test_case):
         """Test get_network_instances method."""

test/eos/mocked_data/test_get_config_sanitized_filtered/normal/expected_result.json

@@ -0,0 +1,5 @@
+{
+  "running": "! Command: show running-config\n!\n! device: localhost (vEOS, EOS-4.15.2.1F)\n!\n! boot system flash:/vEOS-lab.swi\n!\nevent-handler dhclient\n   trigger on-boot\n      action bash sudo /mnt/flash/initialize_ma1.sh\n      !\n      transceiver qsfp default-mode 4x10G\n      !\n      spanning-tree mode mstp\n      !\n      aaa authorization exec default local\n      !\n      aaa root secret 5 <removed>\n      !\n      username admin privilege 15 role network-admin secret 5 <removed>\n      username vagrant privilege 15 role network-admin secret 5 <removed>\n      !\n      interface Ethernet1\n      !\n      interface Ethernet2\n      !\n      interface Management1\n         ip address 10.0.2.15/24\n\t !\n\t no ip routing\n\t !\n\t management api http-commands\n\t    no shutdown\n\t    !\n\t    !\n\t    end\n",
+  "startup": "! Command: show startup-config\n! Startup-config last modified at  Sun Sep 11 18:35:01 2016 by root\n! device: localhost (vEOS, EOS-4.15.2.1F)\n!\n! boot system flash:/vEOS-lab.swi\n!\nevent-handler dhclient\n   trigger on-boot\n      action bash sudo /mnt/flash/initialize_ma1.sh\n      !\n      transceiver qsfp default-mode 4x10G\n      !\n      spanning-tree mode mstp\n      !\n      aaa authorization exec default local\n      !\n      aaa root secret 5 <removed>\n      !\n      username admin privilege 15 role network-admin secret 5 <removed>\n      username vagrant privilege 15 role network-admin secret 5 <removed>\n      !\n      interface Ethernet1\n      !\n      interface Ethernet2\n      !\n      interface Management1\n         ip address 10.0.2.15/24\n\t !\n\t no ip routing\n\t !\n\t management api http-commands\n\t    no shutdown\n\t    !\n\t    !\n\t    end\n",
+  "candidate": ""
+}

test/eos/mocked_data/test_get_config_sanitized_filtered/normal/show_running_config_sanitized.text

@@ -0,0 +1,35 @@
+! Command: show running-config
+!
+! device: localhost (vEOS, EOS-4.15.2.1F)
+!
+! boot system flash:/vEOS-lab.swi
+!
+event-handler dhclient
+   trigger on-boot
+      action bash sudo /mnt/flash/initialize_ma1.sh
+      !
+      transceiver qsfp default-mode 4x10G
+      !
+      spanning-tree mode mstp
+      !
+      aaa authorization exec default local
+      !
+      aaa root secret 5 <removed>
+      !
+      username admin privilege 15 role network-admin secret 5 <removed>
+      username vagrant privilege 15 role network-admin secret 5 <removed>
+      !
+      interface Ethernet1
+      !
+      interface Ethernet2
+      !
+      interface Management1
+         ip address 10.0.2.15/24
+	 !
+	 no ip routing
+	 !
+	 management api http-commands
+	    no shutdown
+	    !
+	    !
+	    end

test/eos/mocked_data/test_get_config_sanitized_filtered/normal/show_startup_config.text

@@ -0,0 +1,35 @@
+! Command: show startup-config
+! Startup-config last modified at  Sun Sep 11 18:35:01 2016 by root
+! device: localhost (vEOS, EOS-4.15.2.1F)
+!
+! boot system flash:/vEOS-lab.swi
+!
+event-handler dhclient
+   trigger on-boot
+      action bash sudo /mnt/flash/initialize_ma1.sh
+      !
+      transceiver qsfp default-mode 4x10G
+      !
+      spanning-tree mode mstp
+      !
+      aaa authorization exec default local
+      !
+      aaa root secret 5 $1$zr7sHZaW$WagNykw1d5wjy4ZhtMgUS/
+      !
+      username admin privilege 15 role network-admin secret 5 $1$VoQBHXed$4z2.EoAeoIY2SFme/Pz3Q/
+      username vagrant privilege 15 role network-admin secret 5 $1$l/6gq.Qs$ifURvKWUYzLm0RqmRgr.W1
+      !
+      interface Ethernet1
+      !
+      interface Ethernet2
+      !
+      interface Management1
+         ip address 10.0.2.15/24
+	 !
+	 no ip routing
+	 !
+	 management api http-commands
+	    no shutdown
+	    !
+	    !
+	    end