@@ -325,7 +325,6 @@ func Test_IssueRead_IFC_InsidersMode(t *testing.T) {
325325 t .Run ("insiders mode disabled omits ifc label" , func (t * testing.T ) {
326326 deps := BaseDeps {
327327 Client : mustNewGHClient (t , makeMockClient (false , 0 )),
328- Flags : FeatureFlags {InsidersMode : false },
329328 }
330329 handler := serverTool .Handler (deps )
331330
@@ -339,8 +338,8 @@ func Test_IssueRead_IFC_InsidersMode(t *testing.T) {
339338
340339 t .Run ("insiders mode enabled on public repo emits public untrusted" , func (t * testing.T ) {
341340 deps := BaseDeps {
342- Client : mustNewGHClient (t , makeMockClient (false , 0 )),
343- Flags : FeatureFlags { InsidersMode : true } ,
341+ Client : mustNewGHClient (t , makeMockClient (false , 0 )),
342+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
344343 }
345344 handler := serverTool .Handler (deps )
346345
@@ -357,8 +356,8 @@ func Test_IssueRead_IFC_InsidersMode(t *testing.T) {
357356
358357 t .Run ("insiders mode enabled on private repo with get_comments emits private untrusted" , func (t * testing.T ) {
359358 deps := BaseDeps {
360- Client : mustNewGHClient (t , makeMockClient (true , 0 )),
361- Flags : FeatureFlags { InsidersMode : true } ,
359+ Client : mustNewGHClient (t , makeMockClient (true , 0 )),
360+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
362361 }
363362 handler := serverTool .Handler (deps )
364363
@@ -375,8 +374,8 @@ func Test_IssueRead_IFC_InsidersMode(t *testing.T) {
375374
376375 t .Run ("insiders mode skips ifc label when visibility lookup fails" , func (t * testing.T ) {
377376 deps := BaseDeps {
378- Client : mustNewGHClient (t , makeMockClient (false , http .StatusInternalServerError )),
379- Flags : FeatureFlags { InsidersMode : true } ,
377+ Client : mustNewGHClient (t , makeMockClient (false , http .StatusInternalServerError )),
378+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
380379 }
381380 handler := serverTool .Handler (deps )
382381
@@ -910,7 +909,6 @@ func Test_SearchIssues_IFC_InsidersMode(t *testing.T) {
910909 searchResult := & github.IssuesSearchResult {Issues : []* github.Issue {makeIssue ("octocat" , "public-repo" , 1 )}}
911910 deps := BaseDeps {
912911 Client : mustNewGHClient (t , makeMockClient (searchResult , []repoFixture {{owner : "octocat" , repo : "public-repo" }})),
913- Flags : FeatureFlags {InsidersMode : false },
914912 }
915913 handler := serverTool .Handler (deps )
916914
@@ -924,8 +922,8 @@ func Test_SearchIssues_IFC_InsidersMode(t *testing.T) {
924922 t .Run ("insiders mode all public emits public untrusted" , func (t * testing.T ) {
925923 searchResult := & github.IssuesSearchResult {Issues : []* github.Issue {makeIssue ("octocat" , "public-repo" , 1 )}}
926924 deps := BaseDeps {
927- Client : mustNewGHClient (t , makeMockClient (searchResult , []repoFixture {{owner : "octocat" , repo : "public-repo" }})),
928- Flags : FeatureFlags { InsidersMode : true } ,
925+ Client : mustNewGHClient (t , makeMockClient (searchResult , []repoFixture {{owner : "octocat" , repo : "public-repo" }})),
926+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
929927 }
930928 handler := serverTool .Handler (deps )
931929
@@ -950,7 +948,7 @@ func Test_SearchIssues_IFC_InsidersMode(t *testing.T) {
950948 {owner : "octocat" , repo : "private-repo" , isPrivate : true },
951949 {owner : "octocat" , repo : "public-repo" },
952950 })),
953- Flags : FeatureFlags { InsidersMode : true } ,
951+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
954952 }
955953 handler := serverTool .Handler (deps )
956954
@@ -971,7 +969,7 @@ func Test_SearchIssues_IFC_InsidersMode(t *testing.T) {
971969 Client : mustNewGHClient (t , makeMockClient (searchResult , []repoFixture {
972970 {owner : "octocat" , repo : "broken" , repoStatus : http .StatusInternalServerError },
973971 })),
974- Flags : FeatureFlags { InsidersMode : true } ,
972+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
975973 }
976974 handler := serverTool .Handler (deps )
977975
@@ -989,8 +987,8 @@ func Test_SearchIssues_IFC_InsidersMode(t *testing.T) {
989987 t .Run ("insiders mode empty results emits public untrusted" , func (t * testing.T ) {
990988 searchResult := & github.IssuesSearchResult {Issues : []* github.Issue {}}
991989 deps := BaseDeps {
992- Client : mustNewGHClient (t , makeMockClient (searchResult , nil )),
993- Flags : FeatureFlags { InsidersMode : true } ,
990+ Client : mustNewGHClient (t , makeMockClient (searchResult , nil )),
991+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
994992 }
995993 handler := serverTool .Handler (deps )
996994
@@ -1876,7 +1874,6 @@ func Test_ListIssues_IFC_InsidersMode(t *testing.T) {
18761874 gqlClient := githubv4 .NewClient (githubv4mock .NewMockedHTTPClient (matcher ))
18771875 deps := BaseDeps {
18781876 GQLClient : gqlClient ,
1879- Flags : FeatureFlags {InsidersMode : false },
18801877 }
18811878 handler := serverTool .Handler (deps )
18821879
@@ -1892,8 +1889,8 @@ func Test_ListIssues_IFC_InsidersMode(t *testing.T) {
18921889 matcher := githubv4mock .NewQueryMatcher (query , vars , makeResponse (false ))
18931890 gqlClient := githubv4 .NewClient (githubv4mock .NewMockedHTTPClient (matcher ))
18941891 deps := BaseDeps {
1895- GQLClient : gqlClient ,
1896- Flags : FeatureFlags { InsidersMode : true } ,
1892+ GQLClient : gqlClient ,
1893+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
18971894 }
18981895 handler := serverTool .Handler (deps )
18991896
@@ -1919,8 +1916,8 @@ func Test_ListIssues_IFC_InsidersMode(t *testing.T) {
19191916 matcher := githubv4mock .NewQueryMatcher (query , vars , makeResponse (true ))
19201917 gqlClient := githubv4 .NewClient (githubv4mock .NewMockedHTTPClient (matcher ))
19211918 deps := BaseDeps {
1922- GQLClient : gqlClient ,
1923- Flags : FeatureFlags { InsidersMode : true } ,
1919+ GQLClient : gqlClient ,
1920+ featureChecker : featureCheckerFor ( FeatureFlagIFCLabels ) ,
19241921 }
19251922 handler := serverTool .Handler (deps )
19261923
0 commit comments