def setUp(self): """Can't make an IRC connection in tests, so a mock will have to do.""" self.client = Client( handlers=[], nick='anick', real_name='Real Name', ) self.client.connection = mock.MagicMock(spec=Connection)
class TestOnConnect(TestCase): def setUp(self): """Can't make an IRC connection in tests, so a mock will have to do.""" self.client = Client( handlers=[], nick='anick', real_name='Real Name', ) self.client.connection = mock.MagicMock(spec=Connection) def test_user_command_sent(self): self.client.on_connect() expected = b'USER anick 0 * :Real Name\r\n' self.client.connection.send.assert_any_call(expected) def test_set_nick_called(self): self.client.on_connect() expected = b'NICK anick\r\n' self.client.connection.send.assert_called_with(expected)