def test_sets_plan_on_deploy(self): config = Mock() config.get_deploy_log = Mock() deployer = Mock() env = Environment(config, "staging", deployer) env.deploy("EXAMPLE-PLAN", "EXAMPLE-PLAN-39") self.assertEquals("EXAMPLE-PLAN", env.get_plan())