class TestAddWriter(unittest.TestCase): def setUp(self): self.addingMachine = AddingMachineWrite() def tearDown(self): pass def testCreate(self): module = self.addingMachine.__module__ self.assertEqual(module, "AddingMachineWrite", "Could not create class") def testGetHeader(self): header = self.addingMachine.getHeader() self.assertTrue(header.startswith('Content-Type: text/html')) def testGetStart(self): start = self.addingMachine.getStart() self.assertTrue(start.startswith('<!DOCTYPE html>\n<html>'), 'Begins with: {0}'.format(start)) def testGetClose(self): text = self.addingMachine.getClose() text = text.strip() self.assertTrue(text.startswith("</body>"), "Begins with: {0}".format(text))
def setUp(self): self.addingMachine = AddingMachineWrite()
def setUpMock(self, MockClass): instance = MockClass.return_value instance.__getitem__ = lambda s, key: self.fields[key] instance.__contains__ = lambda s, key: key in self.fields instance.has_key = lambda key: key in self.fields self.addingMachine = AddingMachineWrite(False)