Fix menu test cases by not reseting the entire menu class cache,

but only specific temporary test menus.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2017-08-10 03:16:03 -04:00
parent 6ec235afc0
commit 31681c66a1
2 changed files with 7 additions and 3 deletions

View File

@@ -62,8 +62,8 @@ class Menu(object):
return cls._registry[name] return cls._registry[name]
@classmethod @classmethod
def reset(cls): def remove(cls, name):
cls._registry = {} del cls._registry[name]
def __init__(self, name, icon=None, label=None): def __init__(self, name, icon=None, label=None):
if name in self.__class__._registry: if name in self.__class__._registry:

View File

@@ -121,12 +121,16 @@ class MenuClassTestCase(GenericViewTestCase):
name=TEST_PERMISSION_NAME, label=TEST_PERMISSION_LABEL name=TEST_PERMISSION_NAME, label=TEST_PERMISSION_LABEL
) )
Menu.reset()
self.menu = Menu(name=TEST_MENU_NAME) self.menu = Menu(name=TEST_MENU_NAME)
self.sub_menu = Menu(name=TEST_SUBMENU_NAME) self.sub_menu = Menu(name=TEST_SUBMENU_NAME)
self.link = Link(text=TEST_LINK_TEXT, view=TEST_VIEW_NAME) self.link = Link(text=TEST_LINK_TEXT, view=TEST_VIEW_NAME)
Permission.invalidate_cache() Permission.invalidate_cache()
def tearDown(self):
Menu.remove(name=TEST_MENU_NAME)
Menu.remove(name=TEST_SUBMENU_NAME)
super(MenuClassTestCase, self).tearDown()
def test_null_source_link_unbinding(self): def test_null_source_link_unbinding(self):
self.menu.bind_links(links=(self.link,)) self.menu.bind_links(links=(self.link,))