Add a websockets tunnel and a test for the proxy's websockets support.

[hlinnaka]

For testing the proxy's websockets support.

I wrote this to test #3822. Unfortunately, that bug can not be reproduced with this tunnel. The bug only appears when the client pipelines the first query with the authentication messages. The tunnel doesn't do that.

---

Update (@conradludgate 2025-01-10):

We have since added some websocket tests, but they manually implemented a very simplistic setup of the postgres protocol. Introducing the tunnel would make more complex testing simpler in the future.

[hlinnaka]

So, this doesn't actually reproduce the bug I wanted to test with this, so this isn't as useful as I thought. But since I wrote it, here's a PR.

Should we commit this anyway, or find some other testing strategy?

[arssher]

Nice! But as you said, it is more useful to test real js client, so let's add something similar to what Stas did at sk/proxy-wss-test-wip branch to pg_clients suite. Arthur created issue for that:
#3819
@petuhovskiy would you pick it up?

[petuhovskiy]

This test looks like a nice way to test basic websocket proxy logic without a dependency on Docker/NodeJS/console. Even if it doesn't reproduce a bug that we have, I think it still will be useful to have such a test.

Also, wanted to look at the logs, but it seems that currently test doesn't work in CI:

 /github/home/.cache/pypoetry/virtualenvs/neon-_pxWMzVK-py3.9/lib/python3.9/site-packages/_pytest/assertion/rewrite.py:170: in exec_module
    exec(co, module.__dict__)
test_runner/regress/test_proxy.py:6: in <module>
    import websocket_tunnel
E     File "/__w/neon/neon/test_runner/websocket_tunnel.py", line 101
E       except* Exception as ex:
E             ^
E   SyntaxError: invalid syntax

[kelvich]

+1 good to have it in the CI.

I was thinking about a similar setup with python test calling docker run with js app inside and check output.

[hlinnaka]

@conradludgate I just remembered this PR. I don't know what we should do with it, and I don't plan to work on it anymore. But maybe you find it useful? If not, feel free to close it.

[conradludgate]

Thanks, I'll take a look

[conradludgate]

Ah yes this will be very useful. I was looking to add some websocket tests last week, glad I checked my notifications again. I'll sort out this PR and merge it in. Thanks!

[github-actions]

7293 tests run: 6929 passed, 0 failed, 364 skipped (full report)

---

Code coverage* (full report)

• functions: 32.7% (8043 of 24618 functions)
• lines: 47.8% (66823 of 139908 lines)

* collected from Rust tests only

---

_{The comment gets automatically updated with the latest test results
6fc1a82 at 2025-01-10T17:51:08.619Z :recycle:}

[ololobus]

This PR has been automatically marked as Draft because it was last updated 450 days ago. Please, consider closing it if it's not needed anymore.

[awarus]

just a nit: it seems like we a couple of extra lines here