mirror of https://github.com/docker/compose.git
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
from .. import mock
|
|
from .testcases import DockerClientTestCase
|
|
from compose.config.types import VolumeSpec
|
|
from compose.project import Project
|
|
from compose.service import ConvergenceStrategy
|
|
|
|
|
|
class ResilienceTest(DockerClientTestCase):
|
|
def setUp(self):
|
|
self.db = self.create_service(
|
|
'db',
|
|
volumes=[VolumeSpec.parse('/var/db')],
|
|
command='top')
|
|
self.project = Project('composetest', [self.db], self.client)
|
|
|
|
container = self.db.create_container()
|
|
self.db.start_container(container)
|
|
self.host_path = container.get_mount('/var/db')['Source']
|
|
|
|
def tearDown(self):
|
|
del self.project
|
|
del self.db
|
|
super(ResilienceTest, self).tearDown()
|
|
|
|
def test_successful_recreate(self):
|
|
self.project.up(strategy=ConvergenceStrategy.always)
|
|
container = self.db.containers()[0]
|
|
assert container.get_mount('/var/db')['Source'] == self.host_path
|
|
|
|
def test_create_failure(self):
|
|
with mock.patch('compose.service.Service.create_container', crash):
|
|
with pytest.raises(Crash):
|
|
self.project.up(strategy=ConvergenceStrategy.always)
|
|
|
|
self.project.up()
|
|
container = self.db.containers()[0]
|
|
assert container.get_mount('/var/db')['Source'] == self.host_path
|
|
|
|
def test_start_failure(self):
|
|
with mock.patch('compose.service.Service.start_container', crash):
|
|
with pytest.raises(Crash):
|
|
self.project.up(strategy=ConvergenceStrategy.always)
|
|
|
|
self.project.up()
|
|
container = self.db.containers()[0]
|
|
assert container.get_mount('/var/db')['Source'] == self.host_path
|
|
|
|
|
|
class Crash(Exception):
|
|
pass
|
|
|
|
|
|
def crash(*args, **kwargs):
|
|
raise Crash()
|