def setUp(self):
yield super(ControlResolvedTest, self).setUp()
config = {"environments": {"firstenv": {"type": "dummy"}}}
self.write_config(dump(config))
self.config.load()
yield self.add_relation_state("wordpress", "mysql")
yield self.add_relation_state("wordpress", "varnish")
self.service1 = yield self.service_state_manager.get_service_state(
"mysql")
self.service_unit1 = yield self.service1.add_unit_state()
self.service_unit2 = yield self.service1.add_unit_state()
self.unit1_workflow = UnitWorkflowState(self.client,
self.service_unit1, None,
self.makeDir())
yield self.unit1_workflow.set_state("started")
self.environment = self.config.get_default()
self.provider = self.environment.get_machine_provider()
self.output = self.capture_logging()
self.stderr = self.capture_stream("stderr")
self.executor = HookExecutor()
def test_resolved_already_resolved(self):
"""
'juju resolved <unit_name>' will report if
the unit is already resolved.
"""
# Mark the unit as resolved and as in an error state.
yield self.service_unit1.set_resolved(RETRY_HOOKS)
yield self.unit1_workflow.set_state("start_error")
unit2_workflow = UnitWorkflowState(self.client, self.service_unit1,
None, self.makeDir())
unit2_workflow.set_state("start_error")
self.assertEqual((yield self.service_unit2.get_resolved()), None)
self.setup_exit(0)
finished = self.setup_cli_reactor()
self.mocker.replay()
main(["resolved", "mysql/0"])
yield finished
self.assertEqual((yield self.service_unit1.get_resolved()),
{"retry": RETRY_HOOKS})
self.assertNotIn("Marked unit 'mysql/0' as resolved",
self.output.getvalue())
self.assertIn("Service unit 'mysql/0' is already marked as resolved.",
self.stderr.getvalue(), "")
def setUp(self):
yield super(RemoteUpgradeCharmTest, self).setUp()
config = {
"environments": {"firstenv": {"type": "dummy"}}}
self.write_config(dump(config))
self.config.load()
charm = CharmDirectory(os.path.join(
test_repository_path, "series", "mysql"))
self.charm_state_manager.add_charm_state(
"cs:series/mysql-1", charm, "")
self.service_state1 = yield self.add_service_from_charm(
"mysql", "cs:series/mysql-1")
self.service_unit1 = yield self.service_state1.add_unit_state()
self.unit1_workflow = UnitWorkflowState(
self.client, self.service_unit1, None, self.makeDir())
yield self.unit1_workflow.set_state("started")
self.environment = self.config.get_default()
self.provider = self.environment.get_machine_provider()
self.output = self.capture_logging()
self.stderr = self.capture_stream("stderr")
def start(self):
"""Start the unit agent process."""
self.service_state_manager = ServiceStateManager(self.client)
self.executor.start()
# Retrieve our unit and configure working directories.
service_name = self.unit_name.split("/")[0]
service_state = yield self.service_state_manager.get_service_state(
service_name)
self.unit_state = yield service_state.get_unit_state(self.unit_name)
self.unit_directory = os.path.join(
self.config["juju_directory"], "units",
self.unit_state.unit_name.replace("/", "-"))
self.state_directory = os.path.join(self.config["juju_directory"],
"state")
# Setup the server portion of the cli api exposed to hooks.
from twisted.internet import reactor
self.api_socket = reactor.listenUNIX(
os.path.join(self.unit_directory, HOOK_SOCKET_FILE),
self.api_factory)
# Setup the unit state's address
address = yield get_unit_address(self.client)
yield self.unit_state.set_public_address(
(yield address.get_public_address()))
yield self.unit_state.set_private_address(
(yield address.get_private_address()))
# Inform the system, we're alive.
yield self.unit_state.connect_agent()
self.lifecycle = UnitLifecycle(self.client, self.unit_state,
service_state, self.unit_directory,
self.executor)
self.workflow = UnitWorkflowState(self.client, self.unit_state,
self.lifecycle, self.state_directory)
if self.get_watch_enabled():
yield self.unit_state.watch_resolved(self.cb_watch_resolved)
yield self.unit_state.watch_hook_debug(self.cb_watch_hook_debug)
yield service_state.watch_config_state(
self.cb_watch_config_changed)
# Fire initial transitions, only if successful
if (yield self.workflow.transition_state("installed")):
yield self.workflow.transition_state("started")
# Upgrade can only be processed if we're in a running state so
# for case of a newly started unit, do it after the unit is started.
if self.get_watch_enabled():
yield self.unit_state.watch_upgrade_flag(
self.cb_watch_upgrade_flag)
def setUp(self):
yield super(UnitWorkflowTest, self).setUp()
yield self.setup_default_test_relation()
self.lifecycle = UnitLifecycle(self.client, self.states["unit"],
self.states["service"],
self.unit_directory, self.executor)
self.juju_directory = self.makeDir()
self.state_directory = self.makeDir(
path=os.path.join(self.juju_directory, "state"))
self.workflow = UnitWorkflowState(self.client, self.states["unit"],
self.lifecycle, self.state_directory)
def setUp(self):
yield super(ControlCharmUpgradeTest, self).setUp()
config = {
"environments": {"firstenv": {"type": "dummy"}}}
self.write_config(dump(config))
self.config.load()
self.service_state1 = yield self.add_service_from_charm("mysql")
self.service_unit1 = yield self.service_state1.add_unit_state()
self.unit1_workflow = UnitWorkflowState(
self.client, self.service_unit1, None, self.makeDir())
yield self.unit1_workflow.set_state("started")
self.environment = self.config.get_default()
self.provider = self.environment.get_machine_provider()
self.output = self.capture_logging()
self.stderr = self.capture_stream("stderr")
def test_resolved_already_running(self):
"""
'juju resolved <unit_name>' will report if
the unit is already running.
"""
# Just verify we don't accidentally mark up another unit of the service
unit2_workflow = UnitWorkflowState(self.client, self.service_unit2,
None, self.makeDir())
unit2_workflow.set_state("start_error")
self.setup_exit(0)
finished = self.setup_cli_reactor()
self.mocker.replay()
main(["resolved", "mysql/0"])
yield finished
self.assertEqual((yield self.service_unit2.get_resolved()), None)
self.assertEqual((yield self.service_unit1.get_resolved()), None)
self.assertNotIn("Unit 'mysql/0 already running: started",
self.output.getvalue())
