package com.easyagents.core.test.util;

import com.easyagents.core.message.UserMessage;
import com.easyagents.core.util.ImageUtil;
import org.junit.Assert;
import org.junit.Test;

import java.util.List;

/**
 * 图片 Data URI 转换测试。
 */
public class ImageUtilTest {

    /**
     * 验证图片字节会生成符合标准的完整 Data URI。
     */
    @Test
    public void shouldAddDataSchemeWhenEncodingImageBytes() {
        String dataUri = ImageUtil.imageBytesToDataUri(new byte[]{1, 2, 3}, "image/png");

        Assert.assertEquals("data:image/png;base64,AQID", dataUri);
    }

    /**
     * 验证用户消息添加图片字节时保留完整 Data URI。
     */
    @Test
    public void shouldStoreCompleteDataUriInUserMessage() {
        UserMessage message = new UserMessage();

        message.addImageBytes(new byte[]{1, 2, 3}, "image/png");

        Assert.assertEquals(List.of("data:image/png;base64,AQID"), message.getImageUrls());
    }
}
