Skip to content
This repository has been archived by the owner on Aug 2, 2022. It is now read-only.

Fix another case of the profile API returns prematurely #353

Merged
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,6 @@ private void prepareProfile(
CommonErrorMessages.FAIL_FETCH_ERR_MSG + detectorId,
false
);

if (profilesToCollect.contains(DetectorProfileName.ERROR)) {
GetRequest getStateRequest = new GetRequest(DetectorInternalState.DETECTOR_STATE_INDEX, detectorId);
client.get(getStateRequest, onGetDetectorState(delegateListener, detectorId, enabledTimeMs));
Expand Down Expand Up @@ -459,8 +458,8 @@ private ActionListener<SearchResponse> onInittedEver(
processInitResponse(detector, profilesToCollect, totalUpdates, false, profileBuilder, listener);
} else {
createRunningStateAndInitProgress(profilesToCollect, profileBuilder);
listener.onResponse(profileBuilder.build());
}
listener.onResponse(profileBuilder.build());
}, exception -> {
if (exception instanceof IndexNotFoundException) {
// anomaly result index is not created yet
Expand Down Expand Up @@ -554,7 +553,6 @@ private void processInitResponse(
} else {
long intervalMins = ((IntervalTimeConfiguration) detector.getDetectionInterval()).toDuration().toMinutes();
InitProgressProfile initProgress = computeInitProgressProfile(totalUpdates, intervalMins);

builder.initProgress(initProgress);
}
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

import org.elasticsearch.action.ActionListener;
import org.elasticsearch.action.get.GetRequest;
Expand Down Expand Up @@ -111,6 +112,7 @@ private void setUpMultiEntityClientGet(DetectorStatus detectorStatus, JobStatus
} else if (request.index().equals(DetectorInternalState.DETECTOR_STATE_INDEX)) {
switch (errorResultStatus) {
case NO_ERROR:
listener.onResponse(null);
break;
case NULL_POINTER_EXCEPTION:
GetResponse response = mock(GetResponse.class);
Expand Down Expand Up @@ -237,7 +239,29 @@ public void testFailGetState() throws IOException, InterruptedException {
assertTrue(inProgressLatch.await(100, TimeUnit.SECONDS));
}

public void testFaiConfirmInitted() throws IOException, InterruptedException {
public void testNoResultsNoError() throws IOException, InterruptedException {
setUpMultiEntityClientGet(DetectorStatus.EXIST, JobStatus.ENABLED, ErrorResultStatus.NO_ERROR);
setUpMultiEntityClientSearch(ADResultStatus.NO_RESULT, CardinalityStatus.NORMAL);
setUpProfileAction();

final AtomicInteger called = new AtomicInteger(0);

runner.profile(detector.getDetectorId(), ActionListener.wrap(response -> {
assertTrue(response.getInitProgress() != null);
called.getAndIncrement();
}, exception -> {
assertTrue("Should not reach here ", false);
called.getAndIncrement();
}), totalInitProgress);

while (called.get() == 0) {
Thread.sleep(100);
}
// should only call onResponse once
assertEquals(1, called.get());
}

public void testFailConfirmInitted() throws IOException, InterruptedException {
setUpMultiEntityClientGet(DetectorStatus.EXIST, JobStatus.ENABLED, ErrorResultStatus.NO_ERROR);
setUpMultiEntityClientSearch(ADResultStatus.EXCEPTION, CardinalityStatus.NORMAL);
setUpProfileAction();
Expand Down