def test_get_filetype_parser_works(self): """Test get_filetype_parser() works.""" self.assertIsNotNone( file_loader._get_filetype_parser('file.yaml', 'string')) self.assertIsNotNone( file_loader._get_filetype_parser('file.yaml', 'file')) self.assertIsNotNone( file_loader._get_filetype_parser('file.json', 'string')) self.assertIsNotNone( file_loader._get_filetype_parser('file.json', 'file'))
def test_get_filetype_parser_raises_errors_for_invalid_parser(self): """Test get_filetype_parser() raises error for invalid parser.""" with self.assertRaises(errors.InvalidParserTypeError): file_loader._get_filetype_parser('path/to/file.yaml', 'asdf')
def test_get_filetype_parser_raises_errors_for_invalid_ext(self): """Test get_filetype_parser() raises error for invalid extension.""" with self.assertRaises(errors.InvalidFileExtensionError): file_loader._get_filetype_parser('invalid/path', 'string')