dshvaika
1 year ago
12 changed files with 227 additions and 104 deletions
@ -0,0 +1,102 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2025 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.common.data.limit; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.thingsboard.server.common.data.tenant.profile.DefaultTenantProfileConfiguration; |
||||
|
|
||||
|
import java.util.Arrays; |
||||
|
import java.util.Map; |
||||
|
import java.util.Set; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.mockito.Mockito.clearInvocations; |
||||
|
import static org.mockito.Mockito.mock; |
||||
|
import static org.mockito.Mockito.verify; |
||||
|
|
||||
|
class LimitedApiTest { |
||||
|
|
||||
|
private DefaultTenantProfileConfiguration config; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
config = mock(DefaultTenantProfileConfiguration.class); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testCorrectConfigExtractorsUsed() { |
||||
|
Map<LimitedApi, Runnable> verifierMap = Map.ofEntries( |
||||
|
Map.entry(LimitedApi.ENTITY_EXPORT, () -> |
||||
|
verify(config).getTenantEntityExportRateLimit()), |
||||
|
Map.entry(LimitedApi.ENTITY_IMPORT, () -> |
||||
|
verify(config).getTenantEntityImportRateLimit()), |
||||
|
Map.entry(LimitedApi.NOTIFICATION_REQUESTS, () -> |
||||
|
verify(config).getTenantNotificationRequestsRateLimit()), |
||||
|
Map.entry(LimitedApi.NOTIFICATION_REQUESTS_PER_RULE, () -> |
||||
|
verify(config).getTenantNotificationRequestsPerRuleRateLimit()), |
||||
|
Map.entry(LimitedApi.REST_REQUESTS_PER_TENANT, () -> |
||||
|
verify(config).getTenantServerRestLimitsConfiguration()), |
||||
|
Map.entry(LimitedApi.REST_REQUESTS_PER_CUSTOMER, () -> |
||||
|
verify(config).getCustomerServerRestLimitsConfiguration()), |
||||
|
Map.entry(LimitedApi.WS_UPDATES_PER_SESSION, () -> |
||||
|
verify(config).getWsUpdatesPerSessionRateLimit()), |
||||
|
Map.entry(LimitedApi.CASSANDRA_WRITE_QUERIES_CORE, () -> |
||||
|
verify(config).getCassandraWriteQueryTenantCoreRateLimits()), |
||||
|
Map.entry(LimitedApi.CASSANDRA_READ_QUERIES_CORE, () -> |
||||
|
verify(config).getCassandraReadQueryTenantCoreRateLimits()), |
||||
|
Map.entry(LimitedApi.CASSANDRA_WRITE_QUERIES_RULE_ENGINE, () -> |
||||
|
verify(config).getCassandraWriteQueryTenantRuleEngineRateLimits()), |
||||
|
Map.entry(LimitedApi.CASSANDRA_READ_QUERIES_RULE_ENGINE, () -> |
||||
|
verify(config).getCassandraReadQueryTenantRuleEngineRateLimits()), |
||||
|
Map.entry(LimitedApi.CASSANDRA_READ_QUERIES_MONOLITH, () -> { |
||||
|
verify(config).getCassandraReadQueryTenantCoreRateLimits(); |
||||
|
verify(config).getCassandraReadQueryTenantRuleEngineRateLimits(); |
||||
|
}), |
||||
|
Map.entry(LimitedApi.CASSANDRA_WRITE_QUERIES_MONOLITH, () -> { |
||||
|
verify(config).getCassandraWriteQueryTenantCoreRateLimits(); |
||||
|
verify(config).getCassandraWriteQueryTenantRuleEngineRateLimits(); |
||||
|
}), |
||||
|
Map.entry(LimitedApi.EDGE_EVENTS, () -> |
||||
|
verify(config).getEdgeEventRateLimits()), |
||||
|
Map.entry(LimitedApi.EDGE_EVENTS_PER_EDGE, () -> |
||||
|
verify(config).getEdgeEventRateLimitsPerEdge()), |
||||
|
Map.entry(LimitedApi.EDGE_UPLINK_MESSAGES, () -> |
||||
|
verify(config).getEdgeUplinkMessagesRateLimits()), |
||||
|
Map.entry(LimitedApi.EDGE_UPLINK_MESSAGES_PER_EDGE, () -> |
||||
|
verify(config).getEdgeUplinkMessagesRateLimitsPerEdge()) |
||||
|
); |
||||
|
|
||||
|
Set<LimitedApi> expected = verifierMap.keySet(); |
||||
|
Set<LimitedApi> actual = Arrays.stream(LimitedApi.values()) |
||||
|
.filter(api -> api.getConfigExtractor() != null) |
||||
|
.collect(Collectors.toSet()); |
||||
|
|
||||
|
assertThat(expected) |
||||
|
.as("Verifier map should cover all LimitedApis with extractors") |
||||
|
.containsExactlyInAnyOrderElementsOf(actual); |
||||
|
|
||||
|
for (Map.Entry<LimitedApi, Runnable> entry : verifierMap.entrySet()) { |
||||
|
LimitedApi api = entry.getKey(); |
||||
|
api.getLimitConfig(config); |
||||
|
entry.getValue().run(); |
||||
|
clearInvocations(config); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue