def setUpClass(cls): super(MoviesTest, cls).setUpClass() cls._clean_movies_and_actors() with _admin_manager_client(cls.app.application) as client: client[cls.api].actors.insert(cls.actors) cls.actors = [format_result(actor) for actor in cls.actors] for movie in cls.movies: movie.update({'actors': cls.actors[cls.movies.index(movie)]['id']}) client[cls.api].movies.insert(cls.movies) cls.movies = [format_result(movie) for movie in cls.movies]
def clean(api, collection): with _admin_manager_client(app) as client: client[api][collection].drop()
def _clean_movies_and_actors(cls): with _admin_manager_client(cls.app.application) as client: client[cls.api].actors.drop() client[cls.api].movies.drop() client[cls.api].stars.drop()
def remove_user(cls, api, user): with _admin_manager_client(cls.app.application) as client: client[api].remove_user(user)
def add_user(cls, api, user, password, roles): with _admin_manager_client(cls.app.application) as client: client[api].add_user(user, password, customData={'roles': roles})
def tearDownClass(cls): with _admin_manager_client(cls.app.application) as client: client[cls.api].authenticate(cls.user, cls.password) client.drop_database(cls.api) super(BaseTest, cls).tearDownClass()