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)
