summaryrefslogtreecommitdiff
path: root/google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py')
-rw-r--r--google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py b/google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py
index e35a576..ead9fa4 100644
--- a/google_compute_engine/distro_lib/freebsd_11/tests/utils_test.py
+++ b/google_compute_engine/distro_lib/freebsd_11/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.IpForwardingUtilsIfconfig')
+ 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)]
+ self.assertEqual(mocks.mock_calls, expected_calls)