def test_simple_models(): expected_models = [ 'Accounting', 'AdmissionRule', 'AssignedResource', 'Challenge', 'EventLog', 'EventLogHostname', 'File', 'FragJob', 'GanttJobsPrediction', 'GanttJobsPredictionsLog', 'GanttJobsPredictionsVisu', 'GanttJobsResource', 'GanttJobsResourcesLog', 'GanttJobsResourcesVisu', 'Job', 'JobDependencie', 'JobResourceDescription', 'JobResourceGroup', 'JobStateLog', 'JobType', 'MoldableJobDescription', 'Queue', 'Resource', 'ResourceLog', 'Scheduler' ] assert set(list(db.models.keys())) == set(expected_models) assert set(list(dict(all_models()).keys())) == set(expected_models) # len(tables) = len(Models) + table schema assert len(dict(all_tables()).keys()) == len(expected_models) + 1
def current_models(self): """ Return a namespace with all mapping classes""" from oar.lib.models import all_models # avoid a circular import return dict(all_models())