docker-compose/tests/unit/cli/docker_client_test.py

from __future__ import absolute_import
from __future__ import unicode_literals

import os

import mock

from compose.cli import docker_client
from tests import unittest


class DockerClientTestCase(unittest.TestCase):

    def test_docker_client_no_home(self):
        with mock.patch.dict(os.environ):
            del os.environ['HOME']
            docker_client.docker_client()

    def test_docker_client_with_custom_timeout(self):
        with mock.patch.dict(os.environ):
            os.environ['DOCKER_CLIENT_TIMEOUT'] = timeout = "300"
            client = docker_client.docker_client()
        self.assertEqual(client.timeout, int(timeout))