def test_app_is_running(service_is_running):
"""Test that checking whether app is running works."""
daemon1 = Daemon('test-daemon-1', 'test-unit-1')
daemon2 = FollowerComponent('test-daemon-2', 'test-unit-2')
daemon2.is_running = Mock()
follower1 = FollowerComponent('test-follower-1')
class TestApp(App):
"""Test app"""
app_id = 'test-app'
app = TestApp()
app.add(daemon1)
app.add(daemon2)
app.add(follower1)
service_is_running.return_value = True
daemon2.is_running.return_value = False
assert not app_is_running(app)
service_is_running.return_value = False
daemon2.is_running.return_value = False
assert not app_is_running(app)
service_is_running.return_value = True
daemon2.is_running.return_value = True
assert app_is_running(app)
def fixture_app_with_components():
"""Setup an app with some components."""
app = AppTest()
app.add(FollowerComponent('test-follower-1'))
app.add(FollowerComponent('test-follower-2'))
app.add(LeaderTest('test-leader-1'))
app.add(LeaderTest('test-leader-2'))
return app
def test_follower_component_initialization():
"""Test that follower component is initialized properly."""
component = FollowerComponent('test-follower-1')
assert not component.is_enabled()
component = FollowerComponent('test-follower-2', False)
assert not component.is_enabled()
component = FollowerComponent('test-follower-3', True)
assert component.is_enabled()
def test_follower_component_set_enabled():
"""Test setting internal enabled state a follower component."""
component = FollowerComponent('test-follower-1', False)
component.set_enabled(True)
assert component.is_enabled()
component.set_enabled(False)
assert not component.is_enabled()
def test_component_has_diagnostics():
"""Test checking if component has diagnostics implemented."""
component = LeaderTest('test-leader-1')
assert component.has_diagnostics()
component = FollowerComponent('test-follower-1')
assert not component.has_diagnostics()
def test_follower_component_disable():
"""Test disabling a follower component."""
component = FollowerComponent('test-follower-1', True)
component.disable()
assert not component.is_enabled()
def test_follower_component_enable():
"""Test enabling a follower component."""
component = FollowerComponent('test-follower-1', False)
component.enable()
assert component.is_enabled()