Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Support component name for general release metrics
Signed-off-by: Prudhvi Godithi <[email protected]>
- Loading branch information
1 parent
8f33f15
commit 310431b
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,144 @@ | ||
| package org.opensearchmetrics.metrics; | ||
|
|
||
| import com.fasterxml.jackson.databind.ObjectMapper; | ||
| import org.junit.jupiter.api.BeforeEach; | ||
| import org.junit.jupiter.api.Test; | ||
| import org.mockito.ArgumentCaptor; | ||
| import org.mockito.ArgumentMatchers; | ||
| import org.mockito.InjectMocks; | ||
| import org.mockito.Mock; | ||
| import org.mockito.Mockito; | ||
| import org.mockito.MockitoAnnotations; | ||
| import org.opensearch.action.search.SearchRequest; | ||
| import org.opensearch.index.query.BoolQueryBuilder; | ||
| import org.opensearchmetrics.metrics.general.*; | ||
| import org.opensearchmetrics.metrics.label.LabelMetrics; | ||
| import org.opensearchmetrics.metrics.release.ReleaseInputs; | ||
| import org.opensearchmetrics.metrics.release.ReleaseMetrics; | ||
| import org.opensearchmetrics.model.label.LabelData; | ||
| import org.opensearchmetrics.model.general.MetricsData; | ||
| import org.opensearchmetrics.model.release.ReleaseMetricsData; | ||
| import org.opensearchmetrics.util.OpenSearchUtil; | ||
|
|
||
| import java.io.IOException; | ||
| import java.security.NoSuchAlgorithmException; | ||
| import java.time.LocalDateTime; | ||
| import java.time.ZoneId; | ||
| import java.util.*; | ||
|
|
||
| import static org.junit.jupiter.api.Assertions.assertArrayEquals; | ||
| import static org.junit.jupiter.api.Assertions.assertEquals; | ||
| import static org.mockito.ArgumentMatchers.any; | ||
| import static org.mockito.Mockito.*; | ||
|
|
||
| public class MetricsCalculationTest { | ||
| @Mock | ||
| private OpenSearchUtil openSearchUtil; | ||
|
|
||
| @Mock | ||
| private ObjectMapper objectMapper; | ||
|
|
||
| @Mock | ||
| private UntriagedIssues untriagedIssues; | ||
| @Mock | ||
| private UncommentedPullRequests uncommentedPullRequests; | ||
| @Mock | ||
| private UnlabelledPullRequests unlabelledPullRequests; | ||
| @Mock | ||
| private UnlabelledIssues unlabelledIssues; | ||
| @Mock | ||
| private MergedPullRequests mergedPullRequests; | ||
| @Mock | ||
| private OpenPullRequests openPullRequests; | ||
| @Mock | ||
| private OpenIssues openIssues; | ||
| @Mock | ||
| private ClosedIssues closedIssues; | ||
| @Mock | ||
| private CreatedIssues createdIssues; | ||
| @Mock | ||
| private IssueComments issueComments; | ||
| @Mock | ||
| private PullComments pullComments; | ||
| @Mock | ||
| private IssuePositiveReactions issuePositiveReactions; | ||
| @Mock | ||
| private IssueNegativeReactions issueNegativeReactions; | ||
| @Mock | ||
| private LabelMetrics labelMetrics; | ||
| @Mock | ||
| private ReleaseMetrics releaseMetrics; | ||
|
|
||
| @InjectMocks | ||
| private MetricsCalculation metricsCalculation; | ||
|
|
||
|
|
||
| @BeforeEach | ||
| void setUp() { | ||
| MockitoAnnotations.openMocks(this); | ||
| metricsCalculation = new MetricsCalculation(openSearchUtil, objectMapper, | ||
| untriagedIssues, uncommentedPullRequests, unlabelledPullRequests, unlabelledIssues, | ||
| mergedPullRequests, openPullRequests, openIssues, closedIssues, createdIssues, | ||
| issueComments, pullComments, issuePositiveReactions, issueNegativeReactions, | ||
| labelMetrics, releaseMetrics); | ||
| } | ||
|
|
||
| @Test | ||
| void testGenerateGeneralMetrics() throws IOException { | ||
| List<String> repositories = Arrays.asList("repo1", "repo2"); | ||
| MetricsData metricsData = new MetricsData(); | ||
| metricsData.setId("id"); | ||
| metricsData.setRepository("repo1"); | ||
| metricsData.setCurrentDate(LocalDateTime.now(ZoneId.of("UTC")).toString()); | ||
| metricsData.setMetricName("metric"); | ||
| metricsData.setMetricCount(10L); | ||
| when(untriagedIssues.getBoolQueryBuilder(any())).thenReturn(new BoolQueryBuilder()); | ||
| when(untriagedIssues.createSearchRequest(any(), any())).thenReturn(new SearchRequest()); | ||
| when(untriagedIssues.performSearch(any(), any())).thenReturn(10L); | ||
| when(objectMapper.writeValueAsString(any())).thenReturn("json"); | ||
| metricsCalculation.generateGeneralMetrics(repositories); | ||
| verify(openSearchUtil).createIndexIfNotExists("opensearch_general_metrics"); | ||
| verify(openSearchUtil).bulkIndex(eq("opensearch_general_metrics"), any(Map.class)); | ||
| } | ||
|
|
||
| @Test | ||
| void testGenerateLabelMetrics() throws IOException { | ||
| List<String> repositories = Arrays.asList("repo1", "repo2"); | ||
| Map<String, List<Long>> labelInfo = new HashMap<>(); | ||
| labelInfo.put("label1", Arrays.asList(5L, 10L)); | ||
| LabelData labelData = new LabelData(); | ||
| labelData.setId("id"); | ||
| labelData.setRepository("repo1"); | ||
| labelData.setCurrentDate(LocalDateTime.now(ZoneId.of("UTC")).toString()); | ||
| labelData.setLabelName("label1"); | ||
| labelData.setLabelPullCount(10L); | ||
| labelData.setLabelIssueCount(5L); | ||
| when(labelMetrics.getLabelInfo(any(), any())).thenReturn(labelInfo); | ||
| when(objectMapper.writeValueAsString(any())).thenReturn("json"); | ||
| metricsCalculation.generateLabelMetrics(repositories); | ||
| verify(openSearchUtil).createIndexIfNotExists("opensearch_label_metrics"); | ||
| verify(openSearchUtil).bulkIndex(eq("opensearch_label_metrics"), any(Map.class)); | ||
| } | ||
|
|
||
| @Test | ||
| void testGenerateReleaseMetrics() { | ||
| Map<String, String> releaseRepos = new HashMap<>(); | ||
| releaseRepos.put("repo1", "component1"); | ||
| releaseRepos.put("repo2", "component2"); | ||
| when(releaseMetrics.getReleaseRepos("2.13.0")).thenReturn(releaseRepos); | ||
| when(releaseMetrics.getReleaseLabelIssues(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "open", false)).thenReturn(10L); | ||
| when(releaseMetrics.getReleaseLabelIssues(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "open", true)).thenReturn(5L); | ||
| when(releaseMetrics.getReleaseLabelIssues(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "closed", false)).thenReturn(20L); | ||
| when(releaseMetrics.getReleaseLabelPulls(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "open")).thenReturn(3L); | ||
| when(releaseMetrics.getReleaseLabelPulls(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "closed")).thenReturn(8L); | ||
| when(releaseMetrics.getReleaseVersionIncrement(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "main")).thenReturn(true); | ||
| when(releaseMetrics.getReleaseNotes(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1", "main")).thenReturn(true); | ||
| when(releaseMetrics.getReleaseBranch(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1")).thenReturn(true); | ||
| when(releaseMetrics.getReleaseOwners(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1")).thenReturn(new String[]{"owner1", "owner2"}); | ||
| when(releaseMetrics.getReleaseIssue(ReleaseInputs.VERSION_2_13_0.getVersion(), "repo1")).thenReturn("release-123"); | ||
| metricsCalculation.generateReleaseMetrics(); | ||
| verify(openSearchUtil).createIndexIfNotExists("opensearch_release_metrics"); | ||
| verify(openSearchUtil).bulkIndex(eq("opensearch_release_metrics"), ArgumentMatchers.anyMap()); | ||
| verify(openSearchUtil, times(1)).createIndexIfNotExists("opensearch_release_metrics"); | ||
| } | ||
| } |
Oops, something went wrong.