Skip to content

Commit dab0930

Browse files
committed
Merge branch '1172-rest-endpoint-to-report-users-and-transform-statistics' of https://github.com/ssl-hep/ServiceX into 1172-rest-endpoint-to-report-users-and-transform-statistics
2 parents 7da71f3 + 51a7d59 commit dab0930

File tree

3 files changed

+38
-14
lines changed

3 files changed

+38
-14
lines changed

servicex_app/servicex_app_test/test_cli_user_commands.py

Lines changed: 26 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,14 @@
33
from 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

1824
class 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

servicex_app/servicex_app_test/web/admin/test_admin_auth_mixin.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,7 @@ def test_is_admin_false_with_jwt_non_admin_user(self, auth_client, mixin, mocker
5858
with auth_client.application.test_request_context():
5959
assert mixin._is_admin() is False
6060

61-
def test_is_admin_false_on_no_authorization_error(
62-
self, auth_client, mixin, mocker
63-
):
61+
def test_is_admin_false_on_no_authorization_error(self, auth_client, mixin, mocker):
6462
mocker.patch(
6563
"servicex_app.web.admin.verify_jwt_in_request",
6664
side_effect=NoAuthorizationError("no token"),

servicex_app/servicex_app_test/web/admin/test_report_views.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,13 @@ def _setup(results):
5454

5555
return _setup
5656

57-
def _make_mock_user(self, name="Jane Doe", email="jane@example.com",
58-
institution="UChicago", experiment="ATLAS"):
57+
def _make_mock_user(
58+
self,
59+
name="Jane Doe",
60+
email="jane@example.com",
61+
institution="UChicago",
62+
experiment="ATLAS",
63+
):
5964
user = MagicMock()
6065
user.name = name
6166
user.email = email
@@ -71,7 +76,10 @@ def test_write_csv_outputs_header_row(self, mock_db_query):
7176
output.seek(0)
7277
rows = list(csv.reader(output))
7378
assert rows[0] == [
74-
"Name", "Email", "Institution", "Experiment",
79+
"Name",
80+
"Email",
81+
"Institution",
82+
"Experiment",
7583
"Transforms (Last 30 Days)",
7684
]
7785

0 commit comments

Comments
 (0)