0ct0pu5/kafka-ui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

wip

Browse source
This commit is contained in:
iliax 2023-08-11 19:33:56 +04:00
parent 5c6cd565a6
commit a02d537f1e
1 changed files with 22 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
kafka-ui-api/src/test/java/com/provectus/kafka/ui/service/quota/ClientQuotaServiceTest.java
View file

@ -34,37 +34,45 @@ class ClientQuotaServiceTest extends AbstractIntegrationTest {
nullValues = "null" nullValues = "null"
) )
void createUpdateDelete(String user, String clientId, String ip) { void createUpdateDelete(String user, String clientId, String ip) {
//creating new var quotas = Map.of(
StepVerifier.create(
quotaService.upsert(cluster, user, clientId, ip,
Map.of(
"producer_byte_rate", 123.0, "producer_byte_rate", 123.0,
"consumer_byte_rate", 234.0, "consumer_byte_rate", 234.0,
"request_percentage", 10.0 "request_percentage", 10.0
) );
)
//creating new
StepVerifier.create(
quotaService.upsert(cluster, user, clientId, ip, quotas)
) )
.assertNext(status -> assertThat(status.value()).isEqualTo(201)) .assertNext(status -> assertThat(status.value()).isEqualTo(201))
.verifyComplete(); .verifyComplete();
assertThat(quotaRecordExisits(new ClientQuotaRecord(user, clientId, ip, quotas)))
.isTrue();
//updating //updating
StepVerifier.create( StepVerifier.create(
quotaService.upsert(cluster, user, clientId, ip, quotaService.upsert(cluster, user, clientId, ip, Map.of("producer_byte_rate", 111111.0))
Map.of(
"producer_byte_rate", 111111.0,
"consumer_byte_rate", 22222.0
)
)
) )
.assertNext(status -> assertThat(status.value()).isEqualTo(200)) .assertNext(status -> assertThat(status.value()).isEqualTo(200))
.verifyComplete(); .verifyComplete();
//deleting just created record assertThat(quotaRecordExisits(new ClientQuotaRecord(user, clientId, ip, Map.of("producer_byte_rate", 111111.0))))
.isTrue();
//deleting created record
StepVerifier.create( StepVerifier.create(
quotaService.upsert(cluster, user, clientId, ip, Map.of()) quotaService.upsert(cluster, user, clientId, ip, Map.of())
) )
.assertNext(status -> assertThat(status.value()).isEqualTo(204)) .assertNext(status -> assertThat(status.value()).isEqualTo(204))
.verifyComplete(); .verifyComplete();
assertThat(quotaRecordExisits(new ClientQuotaRecord(user, clientId, ip, Map.of("producer_byte_rate", 111111.0))))
.isFalse();
}
private boolean quotaRecordExisits(ClientQuotaRecord rec) {
return quotaService.list(cluster).collectList().block().contains(rec);
} }
} }