@@ -3566,15 +3566,16 @@ def test_component_catalog_package_form_add(self):
35663566 "usage_policy" : policy_approved .pk ,
35673567 }
35683568 form = PackageForm (user = self .super_user , data = data )
3569- self .assertEqual ( 0 , len ( form .fields [ "usage_policy" ]. queryset ) )
3569+ self .assertNotIn ( "usage_policy" , form .fields )
35703570 self .assertTrue (form .is_valid ())
35713571 package = form .save ()
35723572 self .assertIsNone (package .usage_policy )
35733573
35743574 data ["filename" ] = "with policy"
35753575 self .super_user = add_perm (self .super_user , "change_usage_policy_on_package" )
35763576 form = PackageForm (user = self .super_user , data = data )
3577- self .assertEqual (1 , len (form .fields ["usage_policy" ].queryset ))
3577+ self .assertIn ("usage_policy" , form .fields )
3578+ self .assertQuerySetEqual ([policy_approved ], form .fields ["usage_policy" ].queryset )
35783579 self .assertTrue (form .is_valid ())
35793580 package = form .save ()
35803581 self .assertEqual (policy_approved , package .usage_policy )
@@ -4386,15 +4387,16 @@ def test_component_catalog_component_form_add(self):
43864387 "usage_policy" : policy_approved .pk ,
43874388 }
43884389 form = ComponentForm (user = self .user , data = data )
4389- self .assertEqual ( 0 , len ( form .fields [ "usage_policy" ]. queryset ) )
4390+ self .assertNotIn ( "usage_policy" , form .fields )
43904391 self .assertTrue (form .is_valid ())
43914392 component = form .save ()
43924393 self .assertIsNone (component .usage_policy )
43934394
43944395 data ["version" ] = "with policy"
43954396 self .user = add_perm (self .user , "change_usage_policy_on_component" )
43964397 form = ComponentForm (user = self .user , data = data )
4397- self .assertEqual (1 , len (form .fields ["usage_policy" ].queryset ))
4398+ self .assertIn ("usage_policy" , form .fields )
4399+ self .assertQuerySetEqual ([policy_approved ], form .fields ["usage_policy" ].queryset )
43984400 self .assertTrue (form .is_valid ())
43994401 component = form .save ()
44004402 self .assertEqual (policy_approved , component .usage_policy )
@@ -4413,6 +4415,7 @@ def test_component_catalog_component_form_add(self):
44134415 "homepage_url" : "https://nexb.com" ,
44144416 "configuration_status" : status .pk ,
44154417 "release_date" : "2019-03-01" ,
4418+ "usage_policy" : policy_approved .pk ,
44164419 "submit" : "Add Component" ,
44174420 }
44184421 form = ComponentForm (user = self .user , data = data )
@@ -4422,6 +4425,7 @@ def test_component_catalog_component_form_add(self):
44224425 self .assertEqual (status , component .configuration_status )
44234426 self .assertEqual (license1 .key , component .license_expression )
44244427 self .assertEqual (["Key1" , "Another keyword" ], component .keywords )
4428+ self .assertEqual (policy_approved , component .usage_policy )
44254429
44264430 def test_component_catalog_component_form_assigned_packages (self ):
44274431 data = {
0 commit comments