package com.zy.ai.gateway;
|
|
import com.zy.ai.gateway.model.AiUsage;
|
import org.junit.jupiter.api.Test;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
|
class AiGatewayModelTest {
|
|
@Test
|
void aiUsageStoresTokenFields() {
|
AiUsage usage = new AiUsage();
|
|
usage.setInputTokens(3);
|
usage.setOutputTokens(4);
|
usage.setTotalTokens(7);
|
|
assertEquals(3, usage.getInputTokens());
|
assertEquals(4, usage.getOutputTokens());
|
assertEquals(7, usage.getTotalTokens());
|
}
|
}
|
