mirror of
https://github.com/QwenLM/qwen-code.git
synced 2026-11-07 06:19:43 -10:00
test(managed-agent): use the managed store in row lock test
This commit is contained in:
1 parent
b7c72867b0
commit
20d556ceca
1 file changed
+5
-9
+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
|
||||
|
||||
Reference in new issue
Block a user