def test_len_cenarios(stub_api, data_inicial, data_fim, resultado):
# colocando como retorno range com o número informado
# para que len(self._get_dados) retorne o esperado
stub_api.get_data.return_value = range(resultado)
cenario = CopomCalendar(data_inicial, data_fim, stub_api)
eventos = len(cenario)
assert eventos == resultado
def test_connection_error(stub_api_erro_requests):
stub_api_erro_requests.get_data.side_effect = ConnectionError
data_inicial = date(2021, 4, 15)
data_fim = date(2021, 6, 14)
caso = CopomCalendar(data_inicial, data_fim, stub_api_erro_requests)
with pytest.raises(ConnectionError):
len(caso)
def base_case():
data_inicial = date(2021, 4, 15)
data_fim = date(2021, 6, 14)
return CopomCalendar(data_inicial, data_fim)
def test_fim_agenda_getter(data_inicial, data_fim, resultado):
cenario = CopomCalendar(data_inicial, data_fim)
fim_agenda = cenario.fim_agenda
assert fim_agenda == resultado
def test_inicio_agenda_setter(data_inicial, data_fim, resultado):
cenario = CopomCalendar(data_inicial, data_fim)
cenario.inicio_agenda = resultado
inicio_agenda = cenario.inicio_agenda
assert inicio_agenda == resultado
def test_has_events_cenarios(stub_api, data_inicial, data_fim, resultado):
stub_api.get_data.return_value = range(resultado)
cenario = CopomCalendar(data_inicial, data_fim, stub_api)
flag = cenario.has_new_events
assert flag == resultado
def test_has_events_as_dict_base_case(stub_api):
stub_api.get_data.return_value = [{'1': 1}, {'2': 2}, {'3': 3}]
cenario = CopomCalendar(date(2021, 4, 15), date(2021, 6, 14), stub_api)
for event in cenario:
assert isinstance(event, dict)
def test_len_raises_type_error(data_inicial, data_fim):
with pytest.raises(TypeError):
cenario = CopomCalendar(data_inicial, data_fim)
len(cenario)
def test_has_events_raises_type_error(data_inicial, data_fim):
with pytest.raises(TypeError):
cenario = CopomCalendar(data_inicial, data_fim)
cenario.has_new_events