from mock import patch

from .test_helper import raises

from kiwi.exceptions import KiwiContainerImageSetupError

from kiwi.container import ContainerImage


class TestContainerImage(object):
    @raises(KiwiContainerImageSetupError)
    def test_container_image_not_implemented(self):
        ContainerImage('foo', 'root_dir')

    @patch('kiwi.container.ContainerImageDocker')
    def test_container_image_docker(self, mock_docker):
        ContainerImage('docker', 'root_dir')
        mock_docker.assert_called_once_with('root_dir', None)

    @patch('kiwi.container.ContainerImageOCI')
    def test_container_image_oci(self, mock_oci):
        ContainerImage('oci', 'root_dir')
        mock_oci.assert_called_once_with('root_dir', None)