test_websocket.py

import pytest
from websocket import create_connection


@pytest.fixture(scope="module")
def ws(request):
    ws = create_connection("ws://localhost:8080/bullet")

    def fin():
        ws.close()
    request.addfinalizer(fin)
    return ws


def test_echo(ws):
    ws.send("Hello, World")
    result = ws.recv()
    assert result == "Hello, World"


def test_fail(ws):
    ws.send("Hello, World")
    result = ws.recv()
    assert result == "Hello, World1"


def test_get_time(ws):
    result = ws.recv()
    assert isinstance(result, str)