3535PRODUCT_TEAM_NAME = "My Product Team"
3636PRODUCT_NAME = "My Product"
3737VERSION = 1
38+ PARTY_KEY = "ABC1234-987654"
3839
3940QUESTIONNAIRE_DATA = {
4041 "ODS Code" : "FH15R" ,
@@ -69,7 +70,7 @@ def mock_epr_product_with_one_message_set_drd() -> (
6970 product = product_team .create_cpm_product (
7071 name = PRODUCT_NAME , product_id = PRODUCT_ID
7172 )
72- product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = "ABC1234-987654" )
73+ product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = PARTY_KEY )
7374 product_repo = CpmProductRepository (
7475 table_name = TABLE_NAME , dynamodb_client = client
7576 )
@@ -79,10 +80,22 @@ def mock_epr_product_with_one_message_set_drd() -> (
7980 QuestionnaireInstance .SPINE_MHS_MESSAGE_SETS
8081 )
8182 questionnaire_response_1 = mhs_message_set_questionnaire .validate (
82- data = {"Interaction ID" : "urn:foo1" , "MHS SN" : "bar" , "MHS IN" : "baz" }
83+ data = {
84+ "Interaction ID" : "bar:baz" ,
85+ "MHS SN" : "bar" ,
86+ "MHS IN" : "baz" ,
87+ "MHS CPA ID" : f"{ PARTY_KEY } :bar:baz" ,
88+ "Unique Identifier" : f"{ PARTY_KEY } :bar:baz" ,
89+ }
8390 )
8491 questionnaire_response_2 = mhs_message_set_questionnaire .validate (
85- data = {"Interaction ID" : "urn:foo2" , "MHS SN" : "bar2" , "MHS IN" : "baz2" }
92+ data = {
93+ "Interaction ID" : "bar2:baz2" ,
94+ "MHS SN" : "bar2" ,
95+ "MHS IN" : "baz2" ,
96+ "MHS CPA ID" : f"{ PARTY_KEY } :bar2:baz2" ,
97+ "Unique Identifier" : f"{ PARTY_KEY } :bar2:baz2" ,
98+ }
8699 )
87100
88101 # Set up DeviceReferenceData in DB
@@ -124,7 +137,7 @@ def mock_epr_product_with_message_sets_drd() -> (
124137 product = product_team .create_cpm_product (
125138 name = PRODUCT_NAME , product_id = PRODUCT_ID
126139 )
127- product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = "ABC1234-987654" )
140+ product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = PARTY_KEY )
128141 product_repo = CpmProductRepository (
129142 table_name = TABLE_NAME , dynamodb_client = client
130143 )
@@ -135,10 +148,22 @@ def mock_epr_product_with_message_sets_drd() -> (
135148 )
136149
137150 questionnaire_response_1 = mhs_message_set_questionnaire .validate (
138- data = {"Interaction ID" : "urn:foo1" , "MHS SN" : "bar" , "MHS IN" : "baz" }
151+ data = {
152+ "Interaction ID" : "bar:baz" ,
153+ "MHS SN" : "bar" ,
154+ "MHS IN" : "baz" ,
155+ "MHS CPA ID" : f"{ PARTY_KEY } :bar:baz" ,
156+ "Unique Identifier" : f"{ PARTY_KEY } :bar:baz" ,
157+ }
139158 )
140159 questionnaire_response_2 = mhs_message_set_questionnaire .validate (
141- data = {"Interaction ID" : "urn:foo2" , "MHS SN" : "bar2" , "MHS IN" : "baz2" }
160+ data = {
161+ "Interaction ID" : "bar2:baz2" ,
162+ "MHS SN" : "bar2" ,
163+ "MHS IN" : "baz2" ,
164+ "MHS CPA ID" : f"{ PARTY_KEY } :bar2:baz2" ,
165+ "Unique Identifier" : f"{ PARTY_KEY } :bar2:baz2" ,
166+ }
142167 )
143168 # Set up DeviceReferenceData in DB
144169 device_reference_data_mhs = product .create_device_reference_data (
@@ -197,7 +222,7 @@ def mock_epr_product_with_more_than_two_message_sets_drd() -> (
197222 product = product_team .create_cpm_product (
198223 name = PRODUCT_NAME , product_id = PRODUCT_ID
199224 )
200- product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = "ABC1234-987654" )
225+ product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = PARTY_KEY )
201226 product_repo = CpmProductRepository (
202227 table_name = TABLE_NAME , dynamodb_client = client
203228 )
@@ -207,10 +232,22 @@ def mock_epr_product_with_more_than_two_message_sets_drd() -> (
207232 QuestionnaireInstance .SPINE_MHS_MESSAGE_SETS
208233 )
209234 questionnaire_response_1 = mhs_message_set_questionnaire_1 .validate (
210- data = {"Interaction ID" : "urn:foo1" , "MHS SN" : "bar" , "MHS IN" : "baz" }
235+ data = {
236+ "Interaction ID" : "bar:baz" ,
237+ "MHS SN" : "bar" ,
238+ "MHS IN" : "baz" ,
239+ "MHS CPA ID" : f"{ PARTY_KEY } :bar:baz" ,
240+ "Unique Identifier" : f"{ PARTY_KEY } :bar:baz" ,
241+ }
211242 )
212243 questionnaire_response_2 = mhs_message_set_questionnaire_1 .validate (
213- data = {"Interaction ID" : "urn:foo2" , "MHS SN" : "bar2" , "MHS IN" : "baz2" }
244+ data = {
245+ "Interaction ID" : "bar2:baz2" ,
246+ "MHS SN" : "bar2" ,
247+ "MHS IN" : "baz2" ,
248+ "MHS CPA ID" : f"{ PARTY_KEY } :bar2:baz2" ,
249+ "Unique Identifier" : f"{ PARTY_KEY } :bar2:baz2" ,
250+ }
214251 )
215252
216253 # Set up DeviceReferenceData in DB
@@ -224,10 +261,22 @@ def mock_epr_product_with_more_than_two_message_sets_drd() -> (
224261 QuestionnaireInstance .SPINE_MHS_MESSAGE_SETS
225262 )
226263 questionnaire_response_3 = mhs_message_set_questionnaire_2 .validate (
227- data = {"Interaction ID" : "urn:foo3" , "MHS SN" : "bar3" , "MHS IN" : "baz3" }
264+ data = {
265+ "Interaction ID" : "bar3:baz3" ,
266+ "MHS SN" : "bar3" ,
267+ "MHS IN" : "baz3" ,
268+ "MHS CPA ID" : f"{ PARTY_KEY } :bar3:baz3" ,
269+ "Unique Identifier" : f"{ PARTY_KEY } :bar3:baz3" ,
270+ }
228271 )
229272 questionnaire_response_4 = mhs_message_set_questionnaire_2 .validate (
230- data = {"Interaction ID" : "urn:foo4" , "MHS SN" : "bar4" , "MHS IN" : "baz4" }
273+ data = {
274+ "Interaction ID" : "bar4:baz4" ,
275+ "MHS SN" : "bar4" ,
276+ "MHS IN" : "baz4" ,
277+ "MHS CPA ID" : f"{ PARTY_KEY } :bar4:baz4" ,
278+ "Unique Identifier" : f"{ PARTY_KEY } :bar4:baz4" ,
279+ }
231280 )
232281
233282 # Set up DeviceReferenceData in DB
@@ -287,7 +336,7 @@ def mock_epr_product_with_two_message_sets_the_same_drd() -> (
287336 product = product_team .create_cpm_product (
288337 name = PRODUCT_NAME , product_id = PRODUCT_ID
289338 )
290- product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = "ABC1234-987654" )
339+ product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = PARTY_KEY )
291340 product_repo = CpmProductRepository (
292341 table_name = TABLE_NAME , dynamodb_client = client
293342 )
@@ -297,10 +346,22 @@ def mock_epr_product_with_two_message_sets_the_same_drd() -> (
297346 QuestionnaireInstance .SPINE_MHS_MESSAGE_SETS
298347 )
299348 questionnaire_response_1 = mhs_message_set_questionnaire_1 .validate (
300- data = {"Interaction ID" : "urn:foo1" , "MHS SN" : "bar" , "MHS IN" : "baz" }
349+ data = {
350+ "Interaction ID" : "bar:baz" ,
351+ "MHS SN" : "bar" ,
352+ "MHS IN" : "baz" ,
353+ "MHS CPA ID" : f"{ PARTY_KEY } :bar:baz" ,
354+ "Unique Identifier" : f"{ PARTY_KEY } :bar:baz" ,
355+ }
301356 )
302357 questionnaire_response_2 = mhs_message_set_questionnaire_1 .validate (
303- data = {"Interaction ID" : "urn:foo2" , "MHS SN" : "bar2" , "MHS IN" : "baz2" }
358+ data = {
359+ "Interaction ID" : "bar2:baz2" ,
360+ "MHS SN" : "ba2r" ,
361+ "MHS IN" : "baz2" ,
362+ "MHS CPA ID" : f"{ PARTY_KEY } :bar2:baz2" ,
363+ "Unique Identifier" : f"{ PARTY_KEY } :bar2:baz2" ,
364+ }
304365 )
305366
306367 # Set up DeviceReferenceData in DB
@@ -314,10 +375,22 @@ def mock_epr_product_with_two_message_sets_the_same_drd() -> (
314375 QuestionnaireInstance .SPINE_MHS_MESSAGE_SETS
315376 )
316377 questionnaire_response_3 = mhs_message_set_questionnaire_2 .validate (
317- data = {"Interaction ID" : "urn:foo3" , "MHS SN" : "bar3" , "MHS IN" : "baz3" }
378+ data = {
379+ "Interaction ID" : "bar3:baz3" ,
380+ "MHS SN" : "bar3" ,
381+ "MHS IN" : "baz3" ,
382+ "MHS CPA ID" : f"{ PARTY_KEY } :bar3:baz3" ,
383+ "Unique Identifier" : f"{ PARTY_KEY } :bar3:baz3" ,
384+ }
318385 )
319386 questionnaire_response_4 = mhs_message_set_questionnaire_2 .validate (
320- data = {"Interaction ID" : "urn:foo4" , "MHS SN" : "bar4" , "MHS IN" : "baz4" }
387+ data = {
388+ "Interaction ID" : "bar4:baz4" ,
389+ "MHS SN" : "bar4" ,
390+ "MHS IN" : "baz4" ,
391+ "MHS CPA ID" : f"{ PARTY_KEY } :bar4:baz4" ,
392+ "Unique Identifier" : f"{ PARTY_KEY } :bar4:baz4" ,
393+ }
321394 )
322395
323396 # Set up DeviceReferenceData in DB
@@ -390,7 +463,7 @@ def mock_epr_product_without_message_set_drd() -> (
390463 product = product_team .create_cpm_product (
391464 name = PRODUCT_NAME , product_id = PRODUCT_ID
392465 )
393- product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = "ABC1234-987654" )
466+ product .add_key (key_type = ProductKeyType .PARTY_KEY , key_value = PARTY_KEY )
394467 product_repo = CpmProductRepository (
395468 table_name = TABLE_NAME , dynamodb_client = client
396469 )
@@ -449,7 +522,7 @@ def test_index() -> None:
449522 )
450523
451524 # Check party_key is added to tags in the created device
452- expected_party_key = (str (ProductKeyType .PARTY_KEY ), "abc1234-987654" )
525+ expected_party_key = (str (ProductKeyType .PARTY_KEY ), PARTY_KEY . lower () )
453526 assert any (expected_party_key in tag .__root__ for tag in created_device .tags )
454527
455528 # Check an ASID is generated and added to the keys.
0 commit comments