33from servicex_app .cli .user_commands import list_users
44
55
6- def _make_user (sub = "sub1" , email = "user@example.com" , name = "Test User" ,
7- institution = "UChicago" , admin = False , pending = False ):
6+ def _make_user (
7+ sub = "sub1" ,
8+ email = "user@example.com" ,
9+ name = "Test User" ,
10+ institution = "UChicago" ,
11+ admin = False ,
12+ pending = False ,
13+ ):
814 user = MagicMock ()
915 user .sub = sub
1016 user .email = email
@@ -17,38 +23,50 @@ def _make_user(sub="sub1", email="user@example.com", name="Test User",
1723
1824class TestListUsers :
1925 def test_header_includes_admin_column (self , mocker , capsys ):
20- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = []
26+ mocker .patch (
27+ "servicex_app.cli.user_commands.UserModel"
28+ ).query .all .return_value = []
2129 list_users ()
2230 assert "Admin" in capsys .readouterr ().out
2331
2432 def test_prints_admin_true_for_admin_user (self , mocker , capsys ):
2533 user = _make_user (admin = True )
26- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = [user ]
34+ mocker .patch (
35+ "servicex_app.cli.user_commands.UserModel"
36+ ).query .all .return_value = [user ]
2737 list_users ()
2838 assert "True" in capsys .readouterr ().out
2939
3040 def test_prints_admin_false_for_non_admin_user (self , mocker , capsys ):
3141 user = _make_user (admin = False )
32- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = [user ]
42+ mocker .patch (
43+ "servicex_app.cli.user_commands.UserModel"
44+ ).query .all .return_value = [user ]
3345 list_users ()
3446 assert "False" in capsys .readouterr ().out
3547
3648 def test_prints_pending_status (self , mocker , capsys ):
3749 user = _make_user (pending = True )
38- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = [user ]
50+ mocker .patch (
51+ "servicex_app.cli.user_commands.UserModel"
52+ ).query .all .return_value = [user ]
3953 list_users ()
4054 assert "Pending" in capsys .readouterr ().out
4155
4256 def test_prints_approved_status (self , mocker , capsys ):
4357 user = _make_user (pending = False )
44- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = [user ]
58+ mocker .patch (
59+ "servicex_app.cli.user_commands.UserModel"
60+ ).query .all .return_value = [user ]
4561 list_users ()
4662 assert "Approved" in capsys .readouterr ().out
4763
4864 def test_lists_all_users_without_filter (self , mocker , capsys ):
4965 user1 = _make_user (sub = "a" , email = "a@x.com" )
5066 user2 = _make_user (sub = "b" , email = "b@x.com" )
51- mocker .patch ("servicex_app.cli.user_commands.UserModel" ).query .all .return_value = [user1 , user2 ]
67+ mocker .patch (
68+ "servicex_app.cli.user_commands.UserModel"
69+ ).query .all .return_value = [user1 , user2 ]
5270 list_users ()
5371 out = capsys .readouterr ().out
5472 assert "a@x.com" in out
0 commit comments