def build_list_occurences(start_date, end_date, request):
""" Creates a list of occurrences
@param start_date Initial date.
@param end_data Final date.
@return List of occurrences
"""
try:
# DAO object from basic occurrence
occurrences_dao = OcorrenciaBasicaDAO()
# list of occurrences
occurrences_list = occurrences_dao.lista_ocorrencias_por_periodo(
start_date, end_date, _MAX_QUERIES)
except (MySQLdb.Error, ResultadoConsultaNuloError):
raise MySQLdb.Error(ResultadoConsultaNuloError)
return occurrences_list
class TestOcorrencia(DAO_Tests):
"""docstring for TestOcorrencia
Class that tests the methods from ocorrencia_basica_dao
"""
def test_existing_ocorrencia_dao_instance(self):
"""
Tests to see if the occurrence was correctly instantiated
"""
self.ocorrencia = OcorrenciaBasicaDAO()
self.assertIsNotNone(self.ocorrencia)
def test_ocorrencia_por_regiao(self):
"""
Tests is the number of occurrences were instantiated, for a defined
and an undefined limit.
Also verifies if the return for the occurrences are in the same state.
@brief Local variables
oco -
A list of occurences on a region.
"""
self.ocorrencia = OcorrenciaBasicaDAO()
_brasilia = 97012
oco = self.ocorrencia.lista_ocorrencias_por_regiao(_brasilia)
self.assertIsNotNone(oco)
oco = self.ocorrencia.lista_ocorrencias_por_regiao(_brasilia, limite=3)
self.assertIsNotNone(oco)
self.assertLess(len(oco), 4)
# verifica se todos os retornos estão no DF
for i in oco:
self.assertIsNotNone(str(i))
self.assertEqual(i.tmuuf, 'DF')
def test_ocorrencia_por_periodo(self):
"""
Tests is the number of occurrences were instantiated in a determined
period, for a defined and an undefined limit.
Also verifies if the return for the occurrences are in the same year.
@brief Local variables
oco -
A list of occurences on a period.
"""
self.ocorrencia = OcorrenciaBasicaDAO()
oco = self.ocorrencia.lista_ocorrencias_por_periodo(
'06/01/06',
'06/12/06'
)
self.assertIsNotNone(oco)
oco = self.ocorrencia.lista_ocorrencias_por_periodo(
'06/01/06', '06/12/06', limite=3)
self.assertIsNotNone(oco)
self.assertLess(len(oco), 4)
# verifica se as ocorrencias aconteceram em 2006
for i in oco:
self.assertEqual(2006, i.ocodataocorrencia.year)
