def test_v2_public(self):
self.positional[0] = fixtureToAbsolute('v2-public.csv')
argv = build_argv(positional=self.positional)
with captured_output(argv) as (out, err):
sut.main()
validate_files(self.actual_file, fieldsToAbsolute('v2-json.txt'))
actual_print = out.getvalue().strip()
self.assertEqual('Using "v2-json.txt" for field mapping', actual_print)
def test_bad_input(self):
self.positional[0] = fixtureToAbsolute('complaints-subset.csv')
argv = build_argv(positional=self.positional)
with self.assertRaises(SystemExit) as ex:
with captured_output(argv) as (out, err):
sut.main()
self.assertEqual(ex.exception.code, 2)
console_output = err.getvalue()
self.assertIn('Unknown field set', console_output)
def test_v1_intake_csv(self):
self.positional[0] = fixtureToAbsolute('v1-intake.csv')
optional = ['--target-format', 'CSV']
argv = build_argv(optional, self.positional)
with captured_output(argv) as (out, err):
sut.main()
validate_files(self.actual_file, fieldsToAbsolute('v1-csv.txt'))
actual_print = out.getvalue().strip()
self.assertEqual('Using "v1-csv.txt" for field mapping', actual_print)
def test_bad_format_argument(self):
self.positional[0] = fixtureToAbsolute('v1-intake.csv')
optional = ['--target-format', 'tsv']
argv = build_argv(optional, self.positional)
with self.assertRaises(SystemExit) as ex:
with captured_output(argv) as (out, err):
sut.main()
self.assertEqual(ex.exception.code, 2)
console_output = err.getvalue()
self.assertIn('usage: choose_field_map', console_output)
self.assertIn('--target-format: invalid choice', console_output)