[airos] added tests and mock backends for defaulted converters

Author: edoput

tests/airos/mock.py

@@ -10,8 +10,10 @@
                                                      Ntpclient, Pwdog, Radio,
                                                      Resolv, Route, Snmp, Sshd,
                                                      Syslog, System, Telnetd,
-                                                     Update, Users, Vlan,
-                                                     Wireless, Wpasupplicant)
+                                                     Tshaper,
+                                                     Unms, Update, Upnpd,
+                                                     Users, Vlan, Wireless,
+                                                     Wpasupplicant)
 
 
 class ConverterTest(TestCase):
@@ -237,6 +239,24 @@ class TelnetdAirOs(AirOs):
     ]
 
 
+class TshaperAirOs(AirOs):
+    """
+    Mock backend with converter for tshaper
+    """
+    converters = [
+        Tshaper,
+    ]
+
+
+class UnmsAirOs(AirOs):
+    """
+    Mock backend with converter for unms
+    """
+    converters = [
+        Unms,
+    ]
+
+
 class UpdateAirOs(AirOs):
     """
     Mock backend with converter for update
@@ -246,6 +266,15 @@ class UpdateAirOs(AirOs):
     ]
 
 
+class UpnpdAirOs(AirOs):
+    """
+    Mock backend with converter for updnd daemon
+    """
+    converters = [
+        Upnpd,
+    ]
+
+
 class UsersAirOs(AirOs):
     """
     Mock backend with converter for users settings

tests/airos/test_dhcpc.py

@@ -1,7 +1,7 @@
 from .mock import ConverterTest, DhcpcAirOs
 
 
-class TestNetmodeConverter(ConverterTest):
+class TestDhcpcConverter(ConverterTest):
 
     backend = DhcpcAirOs
 

tests/airos/test_snmp.py

@@ -0,0 +1,43 @@
+from .mock import SnmpAirOs, ConverterTest
+
+
+class TestSnmpConverter(ConverterTest):
+    """
+    tests for backends.airos.renderers.SystemRenderer
+    """
+    backend = SnmpAirOs
+
+    def test_defaults(self):
+
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [
+            {
+                'community': 'public',
+                'contact': '',
+                'location': '',
+                'status': 'enabled',
+            }
+        ]
+
+        self.assertEqualConfig(o.intermediate_data['snmp'], expected)
+
+    def test_custom_info(self):
+
+        o = self.backend({
+            'general': {
+                'mantainer': 'noone@somedomain.com',
+                'location': 'somewhere in the woods',
+            }
+        })
+        o.to_intermediate()
+        expected = [
+            {
+                'community': 'public',
+                'contact': 'noone@somedomain.com',
+                'location': 'somewhere in the woods',
+                'status': 'enabled',
+            }
+        ]
+
+        self.assertEqualConfig(o.intermediate_data['snmp'], expected)

tests/airos/test_syslog.py

@@ -0,0 +1,17 @@
+from .mock import SyslogAirOs, ConverterTest
+
+
+class TestSyslogConverter(ConverterTest):
+    backend = SyslogAirOs
+
+    def test_active(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [
+            {
+                'remote.port': 514,
+                'remote.status': 'disabled',
+                'status': 'enabled',
+            }
+        ]
+        self.assertEqualConfig(o.intermediate_data['syslog'], expected)

tests/airos/test_system.py

@@ -0,0 +1,19 @@
+from .mock import SystemAirOs, ConverterTest
+
+
+class TestSystemConverter(ConverterTest):
+    backend = SystemAirOs
+
+    def test_active(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [
+            {
+                'airosx.prov.status': 'enabled',
+                'cfg.version': 0,
+                'date.status': 'disabled',
+                'external.reset': 'enabled',
+                'timezone': 'GMT'
+            }
+        ]
+        self.assertEqualConfig(o.intermediate_data['system'], expected)

tests/airos/test_telnetd.py

@@ -0,0 +1,16 @@
+from .mock import TelnetdAirOs, ConverterTest
+
+
+class TestTelnetdConverter(ConverterTest):
+    backend = TelnetdAirOs
+
+    def test_active(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [
+            {
+                'port': 23,
+                'status': 'disabled'
+            }
+        ]
+        self.assertEqualConfig(o.intermediate_data['telnetd'], expected)

tests/airos/test_tshaper.py

@@ -0,0 +1,11 @@
+from .mock import TshaperAirOs, ConverterTest
+
+
+class TestTshaperConverter(ConverterTest):
+    backend = TshaperAirOs
+
+    def test_active(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [{'status': 'disabled'}]
+        self.assertEqualConfig(o.intermediate_data['tshaper'], expected)

tests/airos/test_unms.py

@@ -0,0 +1,11 @@
+from .mock import UnmsAirOs, ConverterTest
+
+
+class TestUnmsConverter(ConverterTest):
+    backend = UnmsAirOs
+
+    def test_active(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [{'status': 'disabled'}]
+        self.assertEqualConfig(o.intermediate_data['unms'], expected)

tests/airos/test_update.py

@@ -0,0 +1,13 @@
+from .mock import UpdateAirOs, ConverterTest
+
+
+class TestUpdateConverter(ConverterTest):
+    backend = UpdateAirOs
+
+    def test_status(self):
+        o = self.backend({})
+        o.to_intermediate()
+        expected = [
+            {'check.status': 'enabled'}
+        ]
+        self.assertEqualConfig(o.intermediate_data['update'], expected)

tests/airos/test_upnpd.py

@@ -0,0 +1,29 @@
+from .mock import UpnpdAirOs, ConverterTest
+
+
+class TestUpnpdConverter(ConverterTest):
+    """
+    tests for backends.airos.renderers.SystemRenderer
+    """
+    backend = UpnpdAirOs
+
+    def test_bridge(self):
+
+        o = self.backend({
+            'netmode': 'bridge',
+        })
+        o.to_intermediate()
+        with self.assertRaises(KeyError):
+            o.intermediate_data['upnpd']
+
+    def test_router(self):
+        o = self.backend({
+            'netmode': 'router',
+        })
+        o.to_intermediate()
+        expected = [
+            {
+                'status': 'disabled',
+            }
+        ]
+        self.assertEqualConfig(o.intermediate_data['upnpd'], expected)

tests/airos/test_users.py

@@ -0,0 +1,31 @@
+from .mock import UsersAirOs, ConverterTest
+
+
+class TestUsersConverter(ConverterTest):
+    """
+    tests for backends.airos.renderers.SystemRenderer
+    """
+    backend = UsersAirOs
+
+    def test_user(self):
+
+        o = self.backend({
+            'user': {
+                'name': 'ubnt',
+                'password': 'changeme',
+                'salt': 'goodsalt',
+            }
+        })
+        o.to_intermediate()
+        expected = [
+            {
+                'status': 'enabled',
+            },
+            {
+                '1.name': 'ubnt',
+                '1.password': '$1$goodsalt$changeme',
+                '1.status': 'enabled',
+            }
+        ]
+
+        self.assertEqualConfig(o.intermediate_data['users'], expected)