def setUp(self): """Registers the dummy modules and recipe to be used in tests.""" modules_manager.ModulesManager.RegisterModules( [modules.DummyModule1, modules.DummyModule2]) self._recipe = resources.Recipe(test_recipe.__doc__, test_recipe.contents, test_recipe.args) self._recipes_manager = recipes_manager.RecipesManager() self._recipes_manager.RegisterRecipe(self._recipe)
def testReadRecipeFromFileObject(self): """Tests the _ReadRecipeFromFileObject function.""" test_manager = manager.RecipesManager() recipe = test_manager._ReadRecipeFromFileObject(io.StringIO(self._JSON)) self.assertIsNotNone(recipe) self.assertEqual(recipe.name, 'test') self.assertEqual(recipe.description, 'test recipe') self.assertEqual(recipe.contents['modules'][0]['name'], 'TestModule') self.assertEqual(len(recipe.args), 1)
def testRegisterRecipes(self): """Tests the RegisterRecipes function.""" test_manager = manager.RecipesManager() test_recipe = TestRecipe() number_of_recipes = len(test_manager._recipes) test_manager.RegisterRecipes([test_recipe]) self.assertEqual(len(test_manager._recipes), number_of_recipes + 1) test_manager.DeregisterRecipe(test_recipe)
def testGetRecipes(self): """Tests the GetRecipes function.""" test_manager = manager.RecipesManager() test_recipe = TestRecipe() test_manager.RegisterRecipe(test_recipe) registered_recipes = test_manager.GetRecipes() self.assertEqual(registered_recipes, [test_recipe]) test_manager.DeregisterRecipe(test_recipe)
def testRegisterRecipes(self): """Tests the RegisterRecipes function.""" test_manager = manager.RecipesManager() test_recipe = TestRecipe() # pylint: disable=protected-access number_of_recipes = len(test_manager._recipes) test_manager.RegisterRecipes([test_recipe]) self.assertEqual(len(test_manager._recipes), number_of_recipes + 1) test_manager.DeregisterRecipe(test_recipe)
def testRecipeRegistration(self): """Tests the RegisterRecipe and DeregisterRecipe functions.""" test_manager = manager.RecipesManager() test_recipe = TestRecipe() number_of_recipes = len(test_manager._recipes) test_manager.RegisterRecipe(test_recipe) self.assertEqual(len(test_manager._recipes), number_of_recipes + 1) with self.assertRaises(KeyError): test_manager.RegisterRecipe(test_recipe) test_manager.DeregisterRecipe(test_recipe) self.assertEqual(len(test_manager._recipes), number_of_recipes)