compose/tests/testcases.py

from docker import Client
from fig.service import Service
from fig.cli.utils import docker_url
from unittest2 import TestCase


class DockerClientTestCase(TestCase):
    @classmethod
    def setUpClass(cls):
        cls.client = Client(docker_url())
        cls.client.pull('ubuntu')

    def setUp(self):
        for c in self.client.containers(all=True):
            if c['Names'] and 'figtest' in c['Names'][0]:
                self.client.kill(c['Id'])
                self.client.remove_container(c['Id'])

    def create_service(self, name, **kwargs):
        return Service(
            project='figtest',
            name=name,
            client=self.client,
            image="ubuntu",
            command=["/bin/sleep", "300"],
            **kwargs
        )