diff options
Diffstat (limited to 'tests/unittests/net/test_ephemeral.py')
-rw-r--r-- | tests/unittests/net/test_ephemeral.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/unittests/net/test_ephemeral.py b/tests/unittests/net/test_ephemeral.py index d2237faf..ddd9912c 100644 --- a/tests/unittests/net/test_ephemeral.py +++ b/tests/unittests/net/test_ephemeral.py @@ -5,6 +5,7 @@ from unittest import mock import pytest from cloudinit.net.ephemeral import EphemeralIPNetwork +from tests.unittests.util import MockDistro M_PATH = "cloudinit.net.ephemeral." @@ -24,14 +25,17 @@ class TestEphemeralIPNetwork: ipv6, ): interface = object() - with EphemeralIPNetwork(interface, ipv4=ipv4, ipv6=ipv6): + distro = MockDistro() + with EphemeralIPNetwork(distro, interface, ipv4=ipv4, ipv6=ipv6): pass expected_call_args_list = [] if ipv4: expected_call_args_list.append( mock.call(m_ephemeral_dhcp_v4.return_value) ) - assert [mock.call(interface)] == m_ephemeral_dhcp_v4.call_args_list + assert [ + mock.call(distro, interface) + ] == m_ephemeral_dhcp_v4.call_args_list else: assert [] == m_ephemeral_dhcp_v4.call_args_list if ipv6: |