@@ -1094,52 +1094,70 @@ def test_fetch_available_plans_is_enterprise_plan(self):
10941094 isProPlan
10951095 isTeamPlan
10961096 isSentryPlan
1097- isTrialPlan
10981097 isFreePlan
1099- }
1100- }
1101- }
1102- """ % (current_org .username )
1103- data = self .gql_request (query , owner = current_org )
1104- assert data ["owner" ]["availablePlans" ][0 ]["isEnterprisePlan" ] is False
1105- assert data ["owner" ]["availablePlans" ][0 ]["isFreePlan" ] is True
1106- assert data ["owner" ]["availablePlans" ][0 ]["isProPlan" ] is False
1107- assert data ["owner" ]["availablePlans" ][0 ]["isTeamPlan" ] is False
1108- assert data ["owner" ]["availablePlans" ][0 ]["isSentryPlan" ] is False
1109- assert data ["owner" ]["availablePlans" ][0 ]["isTrialPlan" ] is False
1110-
1111- def test_fetch_available_plans_is_trial_plan (self ):
1112- current_org = OwnerFactory (
1113- username = "random-plan-user" ,
1114- service = "github" ,
1115- plan = PlanName .TRIAL_PLAN_NAME .value ,
1116- )
1117-
1118- query = """{
1119- owner(username: "%s") {
1120- availablePlans {
11211098 isTrialPlan
11221099 }
11231100 }
11241101 }
11251102 """ % (current_org .username )
11261103 data = self .gql_request (query , owner = current_org )
1127- assert data ["owner" ]["availablePlans" ][0 ]["isTrialPlan" ] is True
1128-
1129- def test_fetch_available_plans_is_pro_plan (self ):
1130- current_org = OwnerFactory (
1131- username = "random-plan-user" ,
1132- service = "github" ,
1133- plan = PlanName .CODECOV_PRO_MONTHLY_LEGACY .value ,
1134- )
1135-
1136- query = """{
1137- owner(username: "%s") {
1138- availablePlans {
1139- isProPlan
1140- }
1104+ assert data == {
1105+ "owner" : {
1106+ "availablePlans" : [
1107+ {
1108+ "value" : "users-basic" ,
1109+ "isEnterprisePlan" : False ,
1110+ "isProPlan" : False ,
1111+ "isTeamPlan" : False ,
1112+ "isSentryPlan" : False ,
1113+ "isFreePlan" : True ,
1114+ "isTrialPlan" : False ,
1115+ },
1116+ {
1117+ "value" : "users-free" ,
1118+ "isEnterprisePlan" : False ,
1119+ "isProPlan" : False ,
1120+ "isTeamPlan" : False ,
1121+ "isSentryPlan" : False ,
1122+ "isFreePlan" : True ,
1123+ "isTrialPlan" : False ,
1124+ },
1125+ {
1126+ "value" : "users-pr-inappm" ,
1127+ "isEnterprisePlan" : False ,
1128+ "isProPlan" : True ,
1129+ "isTeamPlan" : False ,
1130+ "isSentryPlan" : False ,
1131+ "isFreePlan" : False ,
1132+ "isTrialPlan" : False ,
1133+ },
1134+ {
1135+ "value" : "users-pr-inappy" ,
1136+ "isEnterprisePlan" : False ,
1137+ "isProPlan" : True ,
1138+ "isTeamPlan" : False ,
1139+ "isSentryPlan" : False ,
1140+ "isFreePlan" : False ,
1141+ "isTrialPlan" : False ,
1142+ },
1143+ {
1144+ "value" : "users-teamm" ,
1145+ "isEnterprisePlan" : False ,
1146+ "isProPlan" : False ,
1147+ "isTeamPlan" : True ,
1148+ "isSentryPlan" : False ,
1149+ "isFreePlan" : False ,
1150+ "isTrialPlan" : False ,
1151+ },
1152+ {
1153+ "value" : "users-teamy" ,
1154+ "isEnterprisePlan" : False ,
1155+ "isProPlan" : False ,
1156+ "isTeamPlan" : True ,
1157+ "isSentryPlan" : False ,
1158+ "isFreePlan" : False ,
1159+ "isTrialPlan" : False ,
1160+ },
1161+ ]
11411162 }
11421163 }
1143- """ % (current_org .username )
1144- data = self .gql_request (query , owner = current_org )
1145- assert data ["owner" ]["availablePlans" ][0 ]["isProPlan" ] is True
0 commit comments