Andrew Shvayka
3 years ago
committed by
GitHub
GitHub
14 changed files with 318 additions and 119 deletions
@ -0,0 +1,40 @@ |
|||
/** |
|||
* Copyright © 2016-2023 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.queue.util; |
|||
|
|||
import org.thingsboard.server.common.data.StringUtils; |
|||
|
|||
import java.util.HashMap; |
|||
import java.util.Map; |
|||
|
|||
public class PropertyUtils { |
|||
|
|||
public static Map<String, String> getProps(String properties) { |
|||
Map<String, String> configs = new HashMap<>(); |
|||
if (StringUtils.isNotEmpty(properties)) { |
|||
for (String property : properties.split(";")) { |
|||
if (StringUtils.isNotEmpty(property)) { |
|||
int delimiterPosition = property.indexOf(":"); |
|||
String key = property.substring(0, delimiterPosition); |
|||
String value = property.substring(delimiterPosition + 1); |
|||
configs.put(key, value); |
|||
} |
|||
} |
|||
} |
|||
return configs; |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,83 @@ |
|||
/** |
|||
* Copyright © 2016-2023 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.queue.kafka; |
|||
|
|||
import org.junit.jupiter.api.BeforeEach; |
|||
import org.junit.jupiter.api.Test; |
|||
import org.mockito.Mockito; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.boot.test.context.SpringBootTest; |
|||
import org.springframework.test.context.TestPropertySource; |
|||
|
|||
import java.util.Properties; |
|||
|
|||
import static org.assertj.core.api.Assertions.assertThat; |
|||
import static org.mockito.ArgumentMatchers.any; |
|||
import static org.mockito.Mockito.spy; |
|||
|
|||
@SpringBootTest(classes = TbKafkaSettings.class) |
|||
@TestPropertySource(properties = { |
|||
"queue.type=kafka", |
|||
"queue.kafka.bootstrap.servers=localhost:9092", |
|||
"queue.kafka.other-inline=metrics.recording.level:INFO;metrics.sample.window.ms:30000", |
|||
}) |
|||
class TbKafkaSettingsTest { |
|||
|
|||
@Autowired |
|||
TbKafkaSettings settings; |
|||
|
|||
@BeforeEach |
|||
void beforeEach() { |
|||
settings = spy(settings); //SpyBean is not aware on @ConditionalOnProperty, that is why the traditional spy in use
|
|||
} |
|||
|
|||
@Test |
|||
void givenToProps_whenConfigureSSL_thenVerifyOnce() { |
|||
Properties props = settings.toProps(); |
|||
|
|||
assertThat(props).as("TB_QUEUE_KAFKA_REQUEST_TIMEOUT_MS").containsEntry("request.timeout.ms", 30000); |
|||
assertThat(props).as("TB_QUEUE_KAFKA_SESSION_TIMEOUT_MS").containsEntry("session.timeout.ms", 10000); |
|||
|
|||
//other-inline
|
|||
assertThat(props).as("metrics.recording.level").containsEntry("metrics.recording.level", "INFO"); |
|||
assertThat(props).as("TB_QUEUE_KAFKA_SESSION_TIMEOUT_MS").containsEntry("metrics.sample.window.ms", "30000"); |
|||
|
|||
Mockito.verify(settings).toProps(); |
|||
Mockito.verify(settings).configureSSL(any()); |
|||
} |
|||
|
|||
@Test |
|||
void givenToAdminProps_whenConfigureSSL_thenVerifyOnce() { |
|||
settings.toAdminProps(); |
|||
Mockito.verify(settings).toProps(); |
|||
Mockito.verify(settings).configureSSL(any()); |
|||
} |
|||
|
|||
@Test |
|||
void givenToConsumerProps_whenConfigureSSL_thenVerifyOnce() { |
|||
settings.toConsumerProps("main"); |
|||
Mockito.verify(settings).toProps(); |
|||
Mockito.verify(settings).configureSSL(any()); |
|||
} |
|||
|
|||
@Test |
|||
void givenTotoProducerProps_whenConfigureSSL_thenVerifyOnce() { |
|||
settings.toProducerProps(); |
|||
Mockito.verify(settings).toProps(); |
|||
Mockito.verify(settings).configureSSL(any()); |
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,62 @@ |
|||
/** |
|||
* Copyright © 2016-2023 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.queue.util; |
|||
|
|||
import org.junit.jupiter.api.Test; |
|||
|
|||
import java.util.Map; |
|||
|
|||
import static org.assertj.core.api.Assertions.assertThat; |
|||
|
|||
class PropertyUtilsTest { |
|||
|
|||
@Test |
|||
void givenNullOrEmpty_whenGetConfig_thenEmptyMap() { |
|||
assertThat(PropertyUtils.getProps(null)).as("null property").isEmpty(); |
|||
assertThat(PropertyUtils.getProps("")).as("empty property").isEmpty(); |
|||
assertThat(PropertyUtils.getProps(";")).as("ends with ;").isEmpty(); |
|||
} |
|||
|
|||
@Test |
|||
void givenKafkaOtherProperties_whenGetConfig_thenReturnMappedValues() { |
|||
assertThat(PropertyUtils.getProps("metrics.recording.level:INFO;metrics.sample.window.ms:30000")) |
|||
.as("two pairs") |
|||
.isEqualTo(Map.of( |
|||
"metrics.recording.level", "INFO", |
|||
"metrics.sample.window.ms", "30000" |
|||
)); |
|||
|
|||
assertThat(PropertyUtils.getProps("metrics.recording.level:INFO;metrics.sample.window.ms:30000" + ";")) |
|||
.as("two pairs ends with ;") |
|||
.isEqualTo(Map.of( |
|||
"metrics.recording.level", "INFO", |
|||
"metrics.sample.window.ms", "30000" |
|||
)); |
|||
} |
|||
|
|||
@Test |
|||
void givenKafkaTopicProperties_whenGetConfig_thenReturnMappedValues() { |
|||
assertThat(PropertyUtils.getProps("retention.ms:604800000;segment.bytes:26214400;retention.bytes:1048576000;partitions:1;min.insync.replicas:1")) |
|||
.isEqualTo(Map.of( |
|||
"retention.ms", "604800000", |
|||
"segment.bytes", "26214400", |
|||
"retention.bytes", "1048576000", |
|||
"partitions", "1", |
|||
"min.insync.replicas", "1" |
|||
)); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue