def test_insert_remove(): model = MongoModel() model.insert({'a':1,'b':2}) data = model.query({'a':1}) assert data['b'] == 2 model.delete({'a':1}) data = model.query({'a':1}) assert not data
def setup_test_update(): mongo = MongoModel(project='test_api_update',collection='test_api_entry') mongo.insert({'a':1}) user = User() user.create('test_user_api_update','test_pass','*****@*****.**') project = Project() project.create('test api update','scraped') project.add_entry('test api entry','entry','localhost') user.add_project(project.get_id())
def test_delete(): user = User() user.login('test_user','test_pass') api_key = user.user.auth_token mongo = MongoModel(project='scraped',collection='entry') client = webapp.app.test_client() mongo.insert({'a':1}) data = mongo.query({'a':1}) id = str(data['_id']) project = Project() project.find('scraped') entry_id = project.find_entry('entry') url = '/api/db/%s/%s/%s/?api_key=%s' % (project.project.id,entry_id,id,api_key) response = client.delete(url) status = json.loads(response.data) assert status['status'] check = mongo.query({'_id':objectid.ObjectId(id)}) assert not check
def setup_test_update(): model = MongoModel() model.insert({'a':1,'b':2})