Skip to content

Commit 2ef5812

Browse files
committed
test fix
1 parent 3075212 commit 2ef5812

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

plugins/amazonq/codewhisperer/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererTelemetryServiceTest.kt

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ class CodeWhispererTelemetryServiceTest {
154154
@Test
155155
fun `test aggregateUserDecision`() {
156156
fun assertAggregateUserDecision(decisions: List<CodewhispererSuggestionState>, expected: CodewhispererSuggestionState) {
157-
val actual = sut.aggregateUserDecision(decisions)
157+
val actual = sut.aggregateUserDecision(decisions, false)
158158
assertThat(actual).isEqualTo(expected)
159159
}
160160

@@ -202,6 +202,25 @@ class CodeWhispererTelemetryServiceTest {
202202
)
203203
}
204204

205+
@Test
206+
fun `test aggregateUserDecision when there was a previous reject`() {
207+
fun assertAggregateUserDecision(decisions: List<CodewhispererSuggestionState>, expected: CodewhispererSuggestionState) {
208+
val actual = sut.aggregateUserDecision(decisions, true)
209+
assertThat(actual).isEqualTo(expected)
210+
}
211+
212+
assertAggregateUserDecision(
213+
listOf(
214+
CodewhispererSuggestionState.Discard,
215+
CodewhispererSuggestionState.Discard,
216+
CodewhispererSuggestionState.Reject,
217+
CodewhispererSuggestionState.Empty,
218+
CodewhispererSuggestionState.Ignore
219+
),
220+
CodewhispererSuggestionState.Ignore
221+
)
222+
}
223+
205224
@Test
206225
fun `sendUserTriggerDecisionEvent`() {
207226
val timeSinceDocumentChanged = Random.nextDouble(0.0, 1000.0)

0 commit comments

Comments
 (0)