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)