Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

test: rewrite UTs related to agent resolution in URI handling#203

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to ourterms of service andprivacy statement. We’ll occasionally send you account related emails.

Already on GitHub?Sign in to your account

Merged
fioan89 merged 1 commit intomainfromrefactor-agent-resolver-uts
Oct 3, 2025
Merged
Changes fromall commits
Commits
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
237 changes: 147 additions & 90 deletionssrc/test/kotlin/com/coder/toolbox/util/CoderProtocolHandlerTest.kt
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -21,13 +21,27 @@ import kotlinx.coroutines.CoroutineScope
importkotlinx.coroutines.channels.Channel
importkotlinx.coroutines.flow.MutableStateFlow
importkotlinx.coroutines.runBlocking
importorg.junit.jupiter.api.DisplayName
importjava.util.UUID
importkotlin.test.Test
importkotlin.test.assertEquals
importkotlin.test.assertNull

internalclassCoderProtocolHandlerTest {

privatecompanionobject {
valAGENT_RIKER=AgentTestData(name="Riker", id="9a920eee-47fb-4571-9501-e4b3120c12f2")
valAGENT_BILL=AgentTestData(name="Bill", id="fb3daea4-da6b-424d-84c7-36b90574cfef")
valAGENT_BOB=AgentTestData(name="Bob", id="b0e4c54d-9ba9-4413-8512-11ca1e826a24")

valALL_AGENTS=mapOf(
AGENT_BOB.name toAGENT_BOB.id,
AGENT_BILL.name toAGENT_BILL.id,
AGENT_RIKER.name toAGENT_RIKER.id
)

valSINGLE_AGENT=mapOf(AGENT_BOB.name toAGENT_BOB.id)
}

privateval context=CoderToolboxContext(
mockk<ToolboxUi>(relaxed=true),
mockk<EnvironmentUiPageManager>(),
Expand All@@ -51,128 +65,171 @@ internal class CoderProtocolHandlerTest {
MutableStateFlow(false)
)

privateval agents=
mapOf(
"agent_name_bob" to"b0e4c54d-9ba9-4413-8512-11ca1e826a24",
"agent_name_bill" to"fb3daea4-da6b-424d-84c7-36b90574cfef",
"agent_name_riker" to"9a920eee-47fb-4571-9501-e4b3120c12f2",
)
privateval agentBob=
mapOf(
"agent_name_bob" to"b0e4c54d-9ba9-4413-8512-11ca1e826a24",
)

@Test
@DisplayName("given aws with multiple agents, expect the correct agent to be resolved ifitmatches the agent_name query param")
fungetMatchingAgent() {
val ws=DataGen.workspace("ws", agents= agents)

val tests=
listOf(
Pair(
mapOf("agent_name" to"agent_name_riker"),
"9a920eee-47fb-4571-9501-e4b3120c12f2"
),
Pair(
mapOf("agent_name" to"agent_name_bill"),
"fb3daea4-da6b-424d-84c7-36b90574cfef"
),
Pair(
mapOf("agent_name" to"agent_name_bob"),
"b0e4c54d-9ba9-4413-8512-11ca1e826a24"
)
fun`given aworkspace with multiple agents when getMatchingAgent is called with a valid agent name thenitcorrectly resolves resolves an agent`() {
val ws=DataGen.workspace("ws", agents=ALL_AGENTS)

val testCases=listOf(
AgentMatchTestCase(
"resolves agent with name Riker",
mapOf("agent_name" toAGENT_RIKER.name),
AGENT_RIKER.uuid
),
AgentMatchTestCase(
"resolves agent with name Bill",
mapOf("agent_name" toAGENT_BILL.name),
AGENT_BILL.uuid
),
AgentMatchTestCase(
"resolves agent with name Bob",
mapOf("agent_name" toAGENT_BOB.name),
AGENT_BOB.uuid
)
)

runBlocking {
tests.forEach {
assertEquals(UUID.fromString(it.second), protocolHandler.getMatchingAgent(it.first, ws)?.id)
testCases.forEach { testCase->
assertEquals(
testCase.expectedAgentId,
protocolHandler.getMatchingAgent(testCase.params, ws)?.id,
"Failed:${testCase.description}"
)
}
}
}

@Test
@DisplayName("given a ws with only multiple agents expect the agent resolution to fail if none match the agent_name query param")
funfailsToGetMatchingAgent() {
val ws=DataGen.workspace("ws", agents= agents)
val tests=
listOf(
Triple(emptyMap(),MissingArgumentException::class,"Unable to determine"),
Triple(mapOf("agent_name" to""),MissingArgumentException::class,"Unable to determine"),
Triple(mapOf("agent_name" tonull),MissingArgumentException::class,"Unable to determine"),
Triple(mapOf("agent_name" to"not-an-agent-name"),IllegalArgumentException::class,"agent with ID"),
Triple(
mapOf("agent_name" to"agent_name_homer"),
IllegalArgumentException::class,
"agent with name"
)
fun`given a workspace with multiple agents when getMatchingAgent is called with invalid agent names then no agent is resolved`() {
val ws=DataGen.workspace("ws", agents=ALL_AGENTS)

val testCases=listOf(
AgentNullResultTestCase(
"empty parameters (i.e. no agent name) does not return any agent",
emptyMap()
),
AgentNullResultTestCase(
"empty agent_name does not return any agent",
mapOf("agent_name" to"")
),
AgentNullResultTestCase(
"null agent_name does not return any agent",
mapOf("agent_name" tonull)
),
AgentNullResultTestCase(
"non-existent agent does not return any agent",
mapOf("agent_name" to"agent_name_homer")
),
AgentNullResultTestCase(
"UUID instead of name does not return any agent",
mapOf("agent_name" to"not-an-agent-name")
)
)

runBlocking {
tests.forEach {
assertNull(protocolHandler.getMatchingAgent(it.first, ws)?.id)
testCases.forEach { testCase->
assertNull(
protocolHandler.getMatchingAgent(testCase.params, ws)?.id,
"Failed:${testCase.description}"
)
}
}
}

@Test
@DisplayName("given a ws with only one agent, the agent is selected even when agent_name query param was not provided")
fungetsFirstAgentWhenOnlyOne() {
val ws=DataGen.workspace("ws", agents= agentBob)
val tests=
listOf(
fun`given a workspace with a single agent when getMatchingAgent is called with an empty agent name then the default agent is resolved`() {
val ws=DataGen.workspace("ws", agents=SINGLE_AGENT)

val testCases=listOf(
AgentMatchTestCase(
"empty parameters (i.e. no agent name) auto-selects the one and only agent available",
emptyMap(),
AGENT_BOB.uuid
),
AgentMatchTestCase(
"empty agent_name auto-selects the one and only agent available",
mapOf("agent_name" to""),
mapOf("agent_name" tonull)
AGENT_BOB.uuid
),
AgentMatchTestCase(
"null agent_name auto-selects the one and only agent available",
mapOf("agent_name" tonull),
AGENT_BOB.uuid
)
)

runBlocking {
tests.forEach {
testCases.forEach { testCase->
assertEquals(
UUID.fromString("b0e4c54d-9ba9-4413-8512-11ca1e826a24"),
protocolHandler.getMatchingAgent(
it,
ws,
)?.id,
testCase.expectedAgentId,
protocolHandler.getMatchingAgent(testCase.params, ws)?.id,
"Failed:${testCase.description}"
)
}
}
}

@Test
@DisplayName("given a ws with only one agent, the agent is NOT selected when agent_name query param was provided but does not match")
funfailsToGetAgentWhenOnlyOne() {
val wsWithAgentBob=DataGen.workspace("ws", agents= agentBob)
val tests=
listOf(
Triple(
mapOf("agent_name" to"agent_name_garfield"),
IllegalArgumentException::class,
"agent with name"
),
)
fun`given a workspace with a single agent when getMatchingAgent is called with an invalid agent name then no agent is resolved`() {
val ws=DataGen.workspace("ws", agents=SINGLE_AGENT)

val testCase=AgentNullResultTestCase(
"non-matching agent_name with single agent",
mapOf("agent_name" to"agent_name_garfield")
)

runBlocking {
tests.forEach {
assertNull(protocolHandler.getMatchingAgent(it.first, wsWithAgentBob))
}
assertNull(
protocolHandler.getMatchingAgent(testCase.params, ws),
"Failed:${testCase.description}"
)
}
}

@Test
@DisplayName("fails to resolve any agent when the workspace has no agents")
funfailsToGetAgentWhenWorkspaceHasNoAgents() {
val wsWithoutAgents=DataGen.workspace("ws")
val tests=
listOf(
Triple(emptyMap(),IllegalArgumentException::class,"has no agents"),
Triple(mapOf("agent_name" to""),IllegalArgumentException::class,"has no agents"),
Triple(mapOf("agent_name" tonull),IllegalArgumentException::class,"has no agents"),
Triple(
mapOf("agent_name" to"agent_name_riker"),
IllegalArgumentException::class,
"has no agents"
),
fun`given a workspace with no agent when getMatchingAgent is called then no agent is resolved`() {
val ws=DataGen.workspace("ws")

val testCases=listOf(
AgentNullResultTestCase(
"empty parameters (i.e. no agent name) does not return any agent",
emptyMap()
),
AgentNullResultTestCase(
"empty agent_name does not return any agent",
mapOf("agent_name" to"")
),
AgentNullResultTestCase(
"null agent_name does not return any agent",
mapOf("agent_name" tonull)
),
AgentNullResultTestCase(
"valid agent_name does not return any agent",
mapOf("agent_name" toAGENT_RIKER.name)
)
)

runBlocking {
tests.forEach {
assertNull(protocolHandler.getMatchingAgent(it.first, wsWithoutAgents))
testCases.forEach { testCase->
assertNull(
protocolHandler.getMatchingAgent(testCase.params, ws),
"Failed:${testCase.description}"
)
}
}
}
}

internaldata classAgentTestData(valname:String,valid:String) {
val uuid:UUID get()=UUID.fromString(id)
}

internaldata classAgentMatchTestCase(
valdescription:String,
valparams:Map<String,String?>,
valexpectedAgentId:UUID
)

internaldata classAgentNullResultTestCase(
valdescription:String,
valparams:Map<String,String?>
)
}
Loading
[8]ページ先頭
©2009-2025 Movatter.jp