diff --git a/stalker/db/__init__.py b/stalker/db/__init__.py index f5ae661..7f468e9 100644 --- a/stalker/db/__init__.py +++ b/stalker/db/__init__.py @@ -94,12 +94,13 @@ def init(): # register all Actions available for all SOM classes class_names = [ - 'Asset', 'TimeLog', 'Client', 'Daily', 'Department', 'Entity', - 'FilenameTemplate', 'Group', 'ImageFormat', 'Link', 'Message', 'Note', - 'Page', 'Permission', 'Project', 'Repository', 'Review', 'Scene', + 'Asset', 'Client', 'Daily', 'Department', 'Entity', 'FilenameTemplate', + 'Group', 'ImageFormat', 'Link', 'Message', 'Note', 'Page', + 'Permission', 'Project', 'Repository', 'Review', 'Role', 'Scene', 'Sequence', 'Shot', 'SimpleEntity', 'Status', 'StatusList', - 'Structure', 'Studio', 'Tag', 'Task', 'Ticket', 'TicketLog', 'Type', - 'User', 'Vacation', 'Version'] + 'Structure', 'Studio', 'Tag', 'Task', 'Ticket', 'TicketLog', 'TimeLog', + 'Type', 'User', 'Vacation', 'Version' + ] for class_name in class_names: _temp = __import__( diff --git a/tests/db/test_db.py b/tests/db/test_db.py index 25dbad1..d8f3a34 100644 --- a/tests/db/test_db.py +++ b/tests/db/test_db.py @@ -292,10 +292,10 @@ def test_permissions_created_for_all_the_classes(self): class_names = [ 'Asset', 'Client', 'Group', 'Permission', 'User', 'Department', 'SimpleEntity', 'Entity', 'ImageFormat', 'Link', 'Message', 'Note', - 'Page', 'Project', 'Repository', 'Review', 'Scene', 'Sequence', - 'Shot', 'Status', 'StatusList', 'Structure', 'Studio', 'Tag', - 'TimeLog', 'Task', 'FilenameTemplate', 'Ticket', 'TicketLog', - 'Type', 'Vacation', 'Version', 'Daily' + 'Page', 'Project', 'Repository', 'Review', 'Role', 'Scene', + 'Sequence', 'Shot', 'Status', 'StatusList', 'Structure', 'Studio', + 'Tag', 'TimeLog', 'Task', 'FilenameTemplate', 'Ticket', + 'TicketLog', 'Type', 'Vacation', 'Version', 'Daily' ] permission_db = Permission.query.all() @@ -337,7 +337,7 @@ def test_permissions_not_created_over_and_over_again(self): # and we still have correct amount of Permissions permissions = Permission.query.all() - self.assertEqual(len(permissions), 330) + self.assertEqual(len(permissions), 340) # clean the test shutil.rmtree(temp_db_path)