def test_apply_applies_all_formatters_sequentially_to_lines(self):
formatter = Mock()
container = FileContainer(Path('test/files/Test.java'), Path('out'))
container.formatters.append(formatter)
container.apply()
formatter.apply.assert_called_with(container.lines)
def test_write_saves_file_with_formatter_applied(self):
container = FileContainer(Path('test/files/Test.java'), Path('out'))
container.write()
with open('out/test/files/Test.java') as f:
lines = f.readlines()
self.assertEqual(lines, [
'class Foo {\n',
' Foo() {\n',
' }\n',
'}\n',
])
def test_constructor_loads_lines_of_file(self):
container = FileContainer(Path('test/files/Test.java'), Path('out'))
self.assertEqual(container.lines, [
Line('class Foo {\n', 1),
Line(' Foo() {\n', 2),
Line(' }\n', 3),
Line('}\n', 4)
])
def _map_file(self, file: dict) -> FileContainer:
file_container = FileContainer.from_json(
file,
Path(os.getcwd()).joinpath(Path(self.input['out'])))
typer.echo(f"Found {file['path']}")
for layer in file['layers']:
file_container.formatters.append(self._map_layer(layer))
return file_container
def test_from_json_sets_output_path_generator_to_collapse_when_collapse_prop_present(
self):
json = {
'path': 'test/files/Test.java',
'collapse': "_",
}
container = FileContainer.from_json(json, Path('out'))
expected = CollapsePath('_')
self.assertEqual(container.output_path_generator, expected)
def test_from_json_sets_tag_to_None_if_empty_string(self):
json = {'path': 'test/files/Test.java', 'tag': ''}
container = FileContainer.from_json(json, Path('out'))
self.assertIsNone(container.tag)
def test_from_json_sets_tag_to_None_if_not_present(self):
json = {
'path': 'test/files/Test.java',
}
container = FileContainer.from_json(json, Path('out'))
self.assertIsNone(container.tag)
def test_final_path_includes_tag(self):
container = FileContainer(Path('test/files/Test.java'),
Path('out'),
tag='tag')
expected = Path('out/test/files/Test.java_tag')
self.assertEqual(expected, container.final_path())
def test_final_path_combines_output_dir_with_path(self):
container = FileContainer(Path('test/files/Test.java'), Path('out'))
expected = Path('out/test/files/Test.java')
self.assertEqual(expected, container.final_path())