|
1 | 1 | package io.javaoperatorsdk.operator.processing.dependent.kubernetes;
|
2 | 2 |
|
3 |
| -import java.util.Arrays; |
4 | 3 | import java.util.HashMap;
|
5 | 4 | import java.util.List;
|
6 | 5 | import java.util.Map;
|
7 | 6 |
|
8 | 7 | import org.junit.jupiter.api.BeforeEach;
|
9 | 8 | import org.junit.jupiter.api.Test;
|
| 9 | +import org.junit.jupiter.params.ParameterizedTest; |
| 10 | +import org.junit.jupiter.params.provider.ValueSource; |
10 | 11 |
|
11 | 12 | import io.fabric8.kubernetes.api.model.ConfigMap;
|
12 | 13 | import io.fabric8.kubernetes.api.model.HasMetadata;
|
13 | 14 | import io.fabric8.kubernetes.api.model.apps.Deployment;
|
| 15 | +import io.fabric8.kubernetes.api.model.apps.StatefulSet; |
14 | 16 | import io.javaoperatorsdk.operator.MockKubernetesClient;
|
15 | 17 | import io.javaoperatorsdk.operator.ReconcilerUtils;
|
16 | 18 | import io.javaoperatorsdk.operator.api.config.ConfigurationService;
|
|
23 | 25 |
|
24 | 26 | class SSABasedGenericKubernetesResourceMatcherTest {
|
25 | 27 |
|
26 |
| - Context<?> mockedContext = mock(Context.class); |
| 28 | + private final Context<?> mockedContext = mock(); |
27 | 29 |
|
28 |
| - SSABasedGenericKubernetesResourceMatcher<HasMetadata> matcher = |
29 |
| - new SSABasedGenericKubernetesResourceMatcher<>(); |
| 30 | + private final SSABasedGenericKubernetesResourceMatcher<HasMetadata> matcher = |
| 31 | + SSABasedGenericKubernetesResourceMatcher.getInstance(); |
30 | 32 |
|
31 | 33 | @BeforeEach
|
32 | 34 | @SuppressWarnings("unchecked")
|
33 | 35 | void setup() {
|
34 |
| - var controllerConfiguration = mock(ControllerConfiguration.class); |
35 |
| - when(controllerConfiguration.fieldManager()).thenReturn("controller"); |
36 |
| - var configurationService = mock(ConfigurationService.class); |
37 |
| - |
38 | 36 | final var client = MockKubernetesClient.client(HasMetadata.class);
|
39 | 37 | when(mockedContext.getClient()).thenReturn(client);
|
40 | 38 |
|
| 39 | + final var configurationService = mock(ConfigurationService.class); |
| 40 | + final var controllerConfiguration = mock(ControllerConfiguration.class); |
41 | 41 | when(controllerConfiguration.getConfigurationService()).thenReturn(configurationService);
|
| 42 | + when(controllerConfiguration.fieldManager()).thenReturn("controller"); |
42 | 43 | when(mockedContext.getControllerConfiguration()).thenReturn(controllerConfiguration);
|
43 | 44 | }
|
44 | 45 |
|
@@ -119,52 +120,78 @@ void addedLabelInDesiredMakesMatchFail() {
|
119 | 120 | assertThat(matcher.matches(actualConfigMap, desiredConfigMap, mockedContext)).isFalse();
|
120 | 121 | }
|
121 | 122 |
|
122 |
| - private <R> R loadResource(String fileName, Class<R> clazz) { |
123 |
| - return ReconcilerUtils.loadYaml(clazz, SSABasedGenericKubernetesResourceMatcherTest.class, |
124 |
| - fileName); |
125 |
| - } |
126 |
| - |
127 | 123 | @Test
|
128 | 124 | @SuppressWarnings("unchecked")
|
129 | 125 | void sortListItemsTest() {
|
130 |
| - Map<String, Object> nestedMap1 = new HashMap<>(); |
| 126 | + var nestedMap1 = new HashMap<String, Object>(); |
131 | 127 | nestedMap1.put("z", 26);
|
132 | 128 | nestedMap1.put("y", 25);
|
133 | 129 |
|
134 |
| - Map<String, Object> nestedMap2 = new HashMap<>(); |
| 130 | + var nestedMap2 = new HashMap<String, Object>(); |
135 | 131 | nestedMap2.put("b", 26);
|
136 | 132 | nestedMap2.put("c", 25);
|
137 | 133 | nestedMap2.put("a", 24);
|
138 | 134 |
|
139 |
| - List<Object> unsortedListItems = Arrays.asList(1, nestedMap1, nestedMap2); |
140 |
| - List<Object> sortedListItems = matcher.sortListItems(unsortedListItems); |
141 |
| - |
| 135 | + var unsortedListItems = List.<Object>of(1, nestedMap1, nestedMap2); |
| 136 | + var sortedListItems = matcher.sortListItems(unsortedListItems); |
142 | 137 | assertThat(sortedListItems).element(0).isEqualTo(1);
|
143 | 138 |
|
144 |
| - Map<String, Object> sortedNestedMap1 = (Map<String, Object>) sortedListItems.get(1); |
| 139 | + var sortedNestedMap1 = (Map<String, Object>) sortedListItems.get(1); |
145 | 140 | assertThat(sortedNestedMap1.keySet()).containsExactly("y", "z");
|
146 | 141 |
|
147 |
| - Map<String, Object> sortedNestedMap2 = (Map<String, Object>) sortedListItems.get(2); |
| 142 | + var sortedNestedMap2 = (Map<String, Object>) sortedListItems.get(2); |
148 | 143 | assertThat(sortedNestedMap2.keySet()).containsExactly("a", "b", "c");
|
149 | 144 | }
|
150 | 145 |
|
151 | 146 | @Test
|
152 | 147 | @SuppressWarnings("unchecked")
|
153 | 148 | void testSortMapWithNestedMap() {
|
154 |
| - Map<String, Object> nestedMap = new HashMap<>(); |
| 149 | + var nestedMap = new HashMap<String, Object>(); |
155 | 150 | nestedMap.put("z", 26);
|
156 | 151 | nestedMap.put("y", 25);
|
157 | 152 |
|
158 |
| - Map<String, Object> unsortedMap = new HashMap<>(); |
| 153 | + var unsortedMap = new HashMap<String, Object>(); |
159 | 154 | unsortedMap.put("b", nestedMap);
|
160 | 155 | unsortedMap.put("a", 1);
|
161 | 156 | unsortedMap.put("c", 2);
|
162 | 157 |
|
163 |
| - Map<String, Object> sortedMap = matcher.sortMap(unsortedMap); |
164 |
| - |
| 158 | + var sortedMap = matcher.sortMap(unsortedMap); |
165 | 159 | assertThat(sortedMap.keySet()).containsExactly("a", "b", "c");
|
166 | 160 |
|
167 |
| - Map<String, Object> sortedNestedMap = (Map<String, Object>) sortedMap.get("b"); |
| 161 | + var sortedNestedMap = (Map<String, Object>) sortedMap.get("b"); |
168 | 162 | assertThat(sortedNestedMap.keySet()).containsExactly("y", "z");
|
169 | 163 | }
|
| 164 | + |
| 165 | + @ParameterizedTest |
| 166 | + @ValueSource(strings = {"sample-sts-volumeclaimtemplates-desired.yaml", |
| 167 | + "sample-sts-volumeclaimtemplates-desired-with-status.yaml", |
| 168 | + "sample-sts-volumeclaimtemplates-desired-with-volumemode.yaml"}) |
| 169 | + void testSanitizeState_statefulSetWithVolumeClaims(String desiredResourceFileName) { |
| 170 | + var desiredStatefulSet = loadResource(desiredResourceFileName, StatefulSet.class); |
| 171 | + var actualStatefulSet = loadResource("sample-sts-volumeclaimtemplates.yaml", |
| 172 | + StatefulSet.class); |
| 173 | + |
| 174 | + assertThat(matcher.matches(actualStatefulSet, desiredStatefulSet, mockedContext)).isTrue(); |
| 175 | + } |
| 176 | + |
| 177 | + @ParameterizedTest |
| 178 | + @ValueSource(strings = {"sample-sts-volumeclaimtemplates-desired-add.yaml", |
| 179 | + "sample-sts-volumeclaimtemplates-desired-update.yaml", |
| 180 | + "sample-sts-volumeclaimtemplates-desired-with-status-mismatch.yaml", |
| 181 | + "sample-sts-volumeclaimtemplates-desired-with-volumemode-mismatch.yaml"}) |
| 182 | + void testSanitizeState_statefulSetWithVolumeClaims_withMismatch(String desiredResourceFileName) { |
| 183 | + var desiredStatefulSet = loadResource(desiredResourceFileName, StatefulSet.class); |
| 184 | + var actualStatefulSet = loadResource("sample-sts-volumeclaimtemplates.yaml", |
| 185 | + StatefulSet.class); |
| 186 | + |
| 187 | + assertThat(matcher.matches(actualStatefulSet, desiredStatefulSet, mockedContext)).isFalse(); |
| 188 | + } |
| 189 | + |
| 190 | + assertThat(matcher.matches(actualStatefulSet, desiredStatefulSet, mockedContext)).isTrue(); |
| 191 | + } |
| 192 | + |
| 193 | + private static <R> R loadResource(String fileName, Class<R> clazz) { |
| 194 | + return ReconcilerUtils.loadYaml(clazz, SSABasedGenericKubernetesResourceMatcherTest.class, |
| 195 | + fileName); |
| 196 | + } |
170 | 197 | }
|
0 commit comments