def test_get_flight_departures__with_begin_end_datetime__is_converted_to_int_and_flight_departures_object_is_returned(
):
flight_departures_dict_list, expected_flight_departures_list = make_flight_connection_list(
)
response = Mock()
response.status_code = 200
response.content = flight_departures_dict_list
response.json = Mock(return_value=flight_departures_dict_list)
request_handler = Mock()
request_handler.get = Mock(return_value=response)
client = OpenskyNetworkClient(request_handler=request_handler)
timestamp = datetime.now()
params = {'airport': 'EDDF', 'begin': timestamp, 'end': timestamp}
flight_departures = client.get_flight_departures(**params)
assert expected_flight_departures_list == flight_departures
call_args = request_handler.get.call_args[1]
assert call_args['params']['airport'] == params['airport']
assert call_args['params']['begin'] == int(params['begin'].timestamp())
assert call_args['params']['end'] == int(params['end'].timestamp())
def test_get_flight_departures__json_true__flight_departures_dict_is_returned(
):
flight_departures_dict_list, _ = make_flight_connection_list()
response = Mock()
response.status_code = 200
response.content = flight_departures_dict_list
response.json = Mock(return_value=flight_departures_dict_list)
request_handler = Mock()
request_handler.get = Mock(return_value=response)
client = OpenskyNetworkClient(request_handler=request_handler)
params = {
'airport': 'EDDF',
'begin': 1517227200,
'end': 1517230800,
'json': True
}
flight_departures = client.get_flight_departures(**params)
assert flight_departures_dict_list == flight_departures
call_args = request_handler.get.call_args[1]
assert call_args['params']['airport'] == params['airport']
assert call_args['params']['begin'] == params['begin']
assert call_args['params']['end'] == params['end']