33from unittest .mock import patch
44from uuid import UUID
55
6+ from cc_common .data_model .update_tier_enum import UpdateTierEnum
67from common_test .test_constants import (
78 DEFAULT_AA_SUBMITTING_USER_ID ,
89 DEFAULT_DATE_OF_UPDATE_TIMESTAMP ,
@@ -955,9 +956,12 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
955956 provider_user_records = self .config .data_client .get_provider_user_records (
956957 compact = test_privilege_record .compact ,
957958 provider_id = test_privilege_record .providerId ,
959+ include_update_tier = UpdateTierEnum .TIER_THREE
958960 )
959961 update_records = provider_user_records .get_update_records_for_privilege (
960- jurisdiction = test_privilege_record .jurisdiction , license_type = test_privilege_record .licenseType
962+ jurisdiction = test_privilege_record .jurisdiction ,
963+ license_type = test_privilege_record .licenseType ,
964+
961965 )
962966
963967 investigation_update_records = [
@@ -1002,6 +1006,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
10021006 provider_user_records = self .config .data_client .get_provider_user_records (
10031007 compact = test_privilege_record .compact ,
10041008 provider_id = test_privilege_record .providerId ,
1009+ include_update_tier = UpdateTierEnum .TIER_THREE
10051010 )
10061011 updated_privilege_record = provider_user_records .get_privilege_records ()[0 ]
10071012
@@ -1020,6 +1025,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
10201025 provider_user_records = self .config .data_client .get_provider_user_records (
10211026 compact = test_privilege_record .compact ,
10221027 provider_id = test_privilege_record .providerId ,
1028+ include_update_tier = UpdateTierEnum .TIER_THREE
10231029 )
10241030 update_records = provider_user_records .get_update_records_for_privilege (
10251031 jurisdiction = test_privilege_record .jurisdiction , license_type = test_privilege_record .licenseType
@@ -1088,6 +1094,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
10881094 provider_user_records = self .config .data_client .get_provider_user_records (
10891095 compact = test_license_record .compact ,
10901096 provider_id = test_license_record .providerId ,
1097+ include_update_tier = UpdateTierEnum .TIER_THREE
10911098 )
10921099 investigation_records = provider_user_records .get_investigation_records_for_license (
10931100 license_jurisdiction = test_license_record .jurisdiction ,
@@ -1119,6 +1126,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
11191126 provider_user_records = self .config .data_client .get_provider_user_records (
11201127 compact = test_license_record .compact ,
11211128 provider_id = test_license_record .providerId ,
1129+ include_update_tier = UpdateTierEnum .TIER_THREE
11221130 )
11231131 investigation_records = provider_user_records .get_investigation_records_for_license (
11241132 license_jurisdiction = test_license_record .jurisdiction ,
@@ -1154,6 +1162,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
11541162 provider_user_records = self .config .data_client .get_provider_user_records (
11551163 compact = test_license_record .compact ,
11561164 provider_id = test_license_record .providerId ,
1165+ include_update_tier = UpdateTierEnum .TIER_THREE
11571166 )
11581167 updated_license_record = provider_user_records .get_license_records ()[0 ]
11591168
@@ -1188,6 +1197,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
11881197 provider_user_records = self .config .data_client .get_provider_user_records (
11891198 compact = test_license_record .compact ,
11901199 provider_id = test_license_record .providerId ,
1200+ include_update_tier = UpdateTierEnum .TIER_THREE
11911201 )
11921202 update_records = provider_user_records .get_update_records_for_license (
11931203 jurisdiction = test_license_record .jurisdiction , license_type = test_license_record .licenseType
@@ -1235,6 +1245,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
12351245 provider_user_records = self .config .data_client .get_provider_user_records (
12361246 compact = test_license_record .compact ,
12371247 provider_id = test_license_record .providerId ,
1248+ include_update_tier = UpdateTierEnum .TIER_THREE
12381249 )
12391250 updated_license_record = provider_user_records .get_license_records ()[0 ]
12401251
@@ -1253,6 +1264,7 @@ def test_closing_one_of_multiple_investigations_maintains_investigation_status(s
12531264 provider_user_records = self .config .data_client .get_provider_user_records (
12541265 compact = test_license_record .compact ,
12551266 provider_id = test_license_record .providerId ,
1267+ include_update_tier = UpdateTierEnum .TIER_THREE
12561268 )
12571269 update_records = provider_user_records .get_update_records_for_license (
12581270 jurisdiction = test_license_record .jurisdiction , license_type = test_license_record .licenseType
0 commit comments