def test_api_add_child(self): Item('home').save() Item('fireflash').save() with app.test_client() as c: rv = self.get_json(c, '/api/item/home/') data = json.loads(rv.data) data['children'].append({'name':'fireflash'}) rv = self.put_json(c, '/api/item/home/', data) print rv assert rv.status_code == 200 fireflash = Item.find('fireflash') home = Item.find('home') assert fireflash.parents[0].name == 'home' assert home.children[0].name == 'fireflash'
def show_item(name): item = Item.find(name) if not item: abort(404) if request.method in ["POST", "DELETE"]: if "tag-add" in request.form: tag_name = request.form['tag-name'] tag = Tag.find(tag_name) item.add_to(tag) item.save() elif "tag-remove" in request.form: tag_name = request.form['tag-name'] tag = Tag.find(tag_name) item.remove_from(tag) item.save() elif "child-add" in request.form: child_name = request.form['child-name'] child_item = Item.find(child_name) item.add_child(child_item) item.save() elif "child-remove" in request.form: child_name = request.form['child-name'] child_item = Item.find(child_name) item.remove_child(child_item) item.save() elif "item-remove" in request.form: item.remove() return redirect(url_for("list_tags")) else: update_variables(item, request) var_list = [] for var in item.to_data()['vars']: v = var['value'] if type(v) in [str, int]: var['type'] = 'Text' elif type(v) == list: var['type'] = 'List' elif type(v) == dict: var['type'] = 'Map' var_list.append(var) tag_names = [tag.name for tag in item.tags] available_tags = [tag.name for tag in Tag.find_all() if tag.name not in tag_names] return render_template('item.html', item=item, var_list=var_list, available_tags=available_tags)
def show_item(name): item = Item.find(name) if not item: abort(404) if request.method == 'DELETE': item.remove() response = make_response('', 200) return response if request.method == 'PUT': if not request.json: abort(415) try: if request.json['name'] != item.name: abort(400) item.update_from(request.json) item.save() except MelangeException: abort(400) def item_url(item): return url_for('melange_api.show_item', name=item.name) def tag_url(tag): return url_for('melange_api.show_tag', name=tag.name) data = item.to_data(item_href=item_url, tag_href=tag_url) response = make_response( json.dumps(data), 200 ) response.headers['Content-Type'] = 'application/json' return response
def test_api_set_variable(self): Item('fireflash').save() with app.test_client() as c: rv = self.get_json(c, '/api/item/fireflash/') data = json.loads(rv.data) data['vars'].append({'key':'hello', 'value':'world'}) rv = self.put_json(c, '/api/item/fireflash/', data) print rv assert rv.status_code == 200 fireflash = Item.find('fireflash') assert fireflash.variables['hello'] == 'world'
def list_items(): if request.method == "POST": item_name = request.form["item-name"] item = Item.find(item_name) if not item: item = Item(item_name) if 'tag-name' in request.form: tag = Tag.find(request.form['tag-name']) if tag: item.add_to(tag) item.save() return redirect(url_for('show_item', name=item.name)) items = Item.find_all() return render_template('items.html', items=items)
def test_api_create_item(self): Tag('laptop').save() data = { 'name': 'fireflash', } with app.test_client() as c: rv = self.post_json(c, '/api/tag/laptop/', data) print rv assert rv.status_code == 201 fireflash = Item.find('fireflash') assert fireflash is not None assert fireflash.name == 'fireflash' assert len(fireflash.tags) == 1 assert fireflash.tags[0].name == 'laptop'
def test_api_update_variable(self): fireflash = Item('fireflash') fireflash.set_variable('test', 'one') fireflash.save() with app.test_client() as c: rv = self.get_json(c, '/api/item/fireflash/') data = json.loads(rv.data) data['vars'] = [{'key':'test', 'value':'two'}] rv = self.put_json(c, '/api/item/fireflash/', data) print rv assert rv.status_code == 200 fireflash = Item.find('fireflash') assert fireflash.variables['test'] == 'two'
def show_tag(name): tag = Tag.find(name) if not tag: abort(404) if request.method == 'POST': if not request.json: abort(415) data = request.json if not 'name' in data: abort(400) response_code = 200 item = Item.find(data['name']) if not item: item = Item(data['name']) response_code = 201 try: item.update_from(data) except MelangeException: abort(400) item.add_to(tag) item.save() response = make_response('', response_code) response.headers['Content-Location'] = url_for('melange_api.show_item',name=item.name) return response if request.method == 'DELETE': tag.remove() response = make_response('', 200) return response if request.method == 'PUT': if not request.json: abort(415) try: if request.json['name'] != tag.name: abort(400) tag.update_from(request.json) tag.save() except MelangeException: abort(400) def item_url(item): return url_for('melange_api.show_item', name=item.name) data = tag.to_data(item_href=item_url) response = make_response( json.dumps(data), 200 ) response.headers['Content-Type'] = 'application/json' return response
def test_api_set_tag_variable(self): fireflash = Item('fireflash') laptop = Tag('laptop') laptop.set_variable('test', 'one') fireflash.add_to(laptop) laptop.save() with app.test_client() as c: rv = self.get_json(c, '/api/tag/laptop/') data = json.loads(rv.data) data['vars']['test']= 'two' rv = self.put_json(c, '/api/tag/laptop/', data) print rv assert rv.status_code == 200 fireflash = Item.find('fireflash') laptop = Tag.find('laptop') assert laptop.variables['test'] == 'two' assert fireflash.get_all_variables()['test'] == 'two'