def setUpClass(cls):
super(MoviesTest, cls).setUpClass()
cls._clean_movies_and_actors()
with admin(cls.app.application) as client:
client[cls.api].actors.insert_many(cls.actors)
cls.actors = [fix_id(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_many(cls.movies)
cls.movies = [fix_id(movie) for movie in cls.movies]
def setUpClass(cls):
super(MoviesTest, cls).setUpClass()
cls._clean_movies_and_actors()
with admin(cls.app.application) as client:
client[cls.api].actors.insert_many(cls.actors)
cls.actors = [fix_id(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_many(cls.movies)
cls.movies = [fix_id(movie) for movie in cls.movies]
def setUpClass(cls, auth=False):
super(TestAutoApiDumps, cls).setUpClass()
cls.collection = 'movies_with_data'
cls.now = datetime.datetime(2017, 6, 18, 17, 27, 0)
cls.oid = ObjectId()
with admin(cls.app.application) as client:
cls.oid_oid = client[cls.api][cls.collection].insert_one({
'oid': cls.oid
}).inserted_id
cls.oid_date = client[cls.api][cls.collection].insert_one({
'date': cls.now
}).inserted_id
def setUpClass(cls, auth=False):
super(TestAutoApiDumps, cls).setUpClass()
cls.collection = 'movies_with_data'
cls.now = datetime.datetime(2017, 6, 18, 17, 27, 0)
cls.oid = ObjectId()
with admin(cls.app.application) as client:
cls.oid_oid = client[cls.api][cls.collection].insert_one({
'oid':
cls.oid
}).inserted_id
cls.oid_date = client[cls.api][cls.collection].insert_one({
'date':
cls.now
}).inserted_id
def mocked_delete_token(app, api, user, token):
with admin(app) as client:
client[api].command('updateUser', user, customData={})
from auto_api.mongodb import delete_token
delete_token(app, api, user, token)
def remove_user(cls, api, user):
with admin(cls.app.application) as client:
r = client[api].command('usersInfo', {'user': user, 'db': api})
if r['users']:
client[api].command('dropUser', user)
def _clean_movies_and_actors(cls):
with admin(cls.app.application) as client:
client[cls.api].actors.drop()
client[cls.api].movies.drop()
client[cls.api].stars.drop()
def mocked_delete_token(app, api, user, token):
with admin(app) as client:
client[api].command('updateUser', user, customData={})
from auto_api.mongodb import delete_token
delete_token(app, api, user, token)
def tearDownClass(cls):
with admin(cls.app.application) as client:
client[cls.api][cls.collection].drop()
super(TestAutoApiDumps, cls).tearDownClass()
def remove_user(cls, api, user):
with admin(cls.app.application) as client:
r = client[api].command('usersInfo', {'user': user, 'db': api})
if r['users']:
client[api].command('dropUser', user)
def _clean_movies_and_actors(cls):
with admin(cls.app.application) as client:
client[cls.api].actors.drop()
client[cls.api].movies.drop()
client[cls.api].stars.drop()
def tearDownClass(cls):
with admin(cls.app.application) as client:
client[cls.api][cls.collection].drop()
super(TestAutoApiDumps, cls).tearDownClass()