diff options
Diffstat (limited to 'cloudinit/tests/test_netinfo.py')
| -rw-r--r-- | cloudinit/tests/test_netinfo.py | 106 |
1 files changed, 106 insertions, 0 deletions
diff --git a/cloudinit/tests/test_netinfo.py b/cloudinit/tests/test_netinfo.py new file mode 100644 index 00000000..7dea2e41 --- /dev/null +++ b/cloudinit/tests/test_netinfo.py @@ -0,0 +1,106 @@ +# This file is part of cloud-init. See LICENSE file for license information. + +"""Tests netinfo module functions and classes.""" + +from cloudinit.netinfo import netdev_pformat, route_pformat +from cloudinit.tests.helpers import CiTestCase, mock + + +# Example ifconfig and route output +SAMPLE_IFCONFIG_OUT = """\ +enp0s25 Link encap:Ethernet HWaddr 50:7b:9d:2c:af:91 + inet addr:192.168.2.18 Bcast:192.168.2.255 Mask:255.255.255.0 + inet6 addr: fe80::8107:2b92:867e:f8a6/64 Scope:Link + UP BROADCAST RUNNING MULTICAST MTU:1500 Metric:1 + RX packets:8106427 errors:55 dropped:0 overruns:0 frame:37 + TX packets:9339739 errors:0 dropped:0 overruns:0 carrier:0 + collisions:0 txqueuelen:1000 + RX bytes:4953721719 (4.9 GB) TX bytes:7731890194 (7.7 GB) + Interrupt:20 Memory:e1200000-e1220000 + +lo Link encap:Local Loopback + inet addr:127.0.0.1 Mask:255.0.0.0 + inet6 addr: ::1/128 Scope:Host + UP LOOPBACK RUNNING MTU:65536 Metric:1 + RX packets:579230851 errors:0 dropped:0 overruns:0 frame:0 + TX packets:579230851 errors:0 dropped:0 overruns:0 carrier:0 + collisions:0 txqueuelen:1 +""" + +SAMPLE_ROUTE_OUT = '\n'.join([ + '0.0.0.0 192.168.2.1 0.0.0.0 UG 0 0 0' + ' enp0s25', + '0.0.0.0 192.168.2.1 0.0.0.0 UG 0 0 0' + ' wlp3s0', + '192.168.2.0 0.0.0.0 255.255.255.0 U 0 0 0' + ' enp0s25']) + + +NETDEV_FORMATTED_OUT = '\n'.join([ + '+++++++++++++++++++++++++++++++++++++++Net device info+++++++++++++++++++' + '++++++++++++++++++++', + '+---------+------+------------------------------+---------------+-------+' + '-------------------+', + '| Device | Up | Address | Mask | Scope |' + ' Hw-Address |', + '+---------+------+------------------------------+---------------+-------+' + '-------------------+', + '| enp0s25 | True | 192.168.2.18 | 255.255.255.0 | . |' + ' 50:7b:9d:2c:af:91 |', + '| enp0s25 | True | fe80::8107:2b92:867e:f8a6/64 | . | link |' + ' 50:7b:9d:2c:af:91 |', + '| lo | True | 127.0.0.1 | 255.0.0.0 | . |' + ' . |', + '| lo | True | ::1/128 | . | host |' + ' . |', + '+---------+------+------------------------------+---------------+-------+' + '-------------------+']) + +ROUTE_FORMATTED_OUT = '\n'.join([ + '+++++++++++++++++++++++++++++Route IPv4 info++++++++++++++++++++++++++' + '+++', + '+-------+-------------+-------------+---------------+-----------+-----' + '--+', + '| Route | Destination | Gateway | Genmask | Interface | Flags' + ' |', + '+-------+-------------+-------------+---------------+-----------+' + '-------+', + '| 0 | 0.0.0.0 | 192.168.2.1 | 0.0.0.0 | wlp3s0 |' + ' UG |', + '| 1 | 192.168.2.0 | 0.0.0.0 | 255.255.255.0 | enp0s25 |' + ' U |', + '+-------+-------------+-------------+---------------+-----------+' + '-------+', + '++++++++++++++++++++++++++++++++++++++++Route IPv6 info++++++++++' + '++++++++++++++++++++++++++++++', + '+-------+-------------+-------------+---------------+---------------+' + '-----------------+-------+', + '| Route | Proto | Recv-Q | Send-Q | Local Address |' + ' Foreign Address | State |', + '+-------+-------------+-------------+---------------+---------------+' + '-----------------+-------+', + '| 0 | 0.0.0.0 | 192.168.2.1 | 0.0.0.0 | UG |' + ' 0 | 0 |', + '| 1 | 192.168.2.0 | 0.0.0.0 | 255.255.255.0 | U |' + ' 0 | 0 |', + '+-------+-------------+-------------+---------------+---------------+' + '-----------------+-------+']) + + +class TestNetInfo(CiTestCase): + + maxDiff = None + + @mock.patch('cloudinit.netinfo.util.subp') + def test_netdev_pformat(self, m_subp): + """netdev_pformat properly rendering network device information.""" + m_subp.return_value = (SAMPLE_IFCONFIG_OUT, '') + content = netdev_pformat() + self.assertEqual(NETDEV_FORMATTED_OUT, content) + + @mock.patch('cloudinit.netinfo.util.subp') + def test_route_pformat(self, m_subp): + """netdev_pformat properly rendering network device information.""" + m_subp.return_value = (SAMPLE_ROUTE_OUT, '') + content = route_pformat() + self.assertEqual(ROUTE_FORMATTED_OUT, content) |