blob: d2237faf10a6a7b23525e52f09e2894511ffa159 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
# This file is part of cloud-init. See LICENSE file for license information.
from unittest import mock
import pytest
from cloudinit.net.ephemeral import EphemeralIPNetwork
M_PATH = "cloudinit.net.ephemeral."
class TestEphemeralIPNetwork:
@pytest.mark.parametrize("ipv6", [False, True])
@pytest.mark.parametrize("ipv4", [False, True])
@mock.patch(M_PATH + "contextlib.ExitStack")
@mock.patch(M_PATH + "EphemeralIPv6Network")
@mock.patch(M_PATH + "EphemeralDHCPv4")
def test_stack_order(
self,
m_ephemeral_dhcp_v4,
m_ephemeral_ip_v6_network,
m_exit_stack,
ipv4,
ipv6,
):
interface = object()
with EphemeralIPNetwork(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
else:
assert [] == m_ephemeral_dhcp_v4.call_args_list
if ipv6:
expected_call_args_list.append(
mock.call(m_ephemeral_ip_v6_network.return_value)
)
assert [
mock.call(interface)
] == m_ephemeral_ip_v6_network.call_args_list
else:
assert [] == m_ephemeral_ip_v6_network.call_args_list
assert (
expected_call_args_list
== m_exit_stack.return_value.enter_context.call_args_list
)
|
