Test prompt enhancer management access

2026-09-29 21:21:24 +02:00
parent c5a028cba9
commit 3c59512d85
1 changed files with 8 additions and 0 deletions
@@ -97,6 +97,14 @@ class AccessAPITests(unittest.TestCase):
        code,_,headers=self.request('/api/v1/login',{'password':password})
        self.assertEqual(code,200)
        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):
        cookie=self.setup_login();path='/api/v1/huggingface';secret='hf_'+'a'*30
        for value in ({}, {'token':self.token}):