def setUp(self):
yield super(UnitContainerDeploymentTest, self).setUp()
self.juju_home = self.makeDir()
# Setup unit namespace
environ = dict(os.environ)
environ["JUJU_UNIT_NS"] = "ns1"
self.change_environment(**environ)
self.unit_deploy = UnitContainerDeployment(self.unit_name,
self.juju_home)
self.charm = get_charm_from_path(self.sample_dir1)
self.bundle = self.charm.as_bundle()
self.output = self.capture_logging("unit.deploy", level=logging.DEBUG)
def test_origin_usage(self):
"""The machine agent is started with a origin environment variable
"""
mock_container = self.mocker.patch(LXCContainer)
mock_container.is_constructed()
self.mocker.result(True)
mock_container.is_constructed()
self.mocker.result(True)
self.mocker.replay()
environ = dict(os.environ)
environ["JUJU_ORIGIN"] = "lp:~juju/foobar"
self.change_environment(**environ)
unit_deploy = UnitContainerDeployment(self.unit_name, self.juju_home)
container = yield unit_deploy._get_master_template(
"local", "127.0.0.1:1", "abc")
self.assertEqual(container.origin, "lp:~juju/foobar")
self.assertEqual(
container.customize_log,
os.path.join(self.juju_home, "units", "master-customize.log"))
