import unittest
from fastapi import HTTPException
from services.dashscope_voice import VoiceRequest, enrollment_payload


class VoiceEnrollmentTests(unittest.TestCase):
    def request(self, **changes):
        return VoiceRequest.model_validate({"apiUrl": "wss://dashscope.aliyuncs.com/api-ws/v1/inference", "model": "cosyvoice-v3-plus", "action": "create_voice", "audio": "data:audio/wav;base64,UklGRg==", **changes})

    def test_creation_and_region(self):
        url, payload = enrollment_payload(self.request())
        self.assertEqual(url, "https://dashscope.aliyuncs.com/api/v1/services/audio/tts/customization")
        self.assertEqual(payload["input"]["target_model"], "cosyvoice-v3-plus")

    def test_no_credentials_to_arbitrary_host(self):
        with self.assertRaises(HTTPException):
            enrollment_payload(self.request(apiUrl="https://evil.example"))

    def test_query_needs_id(self):
        with self.assertRaises(HTTPException):
            enrollment_payload(self.request(action="query_voice"))

    def test_invalid_audio(self):
        with self.assertRaises(HTTPException):
            enrollment_payload(self.request(audio="data:audio/wav;base64,???"))