@@ -288,11 +288,16 @@ describe('actions/Policy', () => {
288288 const newTagName = 'new tag' ;
289289 const fakePolicyTags = createRandomPolicyTags ( tagListName ) ;
290290
291+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY } ${ fakePolicy . id } ` , fakePolicy ) ;
292+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakePolicyTags ) ;
293+
291294 mockFetch . pause ( ) ;
292295 await waitForBatchedUpdates ( ) ;
293296
297+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
298+
294299 // When creating a new tag
295- createPolicyTag ( fakePolicy . id , newTagName , fakePolicyTags ) ;
300+ createPolicyTag ( policyData . current , newTagName ) ;
296301 await waitForBatchedUpdates ( ) ;
297302
298303 // Then the tag should appear optimistically with pending state so the user sees immediate feedback
@@ -322,12 +327,17 @@ describe('actions/Policy', () => {
322327 const newTagName = 'new tag' ;
323328 const fakePolicyTags = createRandomPolicyTags ( tagListName ) ;
324329
330+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY } ${ fakePolicy . id } ` , fakePolicy ) ;
331+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakePolicyTags ) ;
332+
325333 mockFetch . pause ( ) ;
326334 await waitForBatchedUpdates ( ) ;
327335 mockFetch . fail ( ) ;
328336
337+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
338+
329339 // When the API fails
330- createPolicyTag ( fakePolicy . id , newTagName , fakePolicyTags ) ;
340+ createPolicyTag ( policyData . current , newTagName ) ;
331341 await waitForBatchedUpdates ( ) ;
332342 mockFetch . resume ( ) ;
333343 await waitForBatchedUpdates ( ) ;
@@ -345,11 +355,15 @@ describe('actions/Policy', () => {
345355
346356 const newTagName = 'new tag' ;
347357
358+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY } ${ fakePolicy . id } ` , fakePolicy ) ;
359+
348360 mockFetch . pause ( ) ;
349361 await waitForBatchedUpdates ( ) ;
350362
363+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
364+
351365 // When adding the first tag
352- createPolicyTag ( fakePolicy . id , newTagName , { } ) ;
366+ createPolicyTag ( policyData . current , newTagName ) ;
353367 await waitForBatchedUpdates ( ) ;
354368
355369 // Then the tag should be created in a new list with pending state so the user sees immediate feedback
@@ -389,6 +403,7 @@ describe('actions/Policy', () => {
389403
390404 mockFetch . pause ( ) ;
391405
406+ await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY } ${ fakePolicy . id } ` , fakePolicy ) ;
392407 await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakePolicyTags ) ;
393408 await waitForBatchedUpdates ( ) ;
394409
@@ -398,8 +413,10 @@ describe('actions/Policy', () => {
398413 expect ( result . current [ 0 ] ) . toBeDefined ( ) ;
399414 } ) ;
400415
416+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
417+
401418 // When using data from useOnyx hook
402- createPolicyTag ( fakePolicy . id , newTagName , result . current [ 0 ] ?? { } ) ;
419+ createPolicyTag ( policyData . current , newTagName ) ;
403420 await waitForBatchedUpdates ( ) ;
404421
405422 // Then the tag should appear optimistically with pending state so the user sees immediate feedback
@@ -2243,7 +2260,9 @@ describe('actions/Policy', () => {
22432260 await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakeTags ) ;
22442261 await Onyx . set ( `${ ONYXKEYS . COLLECTION . REPORT } ${ fakeTaskReportID } ` , fakeTaskReport ) ;
22452262
2246- createPolicyTag ( fakePolicy . id , newTagName , fakeTags , fakeTaskReport , undefined , false ) ;
2263+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
2264+
2265+ createPolicyTag ( policyData . current , newTagName , fakeTaskReport ) ;
22472266
22482267 await waitForBatchedUpdates ( ) ;
22492268
@@ -2276,7 +2295,9 @@ describe('actions/Policy', () => {
22762295 await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakeTags ) ;
22772296 await Onyx . set ( `${ ONYXKEYS . COLLECTION . REPORT } ${ fakeTaskReportID } ` , fakeTaskReport ) ;
22782297
2279- createPolicyTag ( fakePolicy . id , newTagName , fakeTags , undefined , fakeTaskReport , true ) ;
2298+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
2299+
2300+ createPolicyTag ( policyData . current , newTagName , undefined , fakeTaskReport , true ) ;
22802301
22812302 await waitForBatchedUpdates ( ) ;
22822303
@@ -2309,7 +2330,9 @@ describe('actions/Policy', () => {
23092330 await Onyx . set ( `${ ONYXKEYS . COLLECTION . POLICY_TAGS } ${ fakePolicy . id } ` , fakeTags ) ;
23102331 await Onyx . set ( `${ ONYXKEYS . COLLECTION . REPORT } ${ fakeTaskReportID } ` , fakeTaskReport ) ;
23112332
2312- createPolicyTag ( fakePolicy . id , newTagName , fakeTags , undefined , fakeTaskReport , false ) ;
2333+ const { result : policyData } = renderHook ( ( ) => usePolicyData ( fakePolicy . id ) , { wrapper : OnyxListItemProvider } ) ;
2334+
2335+ createPolicyTag ( policyData . current , newTagName , undefined , fakeTaskReport , false ) ;
23132336
23142337 await waitForBatchedUpdates ( ) ;
23152338
0 commit comments