def test3_delete_tags(self): tags = Tags(mongodb) tags.add(self.inserted_id, 'ivan', '1') tags.add(self.inserted_id, 'ivan', '2') tags.add(self.inserted_id, 'ajna', '3') sucesso = tags.delete(self.inserted_id, 'ajna', '3') assert sucesso is True lista_tags = tags.list_usuario(self.inserted_id, 'ajna') assert lista_tags == []
def test1_add_tag(self): tags = Tags(mongodb) assert self.inserted_id is not None tags.add(self.inserted_id, 'ivan', '1') tags.add(self.inserted_id, 'ivan', '2') tags.add(self.inserted_id, 'ajna', '3') lista_tags = tags.list(self.inserted_id) assert type(lista_tags) is list assert len(lista_tags) == 3
def test2_list_tags_usuario(self): tags = Tags(mongodb) tags.add(self.inserted_id, 'ivan', '1') tags.add(self.inserted_id, 'ivan', '2') tags.add(self.inserted_id, 'ajna', '3') lista_tags = tags.list_usuario(self.inserted_id, 'ivan') assert len(lista_tags) == 2 lista_tags = tags.list_usuario(self.inserted_id, 'ajna') assert len(lista_tags) == 1 assert lista_tags == [{ 'tag': '3', 'usuario': 'ajna', 'descricao': 'Auditando' }]
def tag_add(): """Função para inserção de tag na imagem Faz update no fs.files, inserindo em um array com o nome do usuário ativo e a tag passada. Args: _id: ObjectId do arquivo tag: String (app usa lista de códigos com tupla (id, desc)) Returns: json['success']: True ou False """ _id = request.values.get('_id') tag = request.values.get('tag') data = {'success': False} try: db = app.config['mongodb'] tags = Tags(db) data['success'] = tags.add(_id=ObjectId(_id), usuario=current_user.id, tag=tag) data['tags'] = tags.list(ObjectId(_id)) except Exception as err: logger.error(err, exc_info=True) data['error'] = str(err) # raise return jsonify(data)
def test4_tagged(self): inserted_id2 = mongodb['fs.files'].insert_one({ 'teste': True }).inserted_id try: tags = Tags(mongodb) tags.add(inserted_id2, 'ivan', '1') tags.add(inserted_id2, 'ivan', '2') tags.add(inserted_id2, 'ajna', '1') tags.add(self.inserted_id, '', '1') lista_tags = list(tags.tagged(usuario='space ghost')) assert lista_tags == [] lista_tags = list(tags.tagged(tag='42')) assert lista_tags == [] lista_tags = list(tags.tagged(tag='1')) assert len(lista_tags) == 2 lista_tags = list(tags.tagged(tag='2')) assert len(lista_tags) == 1 finally: mongodb['fs.files'].delete_one({'_id': ObjectId(inserted_id2)})