def test_write_message_no_subscribers(mock_connection): ac = ActiveConnections() c = MockConnection() ac.connections = [c] ac.write_message({'user': '******'}) assert mock_connection.call_count == 0, 'Called when noone subscribed'
def test_write_message(mock_connection): ac = ActiveConnections() c = MockConnection() c.subscriptions = ['someuser'] ac.connections = [c] ac.write_message({'user': '******'}) mock_connection.assert_called_once_with(json.dumps({'user': '******'}))