Extracting Ns._process_net_params() and creating unit test
[osm/RO.git] / NG-RO / osm_ng_ro / tests / test_ns.py
index 6980b88..6cc5f77 100644 (file)
@@ -1788,3 +1788,503 @@ class TestNs(unittest.TestCase):
 
         self.assertTrue(epa_params.called)
         self.assertDictEqual(result, expected_result)
+
+    def test__ip_profile_to_ro_with_none(self):
+        ip_profile = None
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertIsNone(result)
+
+    def test__ip_profile_to_ro_with_empty_profile(self):
+        ip_profile = {}
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertIsNone(result)
+
+    def test__ip_profile_to_ro_with_wrong_profile(self):
+        ip_profile = {
+            "no-profile": "here",
+        }
+        expected_result = {
+            "ip_version": "IPv4",
+            "subnet_address": None,
+            "gateway_address": None,
+            "dhcp_enabled": False,
+            "dhcp_start_address": None,
+            "dhcp_count": None,
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    def test__ip_profile_to_ro_with_ipv4_profile(self):
+        ip_profile = {
+            "ip-version": "ipv4",
+            "subnet-address": "192.168.0.0/24",
+            "gateway-address": "192.168.0.254",
+            "dhcp-params": {
+                "enabled": True,
+                "start-address": "192.168.0.10",
+                "count": 25,
+            },
+        }
+        expected_result = {
+            "ip_version": "IPv4",
+            "subnet_address": "192.168.0.0/24",
+            "gateway_address": "192.168.0.254",
+            "dhcp_enabled": True,
+            "dhcp_start_address": "192.168.0.10",
+            "dhcp_count": 25,
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    def test__ip_profile_to_ro_with_ipv6_profile(self):
+        ip_profile = {
+            "ip-version": "ipv6",
+            "subnet-address": "2001:0200:0001::/48",
+            "gateway-address": "2001:0200:0001:ffff:ffff:ffff:ffff:fffe",
+            "dhcp-params": {
+                "enabled": True,
+                "start-address": "2001:0200:0001::0010",
+                "count": 25,
+            },
+        }
+        expected_result = {
+            "ip_version": "IPv6",
+            "subnet_address": "2001:0200:0001::/48",
+            "gateway_address": "2001:0200:0001:ffff:ffff:ffff:ffff:fffe",
+            "dhcp_enabled": True,
+            "dhcp_start_address": "2001:0200:0001::0010",
+            "dhcp_count": 25,
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    def test__ip_profile_to_ro_with_dns_server(self):
+        ip_profile = {
+            "ip-version": "ipv4",
+            "subnet-address": "192.168.0.0/24",
+            "gateway-address": "192.168.0.254",
+            "dhcp-params": {
+                "enabled": True,
+                "start-address": "192.168.0.10",
+                "count": 25,
+            },
+            "dns-server": [
+                {
+                    "address": "8.8.8.8",
+                },
+                {
+                    "address": "1.1.1.1",
+                },
+                {
+                    "address": "1.0.0.1",
+                },
+            ],
+        }
+        expected_result = {
+            "ip_version": "IPv4",
+            "subnet_address": "192.168.0.0/24",
+            "gateway_address": "192.168.0.254",
+            "dhcp_enabled": True,
+            "dhcp_start_address": "192.168.0.10",
+            "dhcp_count": 25,
+            "dns_address": "8.8.8.8;1.1.1.1;1.0.0.1",
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    def test__ip_profile_to_ro_with_security_group(self):
+        ip_profile = {
+            "ip-version": "ipv4",
+            "subnet-address": "192.168.0.0/24",
+            "gateway-address": "192.168.0.254",
+            "dhcp-params": {
+                "enabled": True,
+                "start-address": "192.168.0.10",
+                "count": 25,
+            },
+            "security-group": {
+                "some-security-group": "here",
+            },
+        }
+        expected_result = {
+            "ip_version": "IPv4",
+            "subnet_address": "192.168.0.0/24",
+            "gateway_address": "192.168.0.254",
+            "dhcp_enabled": True,
+            "dhcp_start_address": "192.168.0.10",
+            "dhcp_count": 25,
+            "security_group": {
+                "some-security-group": "here",
+            },
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    def test__ip_profile_to_ro(self):
+        ip_profile = {
+            "ip-version": "ipv4",
+            "subnet-address": "192.168.0.0/24",
+            "gateway-address": "192.168.0.254",
+            "dhcp-params": {
+                "enabled": True,
+                "start-address": "192.168.0.10",
+                "count": 25,
+            },
+            "dns-server": [
+                {
+                    "address": "8.8.8.8",
+                },
+                {
+                    "address": "1.1.1.1",
+                },
+                {
+                    "address": "1.0.0.1",
+                },
+            ],
+            "security-group": {
+                "some-security-group": "here",
+            },
+        }
+        expected_result = {
+            "ip_version": "IPv4",
+            "subnet_address": "192.168.0.0/24",
+            "gateway_address": "192.168.0.254",
+            "dhcp_enabled": True,
+            "dhcp_start_address": "192.168.0.10",
+            "dhcp_count": 25,
+            "dns_address": "8.8.8.8;1.1.1.1;1.0.0.1",
+            "security_group": {
+                "some-security-group": "here",
+            },
+        }
+
+        result = Ns._ip_profile_to_ro(
+            ip_profile=ip_profile,
+        )
+
+        self.assertDictEqual(expected_result, result)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_empty_params(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "provider_network": "some-profile-here",
+        }
+        target_record_id = ""
+        expected_result = {
+            "params": {
+                "net_name": "ns-name-vld-name",
+                "net_type": "bridge",
+                "ip_profile": {
+                    "some_ip_profile": "here",
+                },
+                "provider_network_profile": "some-profile-here",
+            }
+        }
+
+        ip_profile_to_ro.return_value = {
+            "some_ip_profile": "here",
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertTrue(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_vim_info_sdn(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "sdn": "some-sdn",
+            "sdn-ports": ["some", "ports", "here"],
+            "vlds": ["some", "vlds", "here"],
+            "type": "sdn-type",
+        }
+        target_record_id = "vld.sdn.something"
+        expected_result = {
+            "params": {
+                "sdn-ports": ["some", "ports", "here"],
+                "vlds": ["some", "vlds", "here"],
+                "type": "sdn-type",
+            }
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertFalse(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_vim_info_sdn_target_vim(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "sdn": "some-sdn",
+            "sdn-ports": ["some", "ports", "here"],
+            "vlds": ["some", "vlds", "here"],
+            "target_vim": "some-vim",
+            "type": "sdn-type",
+        }
+        target_record_id = "vld.sdn.something"
+        expected_result = {
+            "depends_on": ["some-vim vld.sdn"],
+            "params": {
+                "sdn-ports": ["some", "ports", "here"],
+                "vlds": ["some", "vlds", "here"],
+                "target_vim": "some-vim",
+                "type": "sdn-type",
+            },
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertFalse(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_vim_network_name(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "vim_network_name": "some-network-name",
+        }
+        target_record_id = "vld.sdn.something"
+        expected_result = {
+            "find_params": {
+                "filter_dict": {
+                    "name": "some-network-name",
+                },
+            },
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertFalse(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_vim_network_id(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "vim_network_id": "some-network-id",
+        }
+        target_record_id = "vld.sdn.something"
+        expected_result = {
+            "find_params": {
+                "filter_dict": {
+                    "id": "some-network-id",
+                },
+            },
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertFalse(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_mgmt_network(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "id": "vld-id",
+            "name": "vld-name",
+            "mgmt-network": "some-mgmt-network",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {}
+        target_record_id = "vld.sdn.something"
+        expected_result = {
+            "find_params": {
+                "mgmt": True,
+                "name": "vld-id",
+            },
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertFalse(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_underlay_eline(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+            "underlay": "some-underlay-here",
+            "type": "ELINE",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "provider_network": "some-profile-here",
+        }
+        target_record_id = ""
+        expected_result = {
+            "params": {
+                "ip_profile": {
+                    "some_ip_profile": "here",
+                },
+                "net_name": "ns-name-vld-name",
+                "net_type": "ptp",
+                "provider_network_profile": "some-profile-here",
+            }
+        }
+
+        ip_profile_to_ro.return_value = {
+            "some_ip_profile": "here",
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertTrue(ip_profile_to_ro.called)
+
+    @patch("osm_ng_ro.ns.Ns._ip_profile_to_ro")
+    def test__process_net_params_with_underlay_elan(
+        self,
+        ip_profile_to_ro,
+    ):
+        target_vld = {
+            "name": "vld-name",
+            "underlay": "some-underlay-here",
+            "type": "ELAN",
+        }
+        indata = {
+            "name": "ns-name",
+        }
+        vim_info = {
+            "provider_network": "some-profile-here",
+        }
+        target_record_id = ""
+        expected_result = {
+            "params": {
+                "ip_profile": {
+                    "some_ip_profile": "here",
+                },
+                "net_name": "ns-name-vld-name",
+                "net_type": "data",
+                "provider_network_profile": "some-profile-here",
+            }
+        }
+
+        ip_profile_to_ro.return_value = {
+            "some_ip_profile": "here",
+        }
+
+        result = Ns._process_net_params(
+            target_vld=target_vld,
+            indata=indata,
+            vim_info=vim_info,
+            target_record_id=target_record_id,
+        )
+
+        self.assertDictEqual(expected_result, result)
+        self.assertTrue(ip_profile_to_ro.called)