def test_master_task_entries(self, mock_master_seeds): def side_effect(): yield "required" yield "minimal" self.write_ubuntu_structure() output = GerminateOutput(self.config, self.temp_dir) self.config["DIST"] = "raring" mock_master_seeds.side_effect = side_effect self.assertEqual([ "#include <ubuntu/raring/required>", "#include <ubuntu/raring/minimal>", ], list(output.master_task_entries("ubuntu")))
def test_master_task_entries_no_seeds(self, mock_master_seeds): self.write_ubuntu_structure() output = GerminateOutput(self.config, self.temp_dir) self.config["DIST"] = "raring" self.assertRaises( NoMasterSeeds, list, output.master_task_entries("ubuntu"))