DockerSensorTestCase

Complexity

Total Complexity

1

Size/Duplication

Total Lines	29
Duplicated Lines	0 %

1 Method

Rating	Name	Duplication	Size	Complexity
B	test_poll()	0	26	1

import mock

from st2tests.base import BaseSensorTestCase

from docker_container_sensor import DockerSensor

MOCK_CONTAINER_DATA = {
    "Id": "8dfafdbc3a40",
    "Names": ["/boring_feynman"],
    "Image": "ubuntu:latest",
    "Command": "echo 1",
    "Created": 1367854155,
    "Status": "Exit 0",
    "Ports": [{"PrivatePort": 2222, "PublicPort": 3333, "Type": "tcp"}],
    "SizeRw": 12288,
    "SizeRootFs": 0
}


class DockerSensorTestCase(BaseSensorTestCase):
    sensor_cls = DockerSensor

    def test_poll(self):
        sensor = self.get_sensor_instance()

        # No existing and no running containers (e.g. after initial sensor poll)
        sensor._running_containers = {}
        sensor._get_active_containers = mock.Mock()
        sensor._get_active_containers.return_value = {}

        # Initial poll, no containers
        sensor.poll()
        self.assertEqual(self.get_dispatched_triggers(), [])

        # One container started
        sensor._get_active_containers.return_value = {'1': MOCK_CONTAINER_DATA}

        sensor.poll()
        self.assertEqual(len(self.get_dispatched_triggers()), 1)
        self.assertTriggerDispatched(trigger='docker.container_tracker.started',
                                     payload={'container_info': MOCK_CONTAINER_DATA})

        # One container stopped
        sensor._get_active_containers.return_value = {}
        sensor.poll()
        self.assertEqual(len(self.get_dispatched_triggers()), 2)
        self.assertTriggerDispatched(trigger='docker.container_tracker.stopped',
                                     payload={'container_info': MOCK_CONTAINER_DATA})