Run YuE2 music profiles through audio.cpp with WAV API and test UI

test_endpoint.py

@@ -61,6 +61,17 @@ class EndpointTests(unittest.TestCase):
  def request(self,path='/v1/models',data=None,token='A'*32):
   conn=http.client.HTTPConnection('127.0.0.1',self.port,timeout=5);headers={'Content-Type':'application/json','Authorization':'Bearer '+token}
   conn.request('POST' if data is not None else 'GET',path,body=json.dumps(data) if data is not None else None,headers=headers);r=conn.getresponse();body=r.read();conn.close();return r.status,(body if data and data.get('stream') else json.loads(body))
+ def test_music_discovery_and_authenticated_wav_route(self):
+  self.rows.append(dict(id='music',name='Music',kind='music',runnable=True,blockers=[],parameters={},updated_at=1))
+  self.ep.music=Mock();self.ep.music.start.return_value=dict(id='a'*32);self.ep.music.status.return_value=dict(job=dict(id='a'*32,state='complete'));self.ep.music.audio.return_value=b'RIFFsynthetic-WAV'
+  self.enable('music')
+  self.assertEqual(self.request('/v1/audio/music/models')[1]['data'][0]['id'],'Music')
+  self.assertNotIn('Music',[x['id'] for x in self.request()[1]['data']])
+  conn=http.client.HTTPConnection('127.0.0.1',self.port,timeout=5)
+  conn.request('POST','/v1/audio/music',json.dumps(dict(model='Music',lyrics='Synthetic',style='pop')),{'Content-Type':'application/json','Authorization':'Bearer '+'A'*32})
+  response=conn.getresponse();self.assertEqual(response.status,200);self.assertEqual(response.getheader('Content-Type'),'audio/wav');self.assertEqual(response.read(),b'RIFFsynthetic-WAV');conn.close()
+  self.assertEqual(self.request('/v1/audio/music',dict(model='Music',lyrics='Synthetic',style='pop',unsupported=True))[0],400)
+  self.assertEqual(self.request('/v1/audio/music/models',token='bad')[0],401)
  def enable(self,name):self.ep.enable(dict(id=name,enabled=True))
  def test_explicit_publication_auth_rotation_and_missing_audio(self):
   self.assertEqual(self.request()[1]['data'],[]);self.assertEqual(self.request(token='bad')[0],401)