class TestPrinter(TestCase):
def setUp(self):
self.paper = Paper()
self.printer = Printer()
self.printer.insert(self.paper)
def test_addIndentationAtTheStartOfALineAsLongAsItIsNotRemoved(self):
driver = MethodDefinitionPrinterDriver(self.printer)
tree = ast.parse(open("samples/method.py", 'r').read())
driver.visit(tree)
self.assertEqual(self.paper.text, 'public Object basic_method( Object one, Object two){\n}')
class TestPrinter(TestCase):
def setUp(self):
self.paper = Paper()
self.printer = Printer()
self.printer.insert(self.paper)
def test_justVisitAllChildrenIfNoSpecialPrinterIsRegistered(self):
self.printer.write("Some text")
self.assertEqual(self.paper.text, "Some text")
def test_changeLine(self):
self.printer.write("Some text")
self.printer.newline()
self.printer.write("more text")
self.assertEqual(self.paper.text, "Some text\nmore text")
def test_addIndentationAtTheStartOfALine(self):
self.printer.increaseIndentation()
self.printer.write("Some text")
self.assertEqual(self.paper.text, " Some text")
def test_decreaseIndentationLevelInNextLine(self):
self.printer.increaseIndentation()
self.printer.decreaseIndentation()
self.printer.write("Some text")
self.assertEqual(self.paper.text, "Some text")
def test_doNotAllowNegativeIndentationLevels(self):
self.printer.decreaseIndentation()
self.printer.increaseIndentation()
self.printer.write("Some text")
self.assertEqual(self.paper.text, " Some text")
def test_addIndentationAtTheStartOfALineAsLongAsItIsNotRemoved(self):
self.printer.increaseIndentation()
self.printer.write("Some text")
self.printer.newline()
self.printer.write("more text")
self.assertEqual(self.paper.text, " Some text\n more text")
