class TestMessageBox(unittest.TestCase): def setUp(self): self.msg_box = MessageBox('message1', 'message2') def test_init(self): msg_box = MessageBox() self.assertIsInstance(msg_box, MessageBox) def test_init_with_msgs(self): self.assertEqual(self.msg_box, ['message1', 'message2']) self.assertNotEqual(self.msg_box, ['message3', 'message4']) def test_str(self): self.assertEqual(str(self.msg_box), 'message1\nmessage2') def test_add(self): import datetime with patch.object(self.msg_box, '_now') as mock_now: mock_now.return_value = datetime.datetime(2014, 1, 1, 8, 9, 10) self.msg_box.add('message3') self.assertEqual(self.msg_box, ['message1', 'message2', '[08:09:10] message3'])
def setUp(self): self.msg_box = MessageBox('message1', 'message2')