Ejemplo n.º 1
0
    def test_present_and_force(self):
        """
        According following sls,

        .. code-block:: yaml

            image:latest:
              docker_image.present:
                - force: true

        if ``image:latest`` is not downloaded and force is true
        should pull a new image successfully.
        """
        docker_inspect_image = MagicMock(return_value={"Id": "1234567890ab"})
        docker_pull = MagicMock(
            return_value={
                "Layers": {
                    "Pulled": ["abcdefghijkl"]
                },
                "Status": "Image 'image:latest' was pulled",
                "Time_Elapsed": 1.1,
            })
        docker_list_tags = MagicMock(side_effect=[[], ["image:latest"]])
        docker_resolve_tag = MagicMock(return_value="image:latest")
        __salt__ = {
            "docker.list_tags": docker_list_tags,
            "docker.pull": docker_pull,
            "docker.inspect_image": docker_inspect_image,
            "docker.resolve_tag": docker_resolve_tag,
        }
        with patch.dict(docker_state.__dict__, {"__salt__": __salt__}):
            ret = docker_state.present("image:latest", force=True)
        self.assertEqual(
            ret,
            {
                "changes": {
                    "Layers": {
                        "Pulled": ["abcdefghijkl"]
                    },
                    "Status": "Image 'image:latest' was pulled",
                    "Time_Elapsed": 1.1,
                },
                "result": True,
                "comment": "Image 'image:latest' was pulled",
                "name": "image:latest",
            },
        )
Ejemplo n.º 2
0
    def test_present_and_force(self):
        '''
        According following sls,

        .. code-block:: yaml

            image:latest:
              docker_image.present:
                - force: true

        if ``image:latest`` is not downloaded and force is true
        should pull a new image successfuly.
        '''
        docker_inspect_image = MagicMock(return_value={'Id': '1234567890ab'})
        docker_pull = MagicMock(
            return_value={
                'Layers': {
                    'Pulled': ['abcdefghijkl']
                },
                'Status': "Image 'image:latest' was pulled",
                'Time_Elapsed': 1.1
            })
        docker_list_tags = MagicMock(side_effect=[[], ['image:latest']])
        docker_resolve_tag = MagicMock(return_value='image:latest')
        __salt__ = {
            'docker.list_tags': docker_list_tags,
            'docker.pull': docker_pull,
            'docker.inspect_image': docker_inspect_image,
            'docker.resolve_tag': docker_resolve_tag
        }
        with patch.dict(docker_state.__dict__, {'__salt__': __salt__}):
            ret = docker_state.present('image:latest', force=True)
        self.assertEqual(
            ret, {
                'changes': {
                    'Layers': {
                        'Pulled': ['abcdefghijkl']
                    },
                    'Status': "Image 'image:latest' was pulled",
                    'Time_Elapsed': 1.1
                },
                'result': True,
                'comment': "Image 'image:latest' was pulled",
                'name': 'image:latest',
            })
Ejemplo n.º 3
0
    def test_present_already_local(self):
        """
        According following sls,

        .. code-block:: yaml

            image:latest:
              docker_image.present:
                - force: true

        if ``image:latest`` is already downloaded locally the state
        should not report changes.
        """
        docker_inspect_image = MagicMock(return_value={"Id": "abcdefghijkl"})
        docker_pull = MagicMock(
            return_value={
                "Layers": {
                    "Already_Pulled": ["abcdefghijkl"],
                    "Pulled": []
                },
                "Status": "Image is up to date for image:latest",
                "Time_Elapsed": 1.1,
            })
        docker_list_tags = MagicMock(return_value=["image:latest"])
        docker_resolve_tag = MagicMock(return_value="image:latest")
        __salt__ = {
            "docker.list_tags": docker_list_tags,
            "docker.pull": docker_pull,
            "docker.inspect_image": docker_inspect_image,
            "docker.resolve_tag": docker_resolve_tag,
        }
        with patch.dict(docker_state.__dict__, {"__salt__": __salt__}):
            ret = docker_state.present("image:latest", force=True)
            self.assertEqual(
                ret,
                {
                    "changes": {},
                    "result": True,
                    "comment": "Image 'image:latest' was pulled, "
                    "but there were no changes",
                    "name": "image:latest",
                },
            )
Ejemplo n.º 4
0
    def test_present_and_force(self):
        '''
        According following sls,

        .. code-block:: yaml

            image:latest:
              docker_image.present:
                - force: true

        if ``image:latest`` is not downloaded and force is true
        should pull a new image successfuly.
        '''
        docker_inspect_image = Mock(
            side_effect=CommandExecutionError(
                'Error 404: No such image/container: image:latest'))
        docker_pull = Mock(
            return_value={'Layers':
                          {'Already_Pulled': ['abcdefghijk'],
                           'Pulled': ['abcdefghijk']},
                          'Status': "Image 'image:latest' was pulled",
                          'Time_Elapsed': 1.1})
        docker_list_tags = Mock(
            side_effect=[[], ['image:latest']]
        )
        __salt__ = {'docker.list_tags': docker_list_tags,
                    'docker.pull': docker_pull,
                    'docker.inspect_image': docker_inspect_image,
                    }
        with patch.dict(docker_state.__dict__,
                        {'__salt__': __salt__}):
            ret = docker_state.present('image:latest', force=True)
        self.assertEqual(ret,
                         {'changes': {
                             'Layers': {'Already_Pulled': ['abcdefghijk'],
                                        'Pulled': ['abcdefghijk']},
                             'Status': "Image 'image:latest' was pulled",
                             'Time_Elapsed': 1.1},
                             'result': True,
                             'comment': "Image 'image:latest' was pulled",
                             'name': 'image:latest',
                         })
Ejemplo n.º 5
0
    def test_present_already_local(self):
        '''
        According following sls,

        .. code-block:: yaml

            image:latest:
              docker_image.present:
                - force: true

        if ``image:latest`` is already downloaded locally the state
        should not report changes.
        '''
        docker_inspect_image = MagicMock(return_value={'Id': 'abcdefghijkl'})
        docker_pull = MagicMock(
            return_value={
                'Layers': {
                    'Already_Pulled': ['abcdefghijkl'],
                    'Pulled': []
                },
                'Status': 'Image is up to date for image:latest',
                'Time_Elapsed': 1.1
            })
        docker_list_tags = MagicMock(return_value=['image:latest'])
        docker_resolve_tag = MagicMock(return_value='image:latest')
        __salt__ = {
            'docker.list_tags': docker_list_tags,
            'docker.pull': docker_pull,
            'docker.inspect_image': docker_inspect_image,
            'docker.resolve_tag': docker_resolve_tag
        }
        with patch.dict(docker_state.__dict__, {'__salt__': __salt__}):
            ret = docker_state.present('image:latest', force=True)
            self.assertEqual(
                ret, {
                    'changes': {},
                    'result': True,
                    'comment': "Image 'image:latest' was pulled, "
                    "but there were no changes",
                    'name': 'image:latest',
                })