diff options
Diffstat (limited to 'google_compute_engine/distro_lib/debian_9/tests/utils_test.py')
-rw-r--r-- | google_compute_engine/distro_lib/debian_9/tests/utils_test.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/google_compute_engine/distro_lib/debian_9/tests/utils_test.py b/google_compute_engine/distro_lib/debian_9/tests/utils_test.py index ebfde1a..2717f00 100644 --- a/google_compute_engine/distro_lib/debian_9/tests/utils_test.py +++ b/google_compute_engine/distro_lib/debian_9/tests/utils_test.py @@ -44,3 +44,12 @@ class UtilsTest(unittest.TestCase): utils.Utils.HandleClockSync(self.mock_setup, self.mock_logger) expected_calls = [mock.call.call(mock.ANY)] self.assertEqual(mocks.mock_calls, expected_calls) + + @mock.patch('google_compute_engine.distro_lib.ip_forwarding_utils.IpForwardingUtilsIproute') + def testIpForwardingUtils(self, mock_call): + mocks = mock.Mock() + mocks.attach_mock(mock_call, 'call') + + utils.Utils.IpForwardingUtils(self.mock_setup, self.mock_logger, '66') + expected_calls = [mock.call.call(mock.ANY, '66')] + self.assertEqual(mocks.mock_calls, expected_calls) |