def test_update(): user = User() user.login('test_user_api_update','test_pass') api_key = user.user.auth_token project = Project() project.find('test api update') entry_id = project.find_entry('test api entry') print user.get_project() print project.get_id() mongo = MongoModel(project='test_api_update',collection='test_api_entry') data = mongo.query({'a':1}) print data id = str(data['_id']) updated = {'a':2} url = '/api/db/%s/%s/%s/?api_key=%s' % (project.project.id,entry_id,id,api_key) client = webapp.app.test_client() response = client.put(url, data = json.dumps(updated), content_type='application/json') print response.data status = json.loads(response.data) assert status['status'] updated_data = mongo.query({'_id':objectid.ObjectId(id)}) assert updated_data['a'] == 2
def setup_user_project_list(): user = User() user.create('test_user_list','test_pass','*****@*****.**') project = Project() project.create('project list 1','project content 1') user.add_project(project.get_id()) project = Project() project.create('project list 2','project content 2') user.add_project(project.get_id()) project = Project() project.create('project list 3','project content 3') user.add_project(project.get_id())
def setup_user(): user = User() user.create('test_user','test_pass','*****@*****.**') project = Project() project.create('scraped','scraped') project.add_entry('entry','entry','localhost') user.add_project(project.get_id())
def setup_project_update(): user = User() user.create('test_update_user','test_password','*****@*****.**') project = Project() project.create('project update','test project update') user.add_project(project.get_id()) project.add_entry('data','test data','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 setup_user_project(): # now create user user = User() user.create('test_user_update_project','test_pass','*****@*****.**') # create project project = Project() project.create('project update','project update content') # associate project user.add_project(project.get_id())
def test_user_project_withdraw(): user = User() user.login('test_user_list','test_pass') project = Project() project.find('project list 3') user.add_project(project.get_id()) api_key = user.user.auth_token test_client = webapp.app.test_client() url = '/api/project/%s/?api_key=%s' % (project.get_id(),api_key) data = {'project':'project_list_3','action':'withdraw'} result = test_client.put(url,data=json.dumps(data),content_type='application/json') print result.data user.login('test_user_list','test_pass') print user.user.project print project.get_id() print project.project.name assert not str(project.get_id()) in user.user.project
def test_project_delete(): # login user get token user = User() user.login('test_user_delete_api_project','test_pass') api_key = user.user.auth_token print api_key project = Project() project.find('project 1') project_id = project.get_id() # do a delete test_client = webapp.app.test_client() url = '/api/project/%s/?api_key=%s' % (project.project.id,api_key) result = test_client.delete(url) status = json.loads(result.data) # now check user don't exist project = Project() model = MongoModel(project=project.project_,collection=project.collection_) data = model.query({'name':'project 1'}) assert not data test_user = User() test_user.login('test_user_delete','test_pass') assert project_id not in test_user.user.project
def setup_project_delete(): user = User() user.create('test_user_delete_api_project','test_pass','*****@*****.**') project = Project() project.create('project 1','project content 1') user.add_project(project.get_id())