|
1 | 1 | package javasabr.rlib.logger.impl.config.loader.impl; |
2 | 2 |
|
3 | 3 | import static org.assertj.core.api.Assertions.assertThat; |
| 4 | +import static org.assertj.core.api.Assertions.assertThatThrownBy; |
4 | 5 |
|
| 6 | +import java.io.ByteArrayInputStream; |
5 | 7 | import java.io.IOException; |
| 8 | +import java.io.InputStream; |
| 9 | +import java.nio.charset.StandardCharsets; |
| 10 | +import java.util.Map; |
| 11 | +import java.util.Optional; |
6 | 12 | import java.util.Properties; |
| 13 | +import java.util.function.Supplier; |
| 14 | +import java.util.stream.Collectors; |
7 | 15 | import javasabr.rlib.collections.array.Array; |
8 | 16 | import javasabr.rlib.collections.array.UnsafeArray; |
9 | 17 | import javasabr.rlib.logger.api.Logger; |
|
14 | 22 | import javasabr.rlib.logger.impl.config.consumer.LogMessageConsumer; |
15 | 23 | import javasabr.rlib.logger.impl.config.consumer.impl.ConsoleMessageConsumer; |
16 | 24 | import javasabr.rlib.logger.impl.config.impl.DefaultLoggerConfig; |
| 25 | +import javasabr.rlib.logger.impl.config.render.impl.SimpleLogMessageRender; |
17 | 26 | import javasabr.rlib.logger.impl.config.render.impl.pattern.PatternLogMessageRender; |
18 | 27 | import javasabr.rlib.logger.impl.config.render.impl.pattern.node.DateTimePatternRenderNode; |
19 | 28 | import javasabr.rlib.logger.impl.config.render.impl.pattern.node.LevelPatternRenderNode; |
@@ -100,4 +109,187 @@ void shouldLoadLoggerConfigCorrectly() throws IOException { |
100 | 109 | .returns(false, Logger::warnEnabled) |
101 | 110 | .returns(true, Logger::errorEnabled); |
102 | 111 | } |
| 112 | + |
| 113 | + @Test |
| 114 | + void shouldPreferTestPropertiesOverMainProperties() { |
| 115 | + // given: |
| 116 | + var loader = new PropertyLoggerConfigLoader(); |
| 117 | + var contextClassLoader = new ResourceClassLoader(Map.of( |
| 118 | + PropertyLoggerConfigLoader.FILE_TEST, "logger.level.ROOT=TRACE", |
| 119 | + PropertyLoggerConfigLoader.FILE_MAIN, "logger.level.ROOT=ERROR")); |
| 120 | + |
| 121 | + // when: |
| 122 | + Optional<LoggerConfig> loadedConfig = withContextClassLoader(contextClassLoader, loader::tryToLoad); |
| 123 | + |
| 124 | + // then: |
| 125 | + assertThat(loadedConfig).isPresent(); |
| 126 | + |
| 127 | + // when: |
| 128 | + var loggerService = new DefaultLoggerService(loadedConfig.orElseThrow()); |
| 129 | + DefaultLogger logger = loggerService.getLogger("example.logger"); |
| 130 | + |
| 131 | + // then: |
| 132 | + assertThat(logger) |
| 133 | + .returns(true, Logger::traceEnabled) |
| 134 | + .returns(true, Logger::debugEnabled) |
| 135 | + .returns(true, Logger::infoEnabled) |
| 136 | + .returns(true, Logger::warnEnabled) |
| 137 | + .returns(true, Logger::errorEnabled); |
| 138 | + } |
| 139 | + |
| 140 | + @Test |
| 141 | + void shouldFallbackToMainPropertiesWhenTestMissing() { |
| 142 | + // given: |
| 143 | + var loader = new PropertyLoggerConfigLoader(); |
| 144 | + var contextClassLoader = new ResourceClassLoader(Map.of( |
| 145 | + PropertyLoggerConfigLoader.FILE_MAIN, "logger.level.ROOT=ERROR")); |
| 146 | + |
| 147 | + // when: |
| 148 | + Optional<LoggerConfig> loadedConfig = withContextClassLoader(contextClassLoader, loader::tryToLoad); |
| 149 | + |
| 150 | + // then: |
| 151 | + assertThat(loadedConfig).isPresent(); |
| 152 | + |
| 153 | + // when: |
| 154 | + var loggerService = new DefaultLoggerService(loadedConfig.orElseThrow()); |
| 155 | + DefaultLogger logger = loggerService.getLogger("example.logger"); |
| 156 | + |
| 157 | + // then: |
| 158 | + assertThat(logger) |
| 159 | + .returns(false, Logger::traceEnabled) |
| 160 | + .returns(false, Logger::debugEnabled) |
| 161 | + .returns(false, Logger::infoEnabled) |
| 162 | + .returns(false, Logger::warnEnabled) |
| 163 | + .returns(true, Logger::errorEnabled); |
| 164 | + } |
| 165 | + |
| 166 | + @Test |
| 167 | + void shouldReturnEmptyWhenNoPropertiesFound() { |
| 168 | + // given: |
| 169 | + var loader = new PropertyLoggerConfigLoader(); |
| 170 | + var contextClassLoader = new ResourceClassLoader(Map.of()); |
| 171 | + |
| 172 | + // when: |
| 173 | + Optional<LoggerConfig> loadedConfig = withContextClassLoader(contextClassLoader, loader::tryToLoad); |
| 174 | + |
| 175 | + // then: |
| 176 | + assertThat(loadedConfig).isEmpty(); |
| 177 | + } |
| 178 | + |
| 179 | + @Test |
| 180 | + void shouldUseRootLevelForUnconfiguredLogger() { |
| 181 | + // given: |
| 182 | + var loader = new PropertyLoggerConfigLoader(); |
| 183 | + var contextClassLoader = new ResourceClassLoader(Map.of( |
| 184 | + PropertyLoggerConfigLoader.FILE_MAIN, |
| 185 | + "logger.level.ROOT=WARNING\nlogger.level.configured.logger=TRACE")); |
| 186 | + |
| 187 | + // when: |
| 188 | + Optional<LoggerConfig> loadedConfig = withContextClassLoader(contextClassLoader, loader::tryToLoad); |
| 189 | + |
| 190 | + // then: |
| 191 | + assertThat(loadedConfig).isPresent(); |
| 192 | + |
| 193 | + // when: |
| 194 | + var loggerService = new DefaultLoggerService(loadedConfig.orElseThrow()); |
| 195 | + DefaultLogger unconfiguredLogger = loggerService.getLogger("unknown.logger"); |
| 196 | + DefaultLogger configuredLogger = loggerService.getLogger("configured.logger"); |
| 197 | + |
| 198 | + // then: |
| 199 | + assertThat(unconfiguredLogger) |
| 200 | + .returns(false, Logger::traceEnabled) |
| 201 | + .returns(false, Logger::debugEnabled) |
| 202 | + .returns(false, Logger::infoEnabled) |
| 203 | + .returns(true, Logger::warnEnabled) |
| 204 | + .returns(true, Logger::errorEnabled); |
| 205 | + assertThat(configuredLogger) |
| 206 | + .returns(true, Logger::traceEnabled) |
| 207 | + .returns(true, Logger::debugEnabled) |
| 208 | + .returns(true, Logger::infoEnabled) |
| 209 | + .returns(true, Logger::warnEnabled) |
| 210 | + .returns(true, Logger::errorEnabled); |
| 211 | + } |
| 212 | + |
| 213 | + @Test |
| 214 | + void shouldUseSimpleMessageRenderWhenPatternIsAbsent() { |
| 215 | + // given: |
| 216 | + var loader = new PropertyLoggerConfigLoader(); |
| 217 | + var properties = new Properties(); |
| 218 | + properties.setProperty("logger.level.ROOT", "INFO"); |
| 219 | + |
| 220 | + // when: |
| 221 | + LoggerConfig loggerConfig = loader.loadFromProperties(properties); |
| 222 | + |
| 223 | + // then: |
| 224 | + assertThat(loggerConfig) |
| 225 | + .isInstanceOf(DefaultLoggerConfig.class); |
| 226 | + |
| 227 | + // when: |
| 228 | + var defaultLoggerConfig = (DefaultLoggerConfig) loggerConfig; |
| 229 | + var loggerService = new DefaultLoggerService(loggerConfig); |
| 230 | + DefaultLogger logger = loggerService.getLogger("example.logger"); |
| 231 | + UnsafeArray<LogMessageConsumer> consumers = defaultLoggerConfig.resolveConsumers( |
| 232 | + logger, |
| 233 | + LoggerLevel.TRACE); |
| 234 | + |
| 235 | + // then: |
| 236 | + assertThat(consumers.size()) |
| 237 | + .isEqualTo(1); |
| 238 | + assertThat(consumers.get(0)) |
| 239 | + .isInstanceOf(ConsoleMessageConsumer.class) |
| 240 | + .extracting("logMessageRender") |
| 241 | + .isInstanceOf(SimpleLogMessageRender.class); |
| 242 | + } |
| 243 | + |
| 244 | + @Test |
| 245 | + void shouldThrowExceptionWhenLevelValueIsInvalid() { |
| 246 | + // given: |
| 247 | + var loader = new PropertyLoggerConfigLoader(); |
| 248 | + var properties = new Properties(); |
| 249 | + properties.setProperty("logger.level.ROOT", "INF0"); |
| 250 | + |
| 251 | + // when/then: |
| 252 | + assertThatThrownBy(() -> loader.loadFromProperties(properties)) |
| 253 | + .isInstanceOf(IllegalArgumentException.class); |
| 254 | + } |
| 255 | + |
| 256 | + private static <T> T withContextClassLoader(ClassLoader contextClassLoader, Supplier<T> action) { |
| 257 | + Thread currentThread = Thread.currentThread(); |
| 258 | + ClassLoader previousClassLoader = currentThread.getContextClassLoader(); |
| 259 | + try { |
| 260 | + currentThread.setContextClassLoader(contextClassLoader); |
| 261 | + return action.get(); |
| 262 | + } finally { |
| 263 | + currentThread.setContextClassLoader(previousClassLoader); |
| 264 | + } |
| 265 | + } |
| 266 | + |
| 267 | + private static class ResourceClassLoader extends ClassLoader { |
| 268 | + |
| 269 | + private final Map<String, byte[]> resources; |
| 270 | + |
| 271 | + private ResourceClassLoader(Map<String, String> resources) { |
| 272 | + super(Thread |
| 273 | + .currentThread() |
| 274 | + .getContextClassLoader()); |
| 275 | + this.resources = resources |
| 276 | + .entrySet() |
| 277 | + .stream() |
| 278 | + .collect(Collectors.toUnmodifiableMap( |
| 279 | + Map.Entry::getKey, |
| 280 | + entry -> entry.getValue().getBytes(StandardCharsets.UTF_8))); |
| 281 | + } |
| 282 | + |
| 283 | + @Override |
| 284 | + public InputStream getResourceAsStream(String name) { |
| 285 | + byte[] loaded = resources.get(name); |
| 286 | + if (loaded != null) { |
| 287 | + return new ByteArrayInputStream(loaded); |
| 288 | + } |
| 289 | + if (PropertyLoggerConfigLoader.FILE_TEST.equals(name) || PropertyLoggerConfigLoader.FILE_MAIN.equals(name)) { |
| 290 | + return null; |
| 291 | + } |
| 292 | + return super.getResourceAsStream(name); |
| 293 | + } |
| 294 | + } |
103 | 295 | } |
0 commit comments