def test_main(self, dir_path, input_path, output_path, get_files,
save_csv_file, config):
dir_path.return_value = self.data_path
input_path.return_value = self.data_path
output_path.return_value = self.data_path
get_files.return_value = [self.file_path]
save_csv_file.side_effect = None
main(['process_viajes_data', 'input'])
def test_main_with_s3(self, dir_path, input_path, get_files, p_data,
save_csv_file, send_s3):
send_s3.side_effect = mock.MagicMock()
dir_path.return_value = 'dir'
input_path.return_value = 'input'
get_files.return_value = [
os.path.join(self.data_path, '2020-03-01.viaje')
]
save_csv_file.side_effect = None
main(['process_viajes_data', 'input', '--send-to-s3'])
def test_main_with_lower_and_upper_bound(self, dir_path, input_path,
output_path, get_files, p_data,
save_csv_file, config):
dir_path.return_value = self.data_path
input_path.return_value = self.data_path
output_path.return_value = self.data_path
get_files.return_value = [
os.path.join(self.data_path, '2020-03-01.viaje')
]
p_data.return_value = ['2020-03-01', '5930344']
save_csv_file.side_effect = None
main([
'process_viajes_data', 'input', '--lower-bound', '2019-10-01',
'--upper-bound', '2020-01-01'
])
def test_main_with_upper_bound_only(self, dir_path, input_path,
output_path, get_files, p_data,
save_csv_file, config):
dir_path.return_value = self.data_path
input_path.return_value = self.data_path
output_path.return_value = self.data_path
get_files.return_value = [
os.path.join(self.data_path, '2020-03-01.viaje')
]
p_data.return_value = [['2020-03-01', '420420']]
save_csv_file.side_effect = None
with self.assertRaises(SystemExit) as cm:
main([
'process_viajes_data', 'input', '--upper-bound', '2020-10-10'
])