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