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