Skip to content

Commit

Permalink
fixed findingsDto bug (#102)
Browse files Browse the repository at this point in the history
Signed-off-by: Petar Dzepina <petar.dzepina@gmail.com>

Signed-off-by: Petar Dzepina <petar.dzepina@gmail.com>
  • Loading branch information
petardz committed Nov 3, 2022
1 parent 7de91b5 commit 1ef30c8
Show file tree
Hide file tree
Showing 2 changed files with 72 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ public FindingDto(
Instant timestamp,
List<FindingDocument> documents
) {
this.detectorId = id;
this.detectorId = detectorId;
this.id = id;
this.relatedDocIds = relatedDocIds;
this.index = index;
Expand Down Expand Up @@ -89,4 +89,31 @@ public void writeTo(StreamOutput out) throws IOException {
out.writeList(documents);
}

public String getId() {
return id;
}

public List<String> getRelatedDocIds() {
return relatedDocIds;
}

public String getIndex() {
return index;
}

public List<DocLevelQuery> getDocLevelQueries() {
return docLevelQueries;
}

public Instant getTimestamp() {
return timestamp;
}

public List<FindingDocument> getDocuments() {
return documents;
}

public String getDetectorId() {
return detectorId;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
/*
* Copyright OpenSearch Contributors
* SPDX-License-Identifier: Apache-2.0
*/
package org.opensearch.securityanalytics.findings;

import java.time.Instant;
import java.util.List;
import org.opensearch.commons.alerting.model.DocLevelQuery;
import org.opensearch.commons.alerting.model.FindingDocument;
import org.opensearch.securityanalytics.action.FindingDto;
import org.opensearch.test.OpenSearchTestCase;

public class FindingDtoTests extends OpenSearchTestCase {


public void testFindingDTO_creation() {

FindingDocument findingDocument1 = new FindingDocument("test_index1", "doc1", true, "document 1 payload");
FindingDocument findingDocument2 = new FindingDocument("test_index1", "doc2", true, "document 2 payload");
FindingDocument findingDocument3 = new FindingDocument("test_index1", "doc3", true, "document 3 payload");

Instant now = Instant.now();

FindingDto findingDto = new FindingDto(
"detectorId",
"findingId",
List.of("doc1", "doc2", "doc3"),
"my_index",
List.of(new DocLevelQuery("1","myQuery","fieldA:valABC", List.of())),
now,
List.of(findingDocument1, findingDocument2, findingDocument3)
);

assertEquals("detectorId", findingDto.getDetectorId());
assertEquals("findingId", findingDto.getId());
assertEquals(List.of("doc1", "doc2", "doc3"), findingDto.getRelatedDocIds());
assertEquals("my_index", findingDto.getIndex());
assertEquals(List.of(new DocLevelQuery("1","myQuery","fieldA:valABC", List.of())), findingDto.getDocLevelQueries());
assertEquals(now, findingDto.getTimestamp());
assertEquals(List.of(findingDocument1, findingDocument2, findingDocument3), findingDto.getDocuments());
}

}

0 comments on commit 1ef30c8

Please sign in to comment.