test(managed-agent): use the managed store in row lock test

This commit is contained in:
yungsen.hys committed 2026-11-07 20:17:30 +08:00
1 parent b7c72867b0
commit 20d556ceca
1 file changed
+5 -9
@@ -1,7 +1,6 @@
package com.alibaba.qwen.code.managedagent;
import com.alibaba.qwen.code.managedagent.api.TenantContextFilter;
import com.alibaba.qwen.code.managedagent.model.RequestDigests;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;
@@ -303,10 +302,6 @@ class ManagedAgentDefinitionTest {
String tenant = tenant();
String agentId = json(create(tenant, "for-update", DEFINITION)
.andExpect(status().isAccepted())).path("id").asText();
com.alibaba.qwen.code.managedagent.persistence.ManagedAgentDefinitionStore store =
new com.alibaba.qwen.code.managedagent.persistence.ManagedAgentDefinitionStore(
new org.springframework.jdbc.core.JdbcTemplate(
jdbc.getDataSource()), new RequestDigests());
org.springframework.transaction.support.TransactionTemplate template =
new org.springframework.transaction.support.TransactionTemplate(
transactionManager);
@@ -325,7 +320,7 @@ class ManagedAgentDefinitionTest {
contenderEntered.countDown();
template.executeWithoutResult(ignored -> {
long started = System.nanoTime();
store.latestForUpdate(tenant, agentId);
assertThat(definitions.latestForUpdate(tenant, agentId)).isPresent();
contenderReadMillis.set(
(System.nanoTime() - started) / 1_000_000);
});
@@ -335,7 +330,7 @@ class ManagedAgentDefinitionTest {
});
contender.start();
template.executeWithoutResult(ignored -> {
store.latestForUpdate(tenant, agentId);
assertThat(definitions.latestForUpdate(tenant, agentId)).isPresent();
holderLocked.countDown();
try {
// The contender enters its transaction right after the
@@ -348,8 +343,9 @@ class ManagedAgentDefinitionTest {
}
});
contender.join(5000);
assertThat(contenderFailure.get()).as(
"the contender's locking read failed for another reason")
assertThat(contenderFailure.get()).withFailMessage(() ->
"the contender's locking read failed: " + contenderFailure.get()
+ "; cause: " + contenderFailure.get().getCause())
.isNull();
assertThat(contender.isAlive()).isFalse();
// The contender's locking read must block on the holder's row lock