1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
package org.onap.aai.sparky.sync;
import static org.junit.Assert.*;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;
import org.onap.aai.restclient.client.OperationResult;
import org.onap.aai.sparky.dal.ElasticSearchAdapter;
import org.onap.aai.sparky.sync.config.ElasticSearchEndpointConfig;
import org.onap.aai.sparky.sync.config.ElasticSearchSchemaConfig;
import org.onap.aai.sparky.sync.enumeration.OperationState;
import org.onap.aai.sparky.util.TestResourceLoader;
public class ElasticSearchIndexCleanerTest {
private ElasticSearchIndexCleaner esIndexCleaner;
private ElasticSearchAdapter esAdapter;
private ElasticSearchEndpointConfig esRestEndpointConfig;
private ElasticSearchSchemaConfig esSchemaConfig;
@Before
public void init() throws Exception {
esAdapter = Mockito.mock( ElasticSearchAdapter.class);
esRestEndpointConfig = new ElasticSearchEndpointConfig();
esSchemaConfig = new ElasticSearchSchemaConfig();
esRestEndpointConfig.setScrollContextBatchRequestSize(5000);
esRestEndpointConfig.setEsIpAddress("127.0.0.1");
esRestEndpointConfig.setEsServerPort("9200");
esRestEndpointConfig.setScrollContextTimeToLiveInMinutes(5);
}
@Test
public void validateBasicConstruction() throws Exception {
esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
}
@Test
public void validatePreOperationCollection() throws Exception {
String beforeSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
"/sync/ElasticSearch/BeforeSync_ElasticSearch_ScrollApi_Successful.json");
OperationResult scrollApiOpResult = new OperationResult();
scrollApiOpResult.setResultCode(200);
scrollApiOpResult.setResult(beforeSyncScrollApiResponse);
Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
.thenReturn(scrollApiOpResult);
esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
OperationState opState = esIndexCleaner.populatePreOperationCollection();
assertEquals(OperationState.OK, opState);
}
@Test
public void validatePostOperationCollection() throws Exception {
String afterSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
"/sync/ElasticSearch/AfterSync_ElasticSearch_ScrollApi_Successful.json");
OperationResult scrollApiOpResult = new OperationResult();
scrollApiOpResult.setResultCode(200);
scrollApiOpResult.setResult(afterSyncScrollApiResponse);
Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
.thenReturn(scrollApiOpResult);
esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
OperationState opState = esIndexCleaner.populatePostOperationCollection();
assertEquals(OperationState.OK, opState);
}
@Test
public void validatePerformCleanup() throws Exception {
String beforeSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
"/sync/ElasticSearch/BeforeSync_ElasticSearch_ScrollApi_Successful.json");
OperationResult beforeScrollApiOpResult = new OperationResult();
beforeScrollApiOpResult.setResultCode(200);
beforeScrollApiOpResult.setResult(beforeSyncScrollApiResponse);
String afterSyncScrollApiResponse = TestResourceLoader.getTestResourceDataJson(
"/sync/ElasticSearch/AfterSync_ElasticSearch_ScrollApi_Successful.json");
OperationResult afterScrollApiOpResult = new OperationResult();
afterScrollApiOpResult.setResultCode(200);
afterScrollApiOpResult.setResult(afterSyncScrollApiResponse);
Mockito.when(esAdapter.doPost(Mockito.anyString(), Mockito.anyString(), Mockito.anyObject()))
.thenReturn(beforeScrollApiOpResult,afterScrollApiOpResult);
esIndexCleaner = new ElasticSearchIndexCleaner(esAdapter, esRestEndpointConfig, esSchemaConfig);
OperationState beforeOpState = esIndexCleaner.populatePreOperationCollection();
OperationState afterOpState = esIndexCleaner.populatePostOperationCollection();
assertEquals(OperationState.OK, beforeOpState);
assertEquals(OperationState.OK, afterOpState);
/*
* Now we can start the test work
*/
OperationState cleanupState = esIndexCleaner.performCleanup();
assertEquals(OperationState.OK, cleanupState);
}
}
|