@@ -8,18 +8,6 @@ import { MemoryRouter, Route } from 'react-router-dom'
8
8
9
9
import CircleCI from './CircleCI'
10
10
11
- const mocks = vi . hoisted ( ( ) => ( {
12
- useFlags : vi . fn ( ) ,
13
- } ) )
14
-
15
- vi . mock ( 'shared/featureFlags' , async ( ) => {
16
- const actual = await vi . importActual ( 'shared/featureFlags' )
17
- return {
18
- ...actual ,
19
- useFlags : mocks . useFlags ,
20
- }
21
- } )
22
-
23
11
const mockGetRepo = {
24
12
owner : {
25
13
isAdmin : null ,
@@ -45,6 +33,12 @@ const mockGetOrgUploadToken = {
45
33
} ,
46
34
}
47
35
36
+ const mockNoUploadToken = {
37
+ owner : {
38
+ orgUploadToken : null ,
39
+ } ,
40
+ }
41
+
48
42
const queryClient = new QueryClient ( {
49
43
defaultOptions : {
50
44
queries : {
@@ -85,9 +79,6 @@ interface SetupArgs {
85
79
86
80
describe ( 'CircleCI' , ( ) => {
87
81
function setup ( { hasOrgUploadToken = false } : SetupArgs ) {
88
- mocks . useFlags . mockReturnValue ( {
89
- newRepoFlag : hasOrgUploadToken ,
90
- } )
91
82
const mockMetricMutationVariables = vi . fn ( )
92
83
const mockGetItem = vi . spyOn ( window . localStorage . __proto__ , 'getItem' )
93
84
mockGetItem . mockReturnValue ( null )
@@ -97,7 +88,9 @@ describe('CircleCI', () => {
97
88
return HttpResponse . json ( { data : mockGetRepo } )
98
89
} ) ,
99
90
graphql . query ( 'GetOrgUploadToken' , ( ) => {
100
- return HttpResponse . json ( { data : mockGetOrgUploadToken } )
91
+ return HttpResponse . json ( {
92
+ data : hasOrgUploadToken ? mockGetOrgUploadToken : mockNoUploadToken ,
93
+ } )
101
94
} ) ,
102
95
graphql . mutation ( 'storeEventMetric' , ( info ) => {
103
96
mockMetricMutationVariables ( info ?. variables )
0 commit comments