Skip to content

Commit

Permalink
Auto resolve conflict by keeping our changes
Browse files Browse the repository at this point in the history
  • Loading branch information
box-sdk-build committed Feb 20, 2025
2 parents 0da38a1 + f991efd commit 98aee57
Showing 1 changed file with 61 additions and 0 deletions.
61 changes: 61 additions & 0 deletions src/test/java/com/box/sdkgen/test/aistudio/AiStudioITest.java
Original file line number Diff line number Diff line change
@@ -0,0 +1,61 @@
package com.box.sdkgen.test.aistudio;

import static com.box.sdkgen.internal.utils.UtilsManager.convertToString;
import static com.box.sdkgen.internal.utils.UtilsManager.getUuid;
import static com.box.sdkgen.test.commons.CommonsManager.getDefaultClient;

import com.box.sdkgen.client.BoxClient;
import com.box.sdkgen.managers.aistudio.GetAiAgentByIdQueryParams;
import com.box.sdkgen.schemas.aimultipleagentresponse.AiMultipleAgentResponse;
import com.box.sdkgen.schemas.aisingleagentresponsefull.AiSingleAgentResponseFull;
import com.box.sdkgen.schemas.aistudioagentask.AiStudioAgentAsk;
import com.box.sdkgen.schemas.createaiagent.CreateAiAgent;
import java.util.Arrays;
import org.junit.jupiter.api.Test;

public class AiStudioITest {

private static final BoxClient client = getDefaultClient();

@Test
public void testAiStudioCrud() {
String agentName = getUuid();
AiSingleAgentResponseFull createdAgent =
client
.getAiStudio()
.createAiAgent(
new CreateAiAgent.CreateAiAgentBuilder(agentName, "enabled")
.ask(new AiStudioAgentAsk("enabled", "desc1"))
.build());
assert createdAgent.getName().equals(agentName);
AiMultipleAgentResponse agents = client.getAiStudio().getAiAgents();
int numAgents = agents.getEntries().size();
assert convertToString(agents.getEntries().get(0).getType()).equals("ai_agent");
AiSingleAgentResponseFull retrievedAgent =
client
.getAiStudio()
.getAiAgentById(
createdAgent.getId(),
new GetAiAgentByIdQueryParams.GetAiAgentByIdQueryParamsBuilder()
.fields(Arrays.asList("ask"))
.build());
assert retrievedAgent.getName().equals(agentName);
assert convertToString(retrievedAgent.getAccessState()).equals("enabled");
assert convertToString(retrievedAgent.getAsk().getAccessState()).equals("enabled");
assert retrievedAgent.getAsk().getDescription().equals("desc1");
AiSingleAgentResponseFull updatedAgent =
client
.getAiStudio()
.updateAiAgentById(
createdAgent.getId(),
new CreateAiAgent.CreateAiAgentBuilder(agentName, "enabled")
.ask(new AiStudioAgentAsk("disabled", "desc2"))
.build());
assert convertToString(updatedAgent.getAccessState()).equals("enabled");
assert convertToString(updatedAgent.getAsk().getAccessState()).equals("disabled");
assert updatedAgent.getAsk().getDescription().equals("desc2");
client.getAiStudio().deleteAiAgentById(createdAgent.getId());
AiMultipleAgentResponse agentsAfterDelete = client.getAiStudio().getAiAgents();
assert agentsAfterDelete.getEntries().size() == numAgents - 1;
}
}

0 comments on commit 98aee57

Please sign in to comment.