def test1_add_ocorrencia(self):
ocorrencias = Ocorrencias(mongodb)
assert self.inserted_id is not None
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 1')
ocorrencias.add(self.inserted_id, 'ivan', 'Ocorrencia 2')
ocorrencias.add(self.inserted_id, 'ajna', 'Ocorrencia 3')
lista_ocorrencias = ocorrencias.list(self.inserted_id)
assert type(lista_ocorrencias) is list
assert len(lista_ocorrencias) == 3
def image_ocorrencia(_id, success=True):
"""Função para listar ocorrências na imagem
Args:
_id: ObjectId do arquivo
success: Falso se houve erro em operação anterior
Returns:
json['success']: True ou False
"""
data = {'success': success}
try:
db = app.config['mongodb']
ocorrencias = Ocorrencias(db)
data['ocorrencias'] = ocorrencias.list(ObjectId(_id))
except Exception as err:
logger.error(err, exc_info=True)
data['error'] = str(err)
data['success'] = False
# raise
return jsonify(data)