@@ -107,26 +107,27 @@ describe('rebase operation', () => {
107107 expect ( screen . queryAllByTestId ( 'commit-1' ) ) . toHaveLength ( 1 ) ;
108108 } ) ;
109109
110- it ( 'runs rebase operation' , ( ) => {
110+ it ( 'runs rebase operation' , async ( ) => {
111111 dragAndDropCommits ( 'd' , '2' ) ;
112112
113113 const runRebaseButton = screen . getByText ( 'Run Rebase' ) ;
114114 expect ( runRebaseButton ) . toBeInTheDocument ( ) ;
115115
116116 fireEvent . click ( runRebaseButton ) ;
117-
118- expectMessageSentToServer ( {
119- type : 'runOperation' ,
120- operation : {
121- args : [ 'rebase' , '-s' , succeedableRevset ( 'd' ) , '-d' , succeedableRevset ( 'remote/master' ) ] ,
122- id : expect . anything ( ) ,
123- runner : CommandRunner . Sapling ,
124- trackEventName : 'RebaseOperation' ,
125- } ,
117+ await waitFor ( ( ) => {
118+ expectMessageSentToServer ( {
119+ type : 'runOperation' ,
120+ operation : {
121+ args : [ 'rebase' , '-s' , succeedableRevset ( 'd' ) , '-d' , succeedableRevset ( 'remote/master' ) ] ,
122+ id : expect . anything ( ) ,
123+ runner : CommandRunner . Sapling ,
124+ trackEventName : 'RebaseOperation' ,
125+ } ,
126+ } ) ;
126127 } ) ;
127128 } ) ;
128129
129- it ( 'shows optimistic preview of rebase' , ( ) => {
130+ it ( 'shows optimistic preview of rebase' , async ( ) => {
130131 dragAndDropCommits ( 'd' , '2' ) ;
131132
132133 fireEvent . click ( screen . getByText ( 'Run Rebase' ) ) ;
@@ -136,14 +137,16 @@ describe('rebase operation', () => {
136137 // also includes descendants
137138 expect ( screen . queryAllByTestId ( 'commit-e' ) ) . toHaveLength ( 1 ) ;
138139
139- expect ( screen . getByText ( 'rebasing...' ) ) . toBeInTheDocument ( ) ;
140+ await waitFor ( ( ) => {
141+ expect ( screen . getByText ( 'rebasing...' ) ) . toBeInTheDocument ( ) ;
142+ } ) ;
140143
141144 expect (
142145 screen . queryByTestId ( 'commit-d' ) ?. querySelector ( '.commit-preview-rebase-optimistic-root' ) ,
143146 ) . toBeInTheDocument ( ) ;
144147 } ) ;
145148
146- it ( 'allows re-dragging a previously dragged commit back onto the same parent' , ( ) => {
149+ it ( 'allows re-dragging a previously dragged commit back onto the same parent' , async ( ) => {
147150 // Drag and drop 'e' normally onto 'c'
148151 dragCommits ( 'e' , 'd' ) ;
149152 expect ( screen . queryByText ( 'Run Rebase' ) ) . not . toBeInTheDocument ( ) ; // dragging on original parent is noop
@@ -158,15 +161,16 @@ describe('rebase operation', () => {
158161 dropCommits ( getCommitWithPreview ( 'e' , CommitPreview . REBASE_ROOT ) , 'c' ) ;
159162
160163 fireEvent . click ( screen . getByText ( 'Run Rebase' ) ) ;
161-
162- expectMessageSentToServer ( {
163- type : 'runOperation' ,
164- operation : {
165- args : [ 'rebase' , '-s' , succeedableRevset ( 'e' ) , '-d' , succeedableRevset ( 'c' ) ] ,
166- id : expect . anything ( ) ,
167- runner : CommandRunner . Sapling ,
168- trackEventName : 'RebaseOperation' ,
169- } ,
164+ await waitFor ( ( ) => {
165+ expectMessageSentToServer ( {
166+ type : 'runOperation' ,
167+ operation : {
168+ args : [ 'rebase' , '-s' , succeedableRevset ( 'e' ) , '-d' , succeedableRevset ( 'c' ) ] ,
169+ id : expect . anything ( ) ,
170+ runner : CommandRunner . Sapling ,
171+ trackEventName : 'RebaseOperation' ,
172+ } ,
173+ } ) ;
170174 } ) ;
171175 } ) ;
172176
@@ -276,7 +280,7 @@ describe('rebase operation', () => {
276280 expect ( scanForkedBranchHashes ( 'b' ) ) . toEqual ( [ 'd' , 'e' ] ) ;
277281 } ) ;
278282
279- it ( 'can preview drag drop while previous rebase running' , ( ) => {
283+ it ( 'can preview drag drop while previous rebase running' , async ( ) => {
280284 // c
281285 // c | e
282286 // e b |/
@@ -286,6 +290,9 @@ describe('rebase operation', () => {
286290 // a a a
287291 dragAndDropCommits ( 'd' , 'a' ) ;
288292 fireEvent . click ( screen . getByText ( 'Run Rebase' ) ) ;
293+ await waitFor ( ( ) => {
294+ expect ( screen . getByText ( 'rebasing...' ) ) . toBeInTheDocument ( ) ;
295+ } ) ;
289296
290297 dragAndDropCommits (
291298 getCommitWithPreview ( 'e' , CommitPreview . REBASE_OPTIMISTIC_DESCENDANT ) ,
@@ -298,7 +305,7 @@ describe('rebase operation', () => {
298305 expect ( scanForkedBranchHashes ( 'b' ) ) . toEqual ( [ 'e' ] ) ;
299306 } ) ;
300307
301- it ( 'can see optimistic drag drop while previous rebase running' , ( ) => {
308+ it ( 'can see optimistic drag drop while previous rebase running' , async ( ) => {
302309 // c
303310 // c | e
304311 // e b |/
@@ -308,7 +315,9 @@ describe('rebase operation', () => {
308315 // a a a
309316 dragAndDropCommits ( 'd' , 'a' ) ;
310317 fireEvent . click ( screen . getByText ( 'Run Rebase' ) ) ;
311-
318+ await waitFor ( ( ) => {
319+ expect ( screen . getByText ( 'rebasing...' ) ) . toBeInTheDocument ( ) ;
320+ } ) ;
312321 dragAndDropCommits (
313322 getCommitWithPreview ( 'e' , CommitPreview . REBASE_OPTIMISTIC_DESCENDANT ) ,
314323 'b' ,
0 commit comments