Skip to content

Commit 99c3437

Browse files
Will-ShaoHuaWill-ShaoHua
committed
dedupe repeated code
1 parent 3840926 commit 99c3437

File tree

1 file changed

+53
-87
lines changed

1 file changed

+53
-87
lines changed

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

Lines changed: 53 additions & 87 deletions
Original file line numberDiff line numberDiff line change
@@ -72,37 +72,23 @@ class CodeWhispererFileContextProviderTest {
7272
@Test
7373
fun `should use open tabs if project context is empty due to project context is disabled`() = runTest {
7474
sut = spy(sut)
75-
val file1 = "Human machine interface for lab abc computer applications"
76-
val file2 = "A survey of user opinion of computer system response time"
77-
val file3 = "The EPS user interface management system"
78-
val file4 = "System and human system engineering testing of EPS"
79-
val file5 = "Relation of user perceived response time to error measurement"
80-
val file6 = "The generation of random binary unordered trees"
81-
val file7 = "The intersection graph of paths in trees"
82-
val file8 = "Graph minors IV Widths of trees and well quasi ordering"
83-
val file9 = "Graph minors A survey"
84-
val query = "The intersection of graph survey and trees"
85-
86-
val queryPsi = fixture.addFileToProject("Query.java", query)
87-
val file1Psi = fixture.addFileToProject("File1.java", file1)
88-
val file2Psi = fixture.addFileToProject("File2.java", file2)
89-
val file3Psi = fixture.addFileToProject("File3.java", file3)
90-
val file4Psi = fixture.addFileToProject("File4.java", file4)
91-
val file5Psi = fixture.addFileToProject("File5.java", file5)
92-
val file6Psi = fixture.addFileToProject("File6.java", file6)
93-
val file7Psi = fixture.addFileToProject("File7.java", file7)
94-
val file8Psi = fixture.addFileToProject("File8.java", file8)
95-
val file9Psi = fixture.addFileToProject("File9.java", file9)
75+
val queryPsi = fixture.addFileToProject("Query.java", sampleCase.query)
76+
val file1Psi = fixture.addFileToProject("File1.java", sampleCase.file1)
77+
val file2Psi = fixture.addFileToProject("File2.java", sampleCase.file2)
78+
val file3Psi = fixture.addFileToProject("File3.java", sampleCase.file3)
79+
val file4Psi = fixture.addFileToProject("File4.java", sampleCase.file4)
80+
val file5Psi = fixture.addFileToProject("File5.java", sampleCase.file5)
81+
val file6Psi = fixture.addFileToProject("File6.java", sampleCase.file6)
82+
val file7Psi = fixture.addFileToProject("File7.java", sampleCase.file7)
83+
val file8Psi = fixture.addFileToProject("File8.java", sampleCase.file8)
84+
val file9Psi = fixture.addFileToProject("File9.java", sampleCase.file9)
9685

9786
runInEdtAndWait {
9887
fixture.openFileInEditor(file1Psi.viewProvider.virtualFile)
9988
fixture.openFileInEditor(file2Psi.viewProvider.virtualFile)
10089
fixture.openFileInEditor(file3Psi.viewProvider.virtualFile)
101-
10290
fixture.openFileInEditor(file4Psi.viewProvider.virtualFile)
103-
10491
fixture.openFileInEditor(file5Psi.viewProvider.virtualFile)
105-
10692
fixture.openFileInEditor(file6Psi.viewProvider.virtualFile)
10793
fixture.openFileInEditor(file7Psi.viewProvider.virtualFile)
10894
fixture.openFileInEditor(file8Psi.viewProvider.virtualFile)
@@ -124,27 +110,16 @@ class CodeWhispererFileContextProviderTest {
124110
@Test
125111
fun `should use open tabs if project context is empty due to unknown error`() = runTest {
126112
sut = spy(sut)
127-
val file1 = "Human machine interface for lab abc computer applications"
128-
val file2 = "A survey of user opinion of computer system response time"
129-
val file3 = "The EPS user interface management system"
130-
val file4 = "System and human system engineering testing of EPS"
131-
val file5 = "Relation of user perceived response time to error measurement"
132-
val file6 = "The generation of random binary unordered trees"
133-
val file7 = "The intersection graph of paths in trees"
134-
val file8 = "Graph minors IV Widths of trees and well quasi ordering"
135-
val file9 = "Graph minors A survey"
136-
val query = "The intersection of graph survey and trees"
137-
138-
val queryPsi = fixture.addFileToProject("Query.java", query)
139-
val file1Psi = fixture.addFileToProject("File1.java", file1)
140-
val file2Psi = fixture.addFileToProject("File2.java", file2)
141-
val file3Psi = fixture.addFileToProject("File3.java", file3)
142-
val file4Psi = fixture.addFileToProject("File4.java", file4)
143-
val file5Psi = fixture.addFileToProject("File5.java", file5)
144-
val file6Psi = fixture.addFileToProject("File6.java", file6)
145-
val file7Psi = fixture.addFileToProject("File7.java", file7)
146-
val file8Psi = fixture.addFileToProject("File8.java", file8)
147-
val file9Psi = fixture.addFileToProject("File9.java", file9)
113+
val queryPsi = fixture.addFileToProject("Query.java", sampleCase.query)
114+
val file1Psi = fixture.addFileToProject("File1.java", sampleCase.file1)
115+
val file2Psi = fixture.addFileToProject("File2.java", sampleCase.file2)
116+
val file3Psi = fixture.addFileToProject("File3.java", sampleCase.file3)
117+
val file4Psi = fixture.addFileToProject("File4.java", sampleCase.file4)
118+
val file5Psi = fixture.addFileToProject("File5.java", sampleCase.file5)
119+
val file6Psi = fixture.addFileToProject("File6.java", sampleCase.file6)
120+
val file7Psi = fixture.addFileToProject("File7.java", sampleCase.file7)
121+
val file8Psi = fixture.addFileToProject("File8.java", sampleCase.file8)
122+
val file9Psi = fixture.addFileToProject("File9.java", sampleCase.file9)
148123

149124
runInEdtAndWait {
150125
fixture.openFileInEditor(file1Psi.viewProvider.virtualFile)
@@ -178,27 +153,16 @@ class CodeWhispererFileContextProviderTest {
178153
@Test
179154
fun `should use project context if it is present`() = runTest {
180155
sut = spy(sut)
181-
val file1 = "Human machine interface for lab abc computer applications"
182-
val file2 = "A survey of user opinion of computer system response time"
183-
val file3 = "The EPS user interface management system"
184-
val file4 = "System and human system engineering testing of EPS"
185-
val file5 = "Relation of user perceived response time to error measurement"
186-
val file6 = "The generation of random binary unordered trees"
187-
val file7 = "The intersection graph of paths in trees"
188-
val file8 = "Graph minors IV Widths of trees and well quasi ordering"
189-
val file9 = "Graph minors A survey"
190-
val query = "The intersection of graph survey and trees"
191-
192-
val queryPsi = fixture.addFileToProject("Query.java", query)
193-
val file1Psi = fixture.addFileToProject("File1.java", file1)
194-
val file2Psi = fixture.addFileToProject("File2.java", file2)
195-
val file3Psi = fixture.addFileToProject("File3.java", file3)
196-
val file4Psi = fixture.addFileToProject("File4.java", file4)
197-
val file5Psi = fixture.addFileToProject("File5.java", file5)
198-
val file6Psi = fixture.addFileToProject("File6.java", file6)
199-
val file7Psi = fixture.addFileToProject("File7.java", file7)
200-
val file8Psi = fixture.addFileToProject("File8.java", file8)
201-
val file9Psi = fixture.addFileToProject("File9.java", file9)
156+
val queryPsi = fixture.addFileToProject("Query.java", sampleCase.query)
157+
val file1Psi = fixture.addFileToProject("File1.java", sampleCase.file1)
158+
val file2Psi = fixture.addFileToProject("File2.java", sampleCase.file2)
159+
val file3Psi = fixture.addFileToProject("File3.java", sampleCase.file3)
160+
val file4Psi = fixture.addFileToProject("File4.java", sampleCase.file4)
161+
val file5Psi = fixture.addFileToProject("File5.java", sampleCase.file5)
162+
val file6Psi = fixture.addFileToProject("File6.java", sampleCase.file6)
163+
val file7Psi = fixture.addFileToProject("File7.java", sampleCase.file7)
164+
val file8Psi = fixture.addFileToProject("File8.java", sampleCase.file8)
165+
val file9Psi = fixture.addFileToProject("File9.java", sampleCase.file9)
202166

203167
runInEdtAndWait {
204168
fixture.openFileInEditor(file1Psi.viewProvider.virtualFile)
@@ -426,27 +390,16 @@ class CodeWhispererFileContextProviderTest {
426390
// TODO: fix this test, in test env, psiFile.virtualFile == null @psiGist.getFileData(psiFile) { psiFile -> ... }
427391
@Test
428392
fun `extractSupplementalFileContext from src file should extract src`() = runTest {
429-
val file1 = "Human machine interface for lab abc computer applications"
430-
val file2 = "A survey of user opinion of computer system response time"
431-
val file3 = "The EPS user interface management system"
432-
val file4 = "System and human system engineering testing of EPS"
433-
val file5 = "Relation of user perceived response time to error measurement"
434-
val file6 = "The generation of random binary unordered trees"
435-
val file7 = "The intersection graph of paths in trees"
436-
val file8 = "Graph minors IV Widths of trees and well quasi ordering"
437-
val file9 = "Graph minors A survey"
438-
val query = "The intersection of graph survey and trees"
439-
440-
val queryPsi = fixture.addFileToProject("Query.java", query)
441-
val file1Psi = fixture.addFileToProject("File1.java", file1)
442-
val file2Psi = fixture.addFileToProject("File2.java", file2)
443-
val file3Psi = fixture.addFileToProject("File3.java", file3)
444-
val file4Psi = fixture.addFileToProject("File4.java", file4)
445-
val file5Psi = fixture.addFileToProject("File5.java", file5)
446-
val file6Psi = fixture.addFileToProject("File6.java", file6)
447-
val file7Psi = fixture.addFileToProject("File7.java", file7)
448-
val file8Psi = fixture.addFileToProject("File8.java", file8)
449-
val file9Psi = fixture.addFileToProject("File9.java", file9)
393+
val queryPsi = fixture.addFileToProject("Query.java", sampleCase.query)
394+
val file1Psi = fixture.addFileToProject("File1.java", sampleCase.file1)
395+
val file2Psi = fixture.addFileToProject("File2.java", sampleCase.file2)
396+
val file3Psi = fixture.addFileToProject("File3.java", sampleCase.file3)
397+
val file4Psi = fixture.addFileToProject("File4.java", sampleCase.file4)
398+
val file5Psi = fixture.addFileToProject("File5.java", sampleCase.file5)
399+
val file6Psi = fixture.addFileToProject("File6.java", sampleCase.file6)
400+
val file7Psi = fixture.addFileToProject("File7.java", sampleCase.file7)
401+
val file8Psi = fixture.addFileToProject("File8.java", sampleCase.file8)
402+
val file9Psi = fixture.addFileToProject("File9.java", sampleCase.file9)
450403

451404
runInEdtAndWait {
452405
fixture.openFileInEditor(file1Psi.viewProvider.virtualFile)
@@ -568,3 +521,16 @@ class CodeWhispererFileContextProviderTest {
568521
""".trimMargin()
569522
}
570523
}
524+
525+
private object sampleCase {
526+
val file1 = "Human machine interface for lab abc computer applications"
527+
val file2 = "A survey of user opinion of computer system response time"
528+
val file3 = "The EPS user interface management system"
529+
val file4 = "System and human system engineering testing of EPS"
530+
val file5 = "Relation of user perceived response time to error measurement"
531+
val file6 = "The generation of random binary unordered trees"
532+
val file7 = "The intersection graph of paths in trees"
533+
val file8 = "Graph minors IV Widths of trees and well quasi ordering"
534+
val file9 = "Graph minors A survey"
535+
val query = "The intersection of graph survey and trees"
536+
}

0 commit comments

Comments
 (0)