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 test_project_create(): # login user get token user = User() user.login('test_user','test_pass') api_key = user.user.auth_token # do a post url = '/api/project/?api_key=%s' % api_key # now check project in user test_client = webapp.app.test_client() data = {'name':'project create','description':'project content create'} result = test_client.post(url,data=json.dumps(data),content_type='application/json') status = json.loads(result.data) assert status['status'] project = ProjectList() test_user = User() test_user.login('test_user','test_pass') print test_user.user.project registered = False for i in project.all(): if i.get_id() in test_user.get_project(): registered = True assert registered, "project not in user project" exist = False for i in project.all(): if 'project create' == i.project.name: exist = True assert exist, "project created" # now delete it project = Project() model = MongoModel(project=project.project_,collection=project.collection_) model.delete({'name':'project create'})