2608a02d5c
Without this we end up in weird cases where the viewer gets a cache hit and never request the object data, creating link heirarchies where the viewer knows about all the prims but Hippolyzer only knows some of them and orphans them. Since we don't know what viewer the user is using, we scan around the disk for object caches and try to use those. 99% of the time the connection will be coming from localhost so this is fine. Fixes #11
68 lines
2.8 KiB
Python
68 lines
2.8 KiB
Python
import unittest
|
|
|
|
import aiohttp
|
|
import aioresponses
|
|
from yarl import URL
|
|
|
|
from hippolyzer.lib.base.datatypes import UUID
|
|
from hippolyzer.lib.proxy.caps_client import CapsClient
|
|
from hippolyzer.lib.proxy.region import ProxiedRegion
|
|
from hippolyzer.lib.proxy.sessions import SessionManager
|
|
|
|
|
|
class TestCapsClient(unittest.IsolatedAsyncioTestCase):
|
|
def setUp(self) -> None:
|
|
self.session = SessionManager().create_session({
|
|
"session_id": UUID.random(),
|
|
"secure_session_id": UUID.random(),
|
|
"agent_id": UUID.random(),
|
|
"circuit_code": 0,
|
|
"sim_ip": "127.0.0.1",
|
|
"sim_port": "1",
|
|
"region_x": 1,
|
|
"region_y": 2,
|
|
"seed_capability": "https://test.localhost:4/foo",
|
|
})
|
|
self.region = ProxiedRegion(("127.0.0.1", 1), "", self.session)
|
|
self.caps_client = CapsClient(self.region)
|
|
|
|
async def test_bare_url_works(self):
|
|
with aioresponses.aioresponses() as m:
|
|
m.get("https://example.com/", body=b"foo")
|
|
async with self.caps_client.get("https://example.com/") as resp:
|
|
self.assertEqual(await resp.read(), b"foo")
|
|
|
|
async def test_own_session_works(self):
|
|
with aioresponses.aioresponses() as m:
|
|
async with aiohttp.ClientSession() as sess:
|
|
m.get("https://example.com/", body=b"foo")
|
|
async with self.caps_client.get("https://example.com/", session=sess) as resp:
|
|
self.assertEqual(await resp.read(), b"foo")
|
|
|
|
async def test_read_llsd(self):
|
|
with aioresponses.aioresponses() as m:
|
|
m.get("https://example.com/", body=b"<llsd><integer>2</integer></llsd>")
|
|
async with self.caps_client.get("https://example.com/") as resp:
|
|
self.assertEqual(await resp.read_llsd(), 2)
|
|
|
|
async def test_caps(self):
|
|
self.region.update_caps({"Foobar": "https://example.com/"})
|
|
with aioresponses.aioresponses() as m:
|
|
m.post("https://example.com/baz", body=b"ok")
|
|
data = {"hi": "hello"}
|
|
headers = {"Foo": "bar"}
|
|
async with self.caps_client.post("Foobar", path="baz", llsd=data, headers=headers) as resp:
|
|
self.assertEqual(await resp.read(), b"ok")
|
|
|
|
# Our original dict should not have been touched
|
|
self.assertEqual(headers, {"Foo": "bar"})
|
|
|
|
req_key = ("POST", URL("https://example.com/baz"))
|
|
req_body = m.requests[req_key][0].kwargs['data']
|
|
self.assertEqual(req_body, b'<?xml version="1.0" ?><llsd><map><key>hi</key><string>hello'
|
|
b'</string></map></llsd>')
|
|
|
|
with self.assertRaises(KeyError):
|
|
with self.caps_client.get("BadCap"):
|
|
pass
|