def test_drop_database(self, have_role_mock):
have_role_mock.return_value = True
db = model.mongo.db()
db.test_collection.insert({"name": "test thingy"})
response = self.app.get(url(controller='task', action='drop_database'))
assert daemon.job_running('drop_database'), \
"drop_database action didn't start the drop_database job!"
while daemon.job_running('drop_database'):
time.sleep(0.1)
h.assert_equal(db.collection_names(), ['system.js', 'system.indexes'])
def test_remove_dataset(self, have_role_mock):
have_role_mock.return_value = True
ds = model.Dataset(name='mydataset')
ds.save()
response = self.app.get(url(controller='task',
action='remove_dataset',
dataset='mydataset'))
assert daemon.job_running('remove_mydataset'), \
"remove_dataset action didn't start the remove_mydataset job!"
while daemon.job_running('remove_mydataset'):
time.sleep(0.1)
res = model.Dataset.find_one({'name': 'mydataset'})
h.assert_equal(res, None)
