def setUp(self):
self.socket = Socket(None, None)
Socket.USER = None
class TestSocket(object):
def setUp(self):
self.socket = Socket(None, None)
Socket.USER = None
def test_tells_the_user_when_the_socket_is_opened(self):
# Arrange
Socket.USER = Mock()
# Act
self.socket.opened()
# Assert
Socket.USER.on_web_socket_opened.assert_called_once_with(self.socket)
def does_not_raise_exception_when_socket_is_opened_with_user_unknown(self):
# Act
self.socket.opened()
# Assert that no exception is raised
def test_sends_message(self):
# Arrange
self.socket.send = Mock()
# Act
self.socket.send_message('hi')
# Assert
self.socket.send.assert_called_once_with('hi')
def test_tells_the_user_when_the_socket_is_closed(self):
# Arrange
Socket.USER = Mock()
# Act
self.socket.closed(code=123, reason="everything has an end")
# Assert
Socket.USER.on_web_socket_closed.assert_called_once_with()
def does_not_raise_exception_when_socket_is_closed_with_user_unknown(self):
# Arrange
self.socket.opened()
# Act
self.socket.closed()