@@ -23,7 +23,12 @@ describe('getAutocompleteSuggestions', () => {
2323 const suggestionsAfterSpace = getAutocompleteSuggestions (
2424 'WorkflowID = "test" AND c'
2525 ) ;
26- expect ( suggestionsAfterSpace ) . toEqual ( [ 'CloseTime' , 'CloseStatus' ] ) ;
26+ expect ( suggestionsAfterSpace ) . toEqual ( [
27+ 'CloseTime' ,
28+ 'ClusterAttributeScope' ,
29+ 'ClusterAttributeName' ,
30+ 'CloseStatus' ,
31+ ] ) ;
2732 } ) ;
2833
2934 it ( 'suggests logical operators after a complete WorkflowID value' , ( ) => {
@@ -67,6 +72,16 @@ describe('getAutocompleteSuggestions', () => {
6772 expect ( suggestionsRunID ) . toEqual ( [ '""' ] ) ;
6873 } ) ;
6974
75+ it ( 'suggests empty quotes after ClusterAttributeScope' , ( ) => {
76+ const suggestions = getAutocompleteSuggestions ( 'ClusterAttributeScope =' ) ;
77+ expect ( suggestions ) . toEqual ( [ '""' ] ) ;
78+ } ) ;
79+
80+ it ( 'suggests empty quotes after ClusterAttributeName' , ( ) => {
81+ const suggestions = getAutocompleteSuggestions ( 'ClusterAttributeName !=' ) ;
82+ expect ( suggestions ) . toEqual ( [ '""' ] ) ;
83+ } ) ;
84+
7085 it ( 'suggests status values after CloseStatus' , ( ) => {
7186 const suggestions = getAutocompleteSuggestions ( 'CloseStatus =' ) ;
7287 expect ( suggestions ) . toEqual ( STATUSES ) ;
@@ -105,6 +120,12 @@ describe('getAutocompleteSuggestions', () => {
105120 expect ( suggestionsCase ) . toContain ( 'CloseTime' ) ;
106121 } ) ;
107122
123+ it ( 'handles partial attribute matching for Cluster attributes' , ( ) => {
124+ const suggestions = getAutocompleteSuggestions ( 'Cluster' ) ;
125+ expect ( suggestions ) . toContain ( 'ClusterAttributeScope' ) ;
126+ expect ( suggestions ) . toContain ( 'ClusterAttributeName' ) ;
127+ } ) ;
128+
108129 it ( 'suggests logical operators after complete complex query' , ( ) => {
109130 const complexQuery =
110131 'WorkflowID = "test" AND StartTime >= "2023-01-01T00:00:00Z"' ;
@@ -116,6 +137,11 @@ describe('getAutocompleteSuggestions', () => {
116137 const complexQuery =
117138 'WorkflowID = "test" AND StartTime >= "2023-01-01T00:00:00Z" AND c' ;
118139 const suggestions = getAutocompleteSuggestions ( complexQuery ) ;
119- expect ( suggestions ) . toEqual ( [ 'CloseTime' , 'CloseStatus' ] ) ;
140+ expect ( suggestions ) . toEqual ( [
141+ 'CloseTime' ,
142+ 'ClusterAttributeScope' ,
143+ 'ClusterAttributeName' ,
144+ 'CloseStatus' ,
145+ ] ) ;
120146 } ) ;
121147} ) ;
0 commit comments