mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
d4507f1a08
Re-factoring a little following the previous commit, so that the same container connection logic can be used in both places, rather than it being repeated.
217 lines
9.5 KiB
Python
217 lines
9.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
Unit tests for the docker state
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import skipIf, TestCase
|
|
from tests.support.mock import (
|
|
Mock,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON,
|
|
patch
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.dockermod as docker_mod
|
|
import salt.states.docker_network as docker_state
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class DockerNetworkTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test docker_network states
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {
|
|
docker_mod: {
|
|
'__context__': {'docker.docker_version': ''}
|
|
},
|
|
docker_state: {
|
|
'__opts__': {'test': False}
|
|
}
|
|
}
|
|
|
|
def test_present(self):
|
|
'''
|
|
Test docker_network.present
|
|
'''
|
|
docker_create_network = Mock(return_value='created')
|
|
docker_connect_container_to_network = Mock(return_value='connected')
|
|
docker_inspect_container = Mock(return_value={'Id': 'abcd', 'Name': 'container_bar'})
|
|
# Get docker.networks to return a network with a name which is a superset of the name of
|
|
# the network which is to be created, despite this network existing we should still expect
|
|
# that the new network will be created.
|
|
# Regression test for #41982.
|
|
docker_networks = Mock(return_value=[{
|
|
'Name': 'network_foobar',
|
|
'Containers': {'container': {}}
|
|
}])
|
|
__salt__ = {'docker.create_network': docker_create_network,
|
|
'docker.inspect_container': docker_inspect_container,
|
|
'docker.connect_container_to_network': docker_connect_container_to_network,
|
|
'docker.networks': docker_networks,
|
|
}
|
|
with patch.dict(docker_state.__dict__,
|
|
{'__salt__': __salt__}):
|
|
ret = docker_state.present(
|
|
'network_foo',
|
|
containers=['container'],
|
|
gateway='192.168.0.1',
|
|
ip_range='192.168.0.128/25',
|
|
subnet='192.168.0.0/24'
|
|
)
|
|
docker_create_network.assert_called_with('network_foo',
|
|
driver=None,
|
|
driver_opts=None,
|
|
gateway='192.168.0.1',
|
|
ip_range='192.168.0.128/25',
|
|
subnet='192.168.0.0/24')
|
|
docker_connect_container_to_network.assert_called_with('abcd',
|
|
'network_foo')
|
|
self.assertEqual(ret, {'name': 'network_foo',
|
|
'comment': '',
|
|
'changes': {'connected': ['container_bar'],
|
|
'created': 'created'},
|
|
'result': True})
|
|
|
|
def test_present_with_change(self):
|
|
'''
|
|
Test docker_network.present when the specified network has properties differing from the already present network
|
|
'''
|
|
network_details = {
|
|
'Id': 'abcd',
|
|
'Name': 'network_foo',
|
|
'Driver': 'macvlan',
|
|
'Containers': {
|
|
'abcd': {}
|
|
},
|
|
'Options': {
|
|
'parent': 'eth0'
|
|
},
|
|
'IPAM': {
|
|
'Config': [
|
|
{
|
|
'Subnet': '192.168.0.0/24',
|
|
'Gateway': '192.168.0.1'
|
|
}
|
|
]
|
|
}
|
|
}
|
|
docker_networks = Mock(return_value=[network_details])
|
|
network_details['Containers'] = {'abcd': {'Id': 'abcd', 'Name': 'container_bar'}}
|
|
docker_inspect_network = Mock(return_value=network_details)
|
|
docker_inspect_container = Mock(return_value={'Id': 'abcd', 'Name': 'container_bar'})
|
|
docker_disconnect_container_from_network = Mock(return_value='disconnected')
|
|
docker_remove_network = Mock(return_value='removed')
|
|
docker_create_network = Mock(return_value='created')
|
|
docker_connect_container_to_network = Mock(return_value='connected')
|
|
|
|
__salt__ = {'docker.networks': docker_networks,
|
|
'docker.inspect_network': docker_inspect_network,
|
|
'docker.inspect_container': docker_inspect_container,
|
|
'docker.disconnect_container_from_network': docker_disconnect_container_from_network,
|
|
'docker.remove_network': docker_remove_network,
|
|
'docker.create_network': docker_create_network,
|
|
'docker.connect_container_to_network': docker_connect_container_to_network,
|
|
}
|
|
with patch.dict(docker_state.__dict__,
|
|
{'__salt__': __salt__}):
|
|
ret = docker_state.present(
|
|
'network_foo',
|
|
driver='macvlan',
|
|
gateway='192.168.1.1',
|
|
subnet='192.168.1.0/24',
|
|
driver_opts={'parent': 'eth1'},
|
|
containers=['abcd']
|
|
)
|
|
|
|
docker_disconnect_container_from_network.assert_called_with('abcd', 'network_foo')
|
|
docker_remove_network.assert_called_with('network_foo')
|
|
docker_create_network.assert_called_with('network_foo',
|
|
driver='macvlan',
|
|
driver_opts={'parent': 'eth1'},
|
|
gateway='192.168.1.1',
|
|
ip_range=None,
|
|
subnet='192.168.1.0/24')
|
|
docker_connect_container_to_network.assert_called_with('abcd', 'network_foo')
|
|
|
|
self.assertEqual(ret, {'name': 'network_foo',
|
|
'comment': 'Network \'network_foo\' was replaced with updated config',
|
|
'changes': {
|
|
'updated': {'network_foo': {
|
|
'old': {
|
|
'driver_opts': {'parent': 'eth0'},
|
|
'gateway': '192.168.0.1',
|
|
'subnet': '192.168.0.0/24'
|
|
},
|
|
'new': {
|
|
'driver_opts': {'parent': 'eth1'},
|
|
'gateway': '192.168.1.1',
|
|
'subnet': '192.168.1.0/24'
|
|
}
|
|
}},
|
|
'reconnected': ['container_bar']
|
|
},
|
|
'result': True})
|
|
|
|
def test_absent(self):
|
|
'''
|
|
Test docker_network.absent
|
|
'''
|
|
docker_remove_network = Mock(return_value='removed')
|
|
docker_disconnect_container_from_network = Mock(return_value='disconnected')
|
|
docker_networks = Mock(return_value=[{
|
|
'Name': 'network_foo',
|
|
'Containers': {'container': {}}
|
|
}])
|
|
__salt__ = {
|
|
'docker.remove_network': docker_remove_network,
|
|
'docker.disconnect_container_from_network': docker_disconnect_container_from_network,
|
|
'docker.networks': docker_networks,
|
|
}
|
|
with patch.dict(docker_state.__dict__,
|
|
{'__salt__': __salt__}):
|
|
ret = docker_state.absent('network_foo')
|
|
docker_disconnect_container_from_network.assert_called_with('container',
|
|
'network_foo')
|
|
docker_remove_network.assert_called_with('network_foo')
|
|
self.assertEqual(ret, {'name': 'network_foo',
|
|
'comment': '',
|
|
'changes': {'disconnected': 'disconnected',
|
|
'removed': 'removed'},
|
|
'result': True})
|
|
|
|
def test_absent_with_matching_network(self):
|
|
'''
|
|
Test docker_network.absent when the specified network does not exist,
|
|
but another network with a name which is a superset of the specified
|
|
name does exist. In this case we expect there to be no attempt to remove
|
|
any network.
|
|
Regression test for #41982.
|
|
'''
|
|
docker_remove_network = Mock(return_value='removed')
|
|
docker_disconnect_container_from_network = Mock(return_value='disconnected')
|
|
docker_networks = Mock(return_value=[{
|
|
'Name': 'network_foobar',
|
|
'Containers': {'container': {}}
|
|
}])
|
|
__salt__ = {
|
|
'docker.remove_network': docker_remove_network,
|
|
'docker.disconnect_container_from_network': docker_disconnect_container_from_network,
|
|
'docker.networks': docker_networks,
|
|
}
|
|
with patch.dict(docker_state.__dict__,
|
|
{'__salt__': __salt__}):
|
|
ret = docker_state.absent('network_foo')
|
|
docker_disconnect_container_from_network.assert_not_called()
|
|
docker_remove_network.assert_not_called()
|
|
self.assertEqual(ret, {'name': 'network_foo',
|
|
'comment': 'Network \'network_foo\' already absent',
|
|
'changes': {},
|
|
'result': True})
|