mirror of https://github.com/docker/compose.git
32 lines
754 B
Python
32 lines
754 B
Python
from docker import Client
|
|
from plum.service import Service
|
|
import os
|
|
from unittest import TestCase
|
|
|
|
|
|
class DockerClientTestCase(TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
if os.environ.get('DOCKER_URL'):
|
|
cls.client = Client(os.environ['DOCKER_URL'])
|
|
else:
|
|
cls.client = Client()
|
|
cls.client.pull('ubuntu')
|
|
|
|
def setUp(self):
|
|
for c in self.client.containers(all=True):
|
|
self.client.kill(c['Id'])
|
|
self.client.remove_container(c['Id'])
|
|
|
|
def create_service(self, name, **kwargs):
|
|
return Service(
|
|
name=name,
|
|
client=self.client,
|
|
image="ubuntu",
|
|
command=["/bin/sleep", "300"],
|
|
**kwargs
|
|
)
|
|
|
|
|
|
|