Moved tests to root directory

tests/fixtures/simple-dockerfile/Dockerfile

@@ -0,0 +1,2 @@
+FROM ubuntu
+CMD echo "success"

tests/service_collection_test.py

@@ -0,0 +1,62 @@
+from plum.service import Service
+from plum.service_collection import ServiceCollection
+from .testcases import DockerClientTestCase
+
+
+class ServiceCollectionTest(DockerClientTestCase):
+    def test_from_dict(self):
+        collection = ServiceCollection.from_dicts(None, [
+            {
+                'name': 'web',
+                'image': 'ubuntu'
+            },
+            {
+                'name': 'db',
+                'image': 'ubuntu'
+            }
+        ])
+        self.assertEqual(len(collection), 2)
+        self.assertEqual(collection.get('web').name, 'web')
+        self.assertEqual(collection.get('web').options['image'], 'ubuntu')
+        self.assertEqual(collection.get('db').name, 'db')
+        self.assertEqual(collection.get('db').options['image'], 'ubuntu')
+
+    def test_from_dict_sorts_in_dependency_order(self):
+        collection = ServiceCollection.from_dicts(None, [
+            {
+                'name': 'web',
+                'image': 'ubuntu',
+                'links': ['db'],
+            },
+            {
+                'name': 'db',
+                'image': 'ubuntu'
+            }
+        ])
+
+        self.assertEqual(collection[0].name, 'db')
+        self.assertEqual(collection[1].name, 'web')
+
+    def test_get(self):
+        web = self.create_service('web')
+        collection = ServiceCollection([web])
+        self.assertEqual(collection.get('web'), web)
+
+    def test_start_stop(self):
+        collection = ServiceCollection([
+            self.create_service('web'),
+            self.create_service('db'),
+        ])
+
+        collection.start()
+
+        self.assertEqual(len(collection[0].containers), 1)
+        self.assertEqual(len(collection[1].containers), 1)
+
+        collection.stop()
+
+        self.assertEqual(len(collection[0].containers), 0)
+        self.assertEqual(len(collection[1].containers), 0)
+
+
+

tests/service_test.py

@@ -0,0 +1,91 @@
+from plum import Service
+from .testcases import DockerClientTestCase
+
+
+class NameTestCase(DockerClientTestCase):
+    def test_name_validations(self):
+        self.assertRaises(ValueError, lambda: Service(name=''))
+
+        self.assertRaises(ValueError, lambda: Service(name=' '))
+        self.assertRaises(ValueError, lambda: Service(name='/'))
+        self.assertRaises(ValueError, lambda: Service(name='!'))
+        self.assertRaises(ValueError, lambda: Service(name='\xe2'))
+
+        Service('a')
+        Service('foo')
+        Service('foo_bar')
+        Service('__foo_bar__')
+        Service('_')
+        Service('_____')
+
+    def test_containers(self):
+        foo = self.create_service('foo')
+        bar = self.create_service('bar')
+
+        foo.start()
+
+        self.assertEqual(len(foo.containers), 1)
+        self.assertEqual(foo.containers[0]['Names'], ['/foo_1'])
+        self.assertEqual(len(bar.containers), 0)
+
+        bar.scale(2)
+
+        self.assertEqual(len(foo.containers), 1)
+        self.assertEqual(len(bar.containers), 2)
+
+        names = [c['Names'] for c in bar.containers]
+        self.assertIn(['/bar_1'], names)
+        self.assertIn(['/bar_2'], names)
+
+    def test_up_scale_down(self):
+        service = self.create_service('scaling_test')
+        self.assertEqual(len(service.containers), 0)
+
+        service.start()
+        self.assertEqual(len(service.containers), 1)
+
+        service.start()
+        self.assertEqual(len(service.containers), 1)
+
+        service.scale(2)
+        self.assertEqual(len(service.containers), 2)
+
+        service.scale(1)
+        self.assertEqual(len(service.containers), 1)
+
+        service.stop()
+        self.assertEqual(len(service.containers), 0)
+
+        service.stop()
+        self.assertEqual(len(service.containers), 0)
+
+    def test_start_container_passes_through_options(self):
+        db = self.create_service('db')
+        db.start_container(environment={'FOO': 'BAR'})
+        self.assertEqual(db.inspect()[0]['Config']['Env'], ['FOO=BAR'])
+
+    def test_start_container_inherits_options_from_constructor(self):
+        db = self.create_service('db', environment={'FOO': 'BAR'})
+        db.start_container()
+        self.assertEqual(db.inspect()[0]['Config']['Env'], ['FOO=BAR'])
+
+    def test_start_container_creates_links(self):
+        db = self.create_service('db')
+        web = self.create_service('web', links=[db])
+        db.start_container()
+        web.start_container()
+        self.assertIn('/web_1/db_1', db.containers[0]['Names'])
+        db.stop()
+        web.stop()
+
+    def test_start_container_builds_images(self):
+        service = Service(
+            name='test',
+            client=self.client,
+            build='tests/fixtures/simple-dockerfile',
+        )
+        container = service.start()
+        self.client.wait(container)
+        self.assertIn('success', self.client.logs(container))
+        
+

tests/testcases.py

@@ -0,0 +1,31 @@
+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
+        )
+
+
+