def setUp(self):
self.gtk_output = GtkOutput()
class TestGTKOutput(unittest.TestCase):
def setUp(self):
self.gtk_output = GtkOutput()
def tearDown(self):
self.gtk_output.end()
def test_gtk_output(self):
messages = []
def observer(message):
messages.append((message.get_type(), message.get_msg()))
self.gtk_output.subscribe(observer)
self.gtk_output.console('1')
self.gtk_output.information('2')
self.gtk_output.vulnerability('3')
self.gtk_output.debug('4')
self.gtk_output.error('5')
self.gtk_output.unsubscribe(observer)
self.gtk_output.vulnerability('ignores')
EXPECTED = set([
('console', '1'),
('information', '2'),
('vulnerability', '3'),
('debug', ''), # Note that this empty string is correct
('error', '5'), ]
)
self.assertEquals(set(messages), EXPECTED)