def test_valid_routes(self):
routing_table = \
'Iface\tDestination\tGateway \tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU\tWindow\tIRTT \n' \
'eth0\t00000000\tC1BB910A\t0003\t0\t0\t0\t00000000\t0\t0\t0 \n' \
'eth0\tC0BB910A\t00000000\t0001\t0\t0\t0\tC0FFFFFF\t0\t0\t0 \n' \
'eth0\t10813FA8\tC1BB910A\t000F\t0\t0\t0\tFFFFFFFF\t0\t0\t0 \n' \
'eth0\tFEA9FEA9\tC1BB910A\t0007\t0\t0\t0\tFFFFFFFF\t0\t0\t0 \n' \
'docker0\t002BA8C0\t00000000\t0001\t0\t0\t10\t00FFFFFF\t0\t0\t0 \n'
known_sha1_hash = b'\x1e\xd1k\xae[\xf8\x9b\x1a\x13\xd0\xbbT\xa4\xe3Y\xa3\xdd\x0b\xbd\xa9'
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
raw_route_list = osutil.DefaultOSUtil().read_route_table()
self.assertEqual(len(raw_route_list), 6)
self.assertEqual(textutil.hash_strings(raw_route_list), known_sha1_hash)
route_list = osutil.DefaultOSUtil().get_list_of_routes(raw_route_list)
self.assertEqual(len(route_list), 5)
self.assertEqual(route_list[0].gateway_quad(), '10.145.187.193')
self.assertEqual(route_list[1].gateway_quad(), '0.0.0.0')
self.assertEqual(route_list[1].mask_quad(), '255.255.255.192')
self.assertEqual(route_list[2].destination_quad(), '168.63.129.16')
self.assertEqual(route_list[1].flags, 1)
self.assertEqual(route_list[2].flags, 15)
self.assertEqual(route_list[3].flags, 7)
self.assertEqual(route_list[3].metric, 0)
self.assertEqual(route_list[4].metric, 10)
self.assertEqual(route_list[0].interface, 'eth0')
self.assertEqual(route_list[4].interface, 'docker0')
def test_wireserver_route_exists(self):
# setup
dhcp_handler = dhcp.get_dhcp_handler()
self.assertTrue(dhcp_handler.endpoint is None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
# execute
routing_table = "\
Iface Destination Gateway Flags RefCnt Use Metric " \
"Mask MTU Window IRTT \n\
eth0 00000000 10813FA8 0003 0 0 5 " \
"00000000 0 0 0 \n\
eth0 00345B0A 00000000 0001 0 0 5 " \
"00000000 0 0 0 \n\
lo 00000000 01345B0A 0003 0 0 1 " \
"00FCFFFF 0 0 0 \n"
with patch("os.path.exists", return_value=True):
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
self.assertTrue(dhcp_handler.wireserver_route_exists)
# test
self.assertTrue(dhcp_handler.endpoint is not None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
def test_bogus_proc_net_route(self):
routing_table = 'Iface\tDestination\tGateway \tFlags\t\tUse\tMetric\t\neth0\t00000000\t00000000\t0001\t\t0\t0\n'
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
raw_route_list = osutil.DefaultOSUtil().read_route_table()
self.assertEqual(len(osutil.DefaultOSUtil().get_list_of_routes(raw_route_list)), 0)
def test_no_routes(self):
routing_table = 'Iface\tDestination\tGateway \tFlags\tRefCnt\tUse\tMetric\tMask\t\tMTU\tWindow\tIRTT \n'
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
raw_route_list = osutil.DefaultOSUtil().read_route_table()
self.assertEqual(len(osutil.DefaultOSUtil().get_list_of_routes(raw_route_list)), 0)
def test_interface_flags(self):
routing_table = "\
Iface Destination Gateway Flags RefCnt Use Metric Mask MTU Window IRTT \n\
nflg 00000000 01345B0A 0001 0 0 1 00000000 0 0 0 \n\
flgs 00000000 01345B0A 0003 0 0 1 00FCFFFF 0 0 0 \n"
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
self.assertTrue(osutil.DefaultOSUtil().is_primary_interface('flgs'))
def test_dhcp_lease_custom_dns(self):
"""
Validate that the wireserver address is coming from option 245
(on default configurations the address is also available in the domain-name-servers option, but users
may set up a custom dns server on their vnet)
"""
with patch.object(glob, "glob", return_value=['/var/lib/dhcp/dhclient.eth0.leases']):
with patch(open_patch(), mock.mock_open(read_data=load_data("dhcp.leases.custom.dns"))):
endpoint = get_osutil(distro_name='ubuntu', distro_version='14.04').get_dhcp_lease_endpoint() # pylint: disable=assignment-from-none
self.assertEqual(endpoint, "168.63.129.16")
def test_no_interface(self):
routing_table = "\
Iface Destination Gateway Flags RefCnt Use Metric Mask MTU Window IRTT \n\
ndst 00000001 01345B0A 0003 0 0 1 00000000 0 0 0 \n\
nflg 00000000 01345B0A 0001 0 0 1 00FCFFFF 0 0 0 \n"
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
self.assertFalse(osutil.DefaultOSUtil().is_primary_interface('ndst'))
self.assertFalse(osutil.DefaultOSUtil().is_primary_interface('nflg'))
self.assertFalse(osutil.DefaultOSUtil().is_primary_interface('invalid'))
def test_dhcp_skip_cache(self):
handler = dhcp.get_dhcp_handler()
handler.osutil = osutil.DefaultOSUtil()
open_file_mock = mock.mock_open(read_data=TestDHCP.DEFAULT_ROUTING_TABLE)
with patch('os.path.exists', return_value=False):
with patch.object(osutil.DefaultOSUtil, 'get_dhcp_lease_endpoint')\
as patch_dhcp_cache:
with patch.object(dhcp.DhcpHandler, 'send_dhcp_req') \
as patch_dhcp_send:
endpoint = 'foo'
patch_dhcp_cache.return_value = endpoint
# endpoint comes from cache
self.assertFalse(handler.skip_cache)
with patch("os.path.exists", return_value=True):
with patch(open_patch(), open_file_mock):
handler.run()
self.assertTrue(patch_dhcp_cache.call_count == 1)
self.assertTrue(patch_dhcp_send.call_count == 0)
self.assertTrue(handler.endpoint == endpoint)
# reset
handler.skip_cache = True
handler.endpoint = None
# endpoint comes from dhcp request
self.assertTrue(handler.skip_cache)
with patch("os.path.exists", return_value=True):
with patch(open_patch(), open_file_mock):
handler.run()
self.assertTrue(patch_dhcp_cache.call_count == 1)
self.assertTrue(patch_dhcp_send.call_count == 1)
def test_sriov(self):
routing_table = "\
Iface Destination Gateway Flags RefCnt Use Metric Mask MTU Window IRTT \n" \
"bond0 00000000 0100000A 0003 0 0 0 00000000 0 0 0 \n" \
"bond0 0000000A 00000000 0001 0 0 0 00000000 0 0 0 \n" \
"eth0 0000000A 00000000 0001 0 0 0 00000000 0 0 0 \n" \
"bond0 10813FA8 0100000A 0007 0 0 0 00000000 0 0 0 \n" \
"bond0 FEA9FEA9 0100000A 0007 0 0 0 00000000 0 0 0 \n"
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
self.assertFalse(osutil.DefaultOSUtil().is_primary_interface('eth0'))
self.assertTrue(osutil.DefaultOSUtil().is_primary_interface('bond0'))
def test_dhcp_lease_ubuntu(self):
with patch.object(glob, "glob", return_value=['/var/lib/dhcp/dhclient.eth0.leases']):
with patch(open_patch(), mock.mock_open(read_data=load_data("dhcp.leases"))):
endpoint = get_osutil(distro_name='ubuntu', distro_version='12.04').get_dhcp_lease_endpoint() # pylint: disable=assignment-from-none
self.assertTrue(endpoint is not None)
self.assertEqual(endpoint, "168.63.129.16")
endpoint = get_osutil(distro_name='ubuntu', distro_version='12.04').get_dhcp_lease_endpoint() # pylint: disable=assignment-from-none
self.assertTrue(endpoint is not None)
self.assertEqual(endpoint, "168.63.129.16")
endpoint = get_osutil(distro_name='ubuntu', distro_version='14.04').get_dhcp_lease_endpoint() # pylint: disable=assignment-from-none
self.assertTrue(endpoint is not None)
self.assertEqual(endpoint, "168.63.129.16")
def test_wireserver_route_not_exists(self):
# setup
dhcp_handler = dhcp.get_dhcp_handler()
self.assertTrue(dhcp_handler.endpoint is None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
# execute
with patch("os.path.exists", return_value=True):
open_file_mock = mock.mock_open(read_data=TestDHCP.DEFAULT_ROUTING_TABLE)
with patch(open_patch(), open_file_mock):
self.assertFalse(dhcp_handler.wireserver_route_exists)
# test
self.assertTrue(dhcp_handler.endpoint is None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
def test_wireserver_route_exists(self):
# setup
dhcp_handler = dhcp.get_dhcp_handler()
self.assertTrue(dhcp_handler.endpoint is None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
# execute
routing_table_with_wireserver_route = TestDHCP.DEFAULT_ROUTING_TABLE + \
"eth0 00000000 10813FA8 0003 0 0 5 00000000 0 0 0 \n"
with patch("os.path.exists", return_value=True):
open_file_mock = mock.mock_open(read_data=routing_table_with_wireserver_route)
with patch(open_patch(), open_file_mock):
self.assertTrue(dhcp_handler.wireserver_route_exists)
# test
self.assertTrue(dhcp_handler.endpoint is not None)
self.assertTrue(dhcp_handler.routes is None)
self.assertTrue(dhcp_handler.gateway is None)
def test_get_f5_platform_bigip_12_1_1(self):
version_file = textwrap.dedent("""
Product: BIG-IP
Version: 12.1.1
Build: 0.0.184
Sequence: 12.1.1.0.0.184.0
BaseBuild: 0.0.184
Edition: Final
Date: Thu Aug 11 17:09:01 PDT 2016
Built: 160811170901
Changelist: 1874858
JobID: 705993""")
mocked_open = mock.mock_open(read_data=version_file)
with patch(open_patch(), mocked_open):
platform = get_f5_platform()
self.assertTrue(platform[0] == 'bigip')
self.assertTrue(platform[1] == '12.1.1')
self.assertTrue(platform[2] == 'bigip')
self.assertTrue(platform[3] == 'BIG-IP')
def test_get_f5_platform_bigip_12_1_0_hf1(self):
version_file = textwrap.dedent("""
Product: BIG-IP
Version: 12.1.0
Build: 1.0.1447
Sequence: 12.1.0.1.0.1447.0
BaseBuild: 0.0.1434
Edition: Hotfix HF1
Date: Wed Jun 8 13:41:59 PDT 2016
Built: 160608134159
Changelist: 1773831
JobID: 673467""")
mocked_open = mock.mock_open(read_data=version_file)
with patch(open_patch(), mocked_open):
platform = get_f5_platform()
self.assertTrue(platform[0] == 'bigip')
self.assertTrue(platform[1] == '12.1.0')
self.assertTrue(platform[2] == 'bigip')
self.assertTrue(platform[3] == 'BIG-IP')
def test_get_f5_platform_bigiq_5_1_0(self):
version_file = textwrap.dedent("""
Product: BIG-IQ
Version: 5.1.0
Build: 0.0.631
Sequence: 5.1.0.0.0.631.0
BaseBuild: 0.0.631
Edition: Final
Date: Thu Sep 15 19:55:43 PDT 2016
Built: 160915195543
Changelist: 1907534
JobID: 726344""")
mocked_open = mock.mock_open(read_data=version_file)
with patch(open_patch(), mocked_open):
platform = get_f5_platform()
self.assertTrue(platform[0] == 'bigiq')
self.assertTrue(platform[1] == '5.1.0')
self.assertTrue(platform[2] == 'bigiq')
self.assertTrue(platform[3] == 'BIG-IQ')
def test_get_f5_platform_iworkflow_2_0_1(self):
version_file = textwrap.dedent("""
Product: iWorkflow
Version: 2.0.1
Build: 0.0.9842
Sequence: 2.0.1.0.0.9842.0
BaseBuild: 0.0.9842
Edition: Final
Date: Sat Oct 1 22:52:08 PDT 2016
Built: 161001225208
Changelist: 1924048
JobID: 734712""")
mocked_open = mock.mock_open(read_data=version_file)
with patch(open_patch(), mocked_open):
platform = get_f5_platform()
self.assertTrue(platform[0] == 'iworkflow')
self.assertTrue(platform[1] == '2.0.1')
self.assertTrue(platform[2] == 'iworkflow')
self.assertTrue(platform[3] == 'iWorkflow')
def test_get_f5_platform_bigip_12_0_0(self):
version_file = textwrap.dedent("""
Product: BIG-IP
Version: 12.0.0
Build: 0.0.606
Sequence: 12.0.0.0.0.606.0
BaseBuild: 0.0.606
Edition: Final
Date: Fri Aug 21 13:29:22 PDT 2015
Built: 150821132922
Changelist: 1486072
JobID: 536212""")
mocked_open = mock.mock_open(read_data=version_file)
with patch(open_patch(), mocked_open):
platform = get_f5_platform()
self.assertTrue(platform[0] == 'bigip')
self.assertTrue(platform[1] == '12.0.0')
self.assertTrue(platform[2] == 'bigip')
self.assertTrue(platform[3] == 'BIG-IP')
def test_dhcp_lease_multi(self):
with patch.object(glob, "glob", return_value=['/var/lib/dhcp/dhclient.eth0.leases']):
with patch(open_patch(), mock.mock_open(read_data=load_data("dhcp.leases.multi"))):
endpoint = get_osutil(distro_name='ubuntu', distro_version='12.04').get_dhcp_lease_endpoint()
self.assertTrue(endpoint is not None)
self.assertEqual(endpoint, "168.63.129.2")
def test_empty_proc_net_route(self):
routing_table = ""
mo = mock.mock_open(read_data=routing_table)
with patch(open_patch(), mo):
self.assertEqual(len(osutil.DefaultOSUtil().read_route_table()), 0)
