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})
