@@ -1254,21 +1254,27 @@ def setUp(self):
12541254 self .arthur = Student .objects .create (name = "Arthur" )
12551255 self .school = School .objects .create (name = "School of Awesome" )
12561256
1257- def assertActiveButtons (
1258- self , mode , field_name , choose , remove , choose_all = None , remove_all = None
1257+ def assertButtonsDisabled (
1258+ self ,
1259+ mode ,
1260+ field_name ,
1261+ choose_btn_disabled = False ,
1262+ remove_btn_disabled = False ,
1263+ choose_all_btn_disabled = False ,
1264+ remove_all_btn_disabled = False ,
12591265 ):
12601266 choose_button = "#id_%s_add" % field_name
12611267 choose_all_button = "#id_%s_add_all" % field_name
12621268 remove_button = "#id_%s_remove" % field_name
12631269 remove_all_button = "#id_%s_remove_all" % field_name
1264- self .assertEqual (self .has_css_class (choose_button , "active" ), choose )
1265- self .assertEqual (self .has_css_class (remove_button , "active" ), remove )
1270+ self .assertEqual (self .is_disabled (choose_button ), choose_btn_disabled )
1271+ self .assertEqual (self .is_disabled (remove_button ), remove_btn_disabled )
12661272 if mode == "horizontal" :
12671273 self .assertEqual (
1268- self .has_css_class (choose_all_button , "active" ), choose_all
1274+ self .is_disabled (choose_all_button ), choose_all_btn_disabled
12691275 )
12701276 self .assertEqual (
1271- self .has_css_class (remove_all_button , "active" ), remove_all
1277+ self .is_disabled (remove_all_button ), remove_all_btn_disabled
12721278 )
12731279
12741280 def execute_basic_operations (self , mode , field_name ):
@@ -1296,7 +1302,14 @@ def execute_basic_operations(self, mode, field_name):
12961302 ],
12971303 )
12981304 self .assertSelectOptions (to_box , [str (self .lisa .id ), str (self .peter .id )])
1299- self .assertActiveButtons (mode , field_name , False , False , True , True )
1305+ self .assertButtonsDisabled (
1306+ mode ,
1307+ field_name ,
1308+ choose_btn_disabled = True ,
1309+ remove_btn_disabled = True ,
1310+ choose_all_btn_disabled = False ,
1311+ remove_all_btn_disabled = False ,
1312+ )
13001313
13011314 # Click 'Choose all' --------------------------------------------------
13021315 if mode == "horizontal" :
@@ -1323,7 +1336,14 @@ def execute_basic_operations(self, mode, field_name):
13231336 str (self .john .id ),
13241337 ],
13251338 )
1326- self .assertActiveButtons (mode , field_name , False , False , False , True )
1339+ self .assertButtonsDisabled (
1340+ mode ,
1341+ field_name ,
1342+ choose_btn_disabled = True ,
1343+ remove_btn_disabled = True ,
1344+ choose_all_btn_disabled = True ,
1345+ remove_all_btn_disabled = False ,
1346+ )
13271347
13281348 # Click 'Remove all' --------------------------------------------------
13291349 if mode == "horizontal" :
@@ -1350,7 +1370,14 @@ def execute_basic_operations(self, mode, field_name):
13501370 ],
13511371 )
13521372 self .assertSelectOptions (to_box , [])
1353- self .assertActiveButtons (mode , field_name , False , False , True , False )
1373+ self .assertButtonsDisabled (
1374+ mode ,
1375+ field_name ,
1376+ choose_btn_disabled = True ,
1377+ remove_btn_disabled = True ,
1378+ choose_all_btn_disabled = False ,
1379+ remove_all_btn_disabled = True ,
1380+ )
13541381
13551382 # Choose some options ------------------------------------------------
13561383 from_lisa_select_option = self .selenium .find_element (
@@ -1367,9 +1394,23 @@ def execute_basic_operations(self, mode, field_name):
13671394 self .select_option (from_box , str (self .jason .id ))
13681395 self .select_option (from_box , str (self .bob .id ))
13691396 self .select_option (from_box , str (self .john .id ))
1370- self .assertActiveButtons (mode , field_name , True , False , True , False )
1397+ self .assertButtonsDisabled (
1398+ mode ,
1399+ field_name ,
1400+ choose_btn_disabled = False ,
1401+ remove_btn_disabled = True ,
1402+ choose_all_btn_disabled = False ,
1403+ remove_all_btn_disabled = True ,
1404+ )
13711405 self .selenium .find_element (By .ID , choose_button ).click ()
1372- self .assertActiveButtons (mode , field_name , False , False , True , True )
1406+ self .assertButtonsDisabled (
1407+ mode ,
1408+ field_name ,
1409+ choose_btn_disabled = True ,
1410+ remove_btn_disabled = True ,
1411+ choose_all_btn_disabled = False ,
1412+ remove_all_btn_disabled = False ,
1413+ )
13731414
13741415 self .assertSelectOptions (
13751416 from_box ,
@@ -1402,9 +1443,23 @@ def execute_basic_operations(self, mode, field_name):
14021443 # Remove some options -------------------------------------------------
14031444 self .select_option (to_box , str (self .lisa .id ))
14041445 self .select_option (to_box , str (self .bob .id ))
1405- self .assertActiveButtons (mode , field_name , False , True , True , True )
1446+ self .assertButtonsDisabled (
1447+ mode ,
1448+ field_name ,
1449+ choose_btn_disabled = True ,
1450+ remove_btn_disabled = False ,
1451+ choose_all_btn_disabled = False ,
1452+ remove_all_btn_disabled = False ,
1453+ )
14061454 self .selenium .find_element (By .ID , remove_button ).click ()
1407- self .assertActiveButtons (mode , field_name , False , False , True , True )
1455+ self .assertButtonsDisabled (
1456+ mode ,
1457+ field_name ,
1458+ choose_btn_disabled = True ,
1459+ remove_btn_disabled = True ,
1460+ choose_all_btn_disabled = False ,
1461+ remove_all_btn_disabled = False ,
1462+ )
14081463
14091464 self .assertSelectOptions (
14101465 from_box ,
0 commit comments