skygpu
/
skynet
mirror of https://github.com/skygpu/skynet.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
skynet/tests/test_skynet.py

71 lines
1.8 KiB
Python
Raw Normal View History

Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
#!/usr/bin/python
import logging
import trio
Add TLS for encrypted comms, next HMAC
2022-12-11 16:06:07 +00:00
import pynng
import pytest
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
import trio_asyncio
Rework dgpu client to be single task Add a lot of dgpu real image gen tests Modified docker files and environment to allow for quick test relaunch without image rebuild Rename package from skynet_bot to skynet Drop tractor usage cause cuda is oriented to just a single proc managing gpu resources Add ackwnoledge phase to image request for quick gpu disconnected type scenarios Add click entry point for dgpu Add posibility to reuse postgres_db fixture on same session by checking if schema init has been already done
2022-12-17 14:39:42 +00:00
from skynet.brain import run_skynet
from skynet.structs import *
from skynet.frontend import open_skynet_rpc
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
async def test_skynet(skynet_running):
...
Add TLS for encrypted comms, next HMAC
2022-12-11 16:06:07 +00:00
async def test_skynet_attempt_insecure(skynet_running):
with pytest.raises(pynng.exceptions.NNGException) as e:
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
async with open_skynet_rpc('bad-actor'):
...
Add TLS for encrypted comms, next HMAC
2022-12-11 16:06:07 +00:00
assert str(e.value) == 'Connection shutdown'
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
async def test_skynet_dgpu_connection_simple(skynet_running):
Add TLS for encrypted comms, next HMAC
2022-12-11 16:06:07 +00:00
async with open_skynet_rpc(
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
'dgpu-0',
Add TLS for encrypted comms, next HMAC
2022-12-11 16:06:07 +00:00
security=True,
cert_name='whitelist/testing',
key_name='testing'
) as rpc_call:
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
# check 0 nodes are connected
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_workers')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == 0
# check next worker is None
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_next')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == None
# connect 1 dgpu
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_online')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
# check 1 node is connected
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_workers')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == 1
# check next worker is 0
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_next')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == 0
# disconnect 1 dgpu
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_offline')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
# check 0 nodes are connected
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_workers')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == 0
# check next worker is None
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 15:36:02 +00:00
res = await rpc_call('dgpu_next')
Started making roboust testing fixtures to init fresh db and skynet Add simple dgpu worker connection test Make db connection handler manage schema and table init logic Keep tweaking dgpu main handler attemtping to fix subactor hangs Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket Decupled user config request validation from telegram module Fix next_worker logic, now takes in account multiple tasks per dgpu Add dgpu_workers and dgpu_next calls Fixed readme, moved db init code into db module
2022-12-11 14:02:55 +00:00
assert 'ok' in res.result
assert res.result['ok'] == None