michael/Athena-Deck
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test prompt enhancer management access

Browse Source
This commit is contained in:
Mikei386
2026-09-29 21:21:24 +02:00
parent c5a028cba9
commit 3c59512d85
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_auth.py
+8
View File
@@ -97,6 +97,14 @@ class AccessAPITests(unittest.TestCase):
code,_,headers=self.request('/api/v1/login',{'password':password}) code,_,headers=self.request('/api/v1/login',{'password':password})
self.assertEqual(code,200) self.assertEqual(code,200)
return headers['Set-Cookie'].split(';')[0] return headers['Set-Cookie'].split(';')[0]
def test_prompt_enhancer_management_requires_login_and_validates_actions(self):
self.assertEqual(self.request('/api/v1/prompt-enhancers')[0],401)
cookie=self.setup_login()
code,state,_=self.request('/api/v1/prompt-enhancers',cookie=cookie)
self.assertEqual(code,200)
self.assertEqual(set(state['models']),{'t2i','i2i'})
self.assertEqual(self.request('/api/v1/prompt-enhancers/install',{'task':'unknown'},cookie=cookie)[0],400)
self.assertEqual(self.request('/api/v1/profiles/prompt-enhancer',{'id':'missing','revision':0,'prompt_enhancer':{'t2i':None,'i2i':None,'device':'auto'}},cookie=cookie)[0],400)
def test_hub_credentials_are_session_only_and_never_returned(self): def test_hub_credentials_are_session_only_and_never_returned(self):
cookie=self.setup_login();path='/api/v1/huggingface';secret='hf_'+'a'*30 cookie=self.setup_login();path='/api/v1/huggingface';secret='hf_'+'a'*30
for value in ({}, {'token':self.token}): for value in ({}, {'token':self.token}):