|
1 | 1 | package com.google.api.logging; |
2 | 2 |
|
| 3 | +import static org.mockito.ArgumentMatchers.any; |
| 4 | +import static org.mockito.Mockito.mock; |
| 5 | +import static org.mockito.Mockito.verify; |
3 | 6 | import static org.mockito.Mockito.when; |
4 | 7 |
|
5 | 8 | import ch.qos.logback.classic.spi.ILoggingEvent; |
6 | 9 | import com.fasterxml.jackson.core.JsonGenerator; |
7 | | -import java.util.LinkedHashMap; |
| 10 | +import java.io.IOException; |
| 11 | +import java.util.HashMap; |
8 | 12 | import java.util.Map; |
9 | 13 | import org.junit.jupiter.api.BeforeEach; |
10 | | -import org.mockito.Mock; |
| 14 | +import org.junit.jupiter.api.Test; |
11 | 15 |
|
12 | 16 | public class SDKLoggingMdcJsonProviderTest { |
13 | 17 | private SDKLoggingMdcJsonProvider provider = new SDKLoggingMdcJsonProvider(); |
14 | 18 |
|
15 | | - @Mock |
16 | | - private JsonGenerator generator; |
| 19 | + private JsonGenerator generator = mock(JsonGenerator.class); |
17 | 20 |
|
18 | | - @Mock |
19 | | - private ILoggingEvent event; |
| 21 | + private ILoggingEvent event = mock(ILoggingEvent.class); |
20 | 22 |
|
21 | 23 | private Map<String, String> mdc; |
22 | 24 |
|
23 | 25 | @BeforeEach |
24 | 26 | public void setup() { |
25 | | - mdc = new LinkedHashMap<>(); |
26 | | - mdc.put("name1", "value1"); |
27 | | - mdc.put("name2", "value2"); |
28 | | - mdc.put("name3", "value3"); |
| 27 | + mdc = new HashMap<>(); |
| 28 | + mdc.put("json1", "{\n" |
| 29 | + + " \"@version\": \"1\",\n" |
| 30 | + + " \"textPayload\": \"Received response\",\n" |
| 31 | + + " \"response.payload\": {\n" |
| 32 | + + " \"name\": \"example\",\n" |
| 33 | + + " \"state\": \"ACTIVE\"\n" |
| 34 | + + " }\n" |
| 35 | + + "}"); |
29 | 36 | when(event.getMDCPropertyMap()).thenReturn(mdc); |
30 | 37 | } |
| 38 | + |
| 39 | + @Test |
| 40 | + void testUnwrapped() throws IOException { |
| 41 | + |
| 42 | + provider.writeTo(generator, event); |
| 43 | + |
| 44 | + verify(generator).writeFieldName("json1"); |
| 45 | + verify(generator).writeTree(any()); |
| 46 | + } |
31 | 47 | } |
0 commit comments