def test_collect_includes_with_file(self):
mock_node = mock.MagicMock()
node = pre_ast.File(mock_node)
mock_node.accept.return_value = 42
actual = self.include_collector.collect_includes(node)
expected = 42
self.assertEqual(actual, expected)
def test_resolve_with_file(self):
mock_node = mock.MagicMock()
node = pre_ast.File(mock_node)
self.include_linking_visitor.resolve(node, self.files)
mock_node.accept.assert_called_once_with(
self.include_linking_visitor,
self.files,
)
def test_to_string_with_file(self):
mock_node = mock.MagicMock()
node = pre_ast.File(mock_node)
mock_node.accept.side_effect = (lambda visitor, parts: parts.extend(
('foo', 'bar', '42')))
actual = self.to_string_visitor.to_string(node)
expected = 'foo bar 42'
self.assertEqual(actual, expected)
def visit_file(self, file_):
content = file_.content.accept(self)
return pre_ast.File(content)