def test_run_dir_ci(self, mock_print):
with self.assertRaises(CIFailure):
run(
self.test_data,
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=True,
print=True,
header=False,
subconfig=False),
)
def test_run_dir_subconfig_invalid(self, mock_print):
config_file = self.test_data / IMPORTANIZE_JSON_CONFIG
config_file.write_text("invalid json")
try:
consume(
run(
self.test_data,
Config(CONFIG),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=True,
length=None,
list=False,
),
)
)
mock_print.assert_has_calls([mock.call(self.output_grouped)])
finally:
config_file.unlink()
def test_run_dir_subconfig_valid(self, mock_print):
config = deepcopy(CONFIG)
config['after_imports_new_lines'] = 1
config_file = self.test_data / IMPORTANIZE_CONFIG
config_file.write_text(force_text(json.dumps(config)))
try:
actual = run(
self.test_data,
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=True),
)
self.assertIsNone(actual)
mock_print.assert_has_calls([
mock.call(self.output_grouped_single_line),
])
finally:
config_file.unlink()
def test_run_text_print(self, mock_print):
actual = next(
run(
self.input_text,
CONFIG,
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=True,
subconfig=False,
length=None,
list=False,
),
Path("foo"),
)
)
self.assertEqual(actual, self.output_grouped)
mock_print.assert_has_calls(
[
mock.call("==="),
mock.call("foo"),
mock.call("---"),
mock.call(self.output_grouped),
]
)
def test_run_dir_subconfig_valid(self, mock_print):
config = deepcopy(CONFIG)
config["after_imports_new_lines"] = 1
config_file = self.test_data / IMPORTANIZE_JSON_CONFIG
config_file.write_text(force_text(json.dumps(config)))
try:
consume(
run(
self.test_data,
Config(CONFIG),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=True,
length=None,
list=False,
),
)
)
mock_print.assert_has_calls(
[mock.call(self.output_grouped_single_line)]
)
finally:
config_file.unlink()
def test_run_text_to_file_nothing_to_do(self, mock_write_text):
actual = run(
self.output_grouped,
CONFIG,
mock.Mock(formatter='grouped',
ci=False,
print=False,
subconfig=False),
Path(__file__),
)
self.assertEqual(actual, self.output_grouped)
mock_write_text.assert_not_called()
def test_run_file(self, mock_print):
actual = run(
self.test_data / 'input.py',
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False),
)
self.assertEqual(actual, self.output_grouped)
mock_print.assert_called_once_with(self.output_grouped)
def test_run_text_to_file_organized(self, mock_write_text):
actual = run(
self.input_text,
CONFIG,
mock.Mock(formatter='grouped',
ci=False,
print=False,
subconfig=False),
Path(__file__),
)
self.assertEqual(actual, self.output_grouped)
mock_write_text.assert_called_once_with(self.output_grouped)
def test_run_dir(self, mock_print):
actual = run(
self.test_data,
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False),
)
self.assertIsNone(actual)
mock_print.assert_has_calls([
mock.call(self.output_grouped),
])
def test_run_text_print_no_file(self, mock_print):
actual = run(
self.input_text,
CONFIG,
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=True,
subconfig=False),
)
self.assertEqual(actual, self.output_grouped)
mock_print.assert_has_calls([
mock.call(self.output_grouped),
])
def test_run_dir_skipped(self, mock_print):
config = deepcopy(CONFIG)
config['exclude'] = ['*/test_data']
actual = run(
self.test_data,
Config(config),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False),
)
self.assertIsNone(actual)
mock_print.assert_not_called()
def test_run_dir_ci(self, mock_print):
with self.assertRaises(CIFailure):
consume(
run(
self.test_data,
Config(CONFIG),
mock.Mock(
formatter="grouped",
ci=True,
print=True,
header=False,
subconfig=False,
length=None,
list=False,
),
)
)
def test_run_dir(self, mock_print):
consume(
run(
self.test_data,
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False),
))
mock_print.assert_has_calls([
mock.call(self.output_grouped),
])
def test_run_dir_skipped(self, mock_print):
config = deepcopy(CONFIG)
config['exclude'] = ['*/test_data']
consume(
run(
self.test_data,
Config(config),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False),
))
mock_print.assert_not_called()
def test_run_dir(self, mock_print):
consume(
run(
self.test_data,
Config(CONFIG),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False,
),
)
)
mock_print.assert_has_calls([mock.call(self.output_grouped)])
def test_run_file(self, mock_print):
actual = next(
run(
self.test_data / "input.py",
Config(CONFIG),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False,
),
)
)
self.assertEqual(actual, self.output_grouped)
mock_print.assert_called_once_with(self.output_grouped)
def test_run_text_to_file_nothing_to_do(self, mock_write_text):
actual = next(
run(
self.output_grouped,
CONFIG,
mock.Mock(
formatter="grouped",
ci=False,
print=False,
subconfig=False,
length=None,
list=False,
),
Path(__file__),
)
)
self.assertEqual(actual, self.output_grouped)
mock_write_text.assert_not_called()
def test_run_text_print_no_header(self, mock_print):
actual = next(
run(
self.input_text,
CONFIG,
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False),
Path('foo'),
))
self.assertEqual(actual, self.output_grouped)
mock_print.assert_has_calls([
mock.call(self.output_grouped),
])
def test_run_file_skipped(self, mock_print):
config = deepcopy(CONFIG)
config["exclude"] = ["*/test_data/*.py"]
consume(
run(
self.test_data / "input.py",
Config(config),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False,
),
)
)
mock_print.assert_not_called()
def test_run_dir_skipped(self, mock_print):
config = deepcopy(CONFIG)
config["exclude"] = ["*/test_data"]
consume(
run(
self.test_data,
Config(config),
mock.Mock(
formatter="grouped",
ci=False,
print=True,
header=False,
subconfig=False,
length=None,
list=False,
),
)
)
mock_print.assert_not_called()
def test_run_dir_subconfig_invalid(self, mock_print):
config_file = self.test_data / IMPORTANIZE_CONFIG
config_file.write_text('invalid json')
try:
actual = run(
self.test_data,
Config(CONFIG),
mock.Mock(formatter='grouped',
ci=False,
print=True,
header=False,
subconfig=True),
)
self.assertIsNone(actual)
mock_print.assert_has_calls([
mock.call(self.output_grouped),
])
finally:
config_file.unlink()
