diff options
author | creatiwit <shrin.krishnan@gmail.com> | 2019-07-12 17:00:11 -0700 |
---|---|---|
committer | Liam Hopkins <liamh@google.com> | 2019-07-12 17:00:11 -0700 |
commit | e6eab0c12df9192208233b31d8e2ac7e79cba44c (patch) | |
tree | e829053eddb8af77513c7e780c1a72c1015ff52f /packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py | |
parent | f8d3beb109f692a04ae61de314654624309e80ed (diff) | |
download | google-compute-image-packages-e6eab0c12df9192208233b31d8e2ac7e79cba44c.tar.gz |
Support for Google Private Access over IPv6 (#790)
* Enable IPv6 on interfaces based on metadata key.
* Support for FreeBSD and SLES images is still needed.
* Support for multi-NIC is still needed.
* NOTE: `-6` argument to dhclient silently changes pid-file to `/run/dhclient6.pid`
Diffstat (limited to 'packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py')
-rw-r--r-- | packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py | 24 |
1 files changed, 18 insertions, 6 deletions
diff --git a/packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py b/packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py index f47f5c7..1dcbc60 100644 --- a/packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py +++ b/packages/python-google-compute-engine/google_compute_engine/distro_lib/el_6/tests/utils_test.py @@ -30,15 +30,27 @@ class UtilsTest(unittest.TestCase): pass @mock.patch('google_compute_engine.distro_lib.helpers.CallDhclientIpv6') - def testEnableIpv6(self, mock_call): + @mock.patch('google_compute_engine.distro_lib.helpers.CallEnableRouteAdvertisements') + def testEnableIpv6(self, mock_call_enable_ra, mock_call_dhclient): mocks = mock.Mock() - mocks.attach_mock(mock_call, 'call') + mocks.attach_mock(mock_call_dhclient, 'dhclient') + mocks.attach_mock(mock_call_enable_ra, 'enable_ra') - utils.Utils.EnableIpv6( - self.mock_setup, ['A', 'B'], self.mock_logger, - dhclient_script='test_script') + utils.Utils.EnableIpv6(self.mock_setup, ['A', 'B'], self.mock_logger) expected_calls = [ - mock.call.call(['A', 'B'], mock.ANY, dhclient_script='test_script'), + mock.call.enable_ra(['A', 'B'], mock.ANY), + mock.call.dhclient(['A', 'B'], mock.ANY, dhclient_script=None), + ] + self.assertEqual(mocks.mock_calls, expected_calls) + + @mock.patch('google_compute_engine.distro_lib.helpers.CallDhclientIpv6') + def testDisableIpv6(self, mock_call_dhclient): + mocks = mock.Mock() + mocks.attach_mock(mock_call_dhclient, 'dhclient') + + utils.Utils.DisableIpv6(self.mock_setup, ['A', 'B'], self.mock_logger) + expected_calls = [ + mock.call.dhclient(['A', 'B'], mock.ANY, None, release_lease=True), ] self.assertEqual(mocks.mock_calls, expected_calls) |