Add owned OpenAI endpoint and coordinated native model switching
This commit is contained in:
+3
-3
@@ -77,7 +77,7 @@ class AccessAPITests(unittest.TestCase):
|
||||
self.base=f'http://127.0.0.1:{self.server.server_port}'
|
||||
self.password=secrets.token_urlsafe(32);self.token=secrets.token_urlsafe(32)
|
||||
def tearDown(self):
|
||||
self.server.shutdown();self.server.demo.stop();self.server.server_close();self.thread.join();self.directory.cleanup()
|
||||
self.server.shutdown();self.server.endpoint.close();self.server.server_close();self.thread.join();self.directory.cleanup()
|
||||
def request(self,path,body=None,cookie=None,token=None,headers=None):
|
||||
h={**(headers or {})}
|
||||
if body is not None:h.update({'Content-Type':'application/json','X-Athena-Deck':'1'})
|
||||
@@ -127,8 +127,8 @@ class AccessAPITests(unittest.TestCase):
|
||||
for secret in (new,self.token,self.password,'api_token_hash','salt'):self.assertNotIn(secret,raw)
|
||||
def test_api_client_without_browser_headers(self):
|
||||
self.setup_login()
|
||||
req=urllib.request.Request(self.base+'/api/v1/demo/start',method='POST',headers={'Authorization':'Bearer '+self.token})
|
||||
with urllib.request.urlopen(req) as r:self.assertTrue(json.load(r)['reachable'])
|
||||
req=urllib.request.Request(self.base+'/api/v1/status',method='GET',headers={'Authorization':'Bearer '+self.token})
|
||||
with urllib.request.urlopen(req) as r:self.assertEqual(json.load(r)['name'],'Athena Deck')
|
||||
def test_wrong_password_and_logout(self):
|
||||
cookie=self.setup_login()
|
||||
self.assertEqual(self.request('/api/v1/auth/token',dict(current_password='wrong',new_token=secrets.token_urlsafe(32)),cookie=cookie)[0],400)
|
||||
|
||||
Reference in New Issue
Block a user