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