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()