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()
def clean(api, collection):
with _admin_manager_client(app) as client:
client[api][collection].drop()
