def test_desk_staff_have_infinite_quota(mysql_connection):
assert (
get_quota(mysql_connection, 'pubstaff') ==
UserQuota('pubstaff', 500, 500)
)
assert (
get_quota(mysql_connection, 'testopstaff') ==
UserQuota('testopstaff', 500, 500)
)
def assert_quota(c, user, diff_daily, diff_semesterly):
"""Assert the quota for a user is what we expect.
Typically, you want to pass a negative number for diff_daily and
diff_semesterly. This number is added to the start quota before assertion.
"""
start = 1000, 10000
with mock.patch('ocflib.printing.quota.daily_quota', return_value=start[0]), \
mock.patch('ocflib.printing.quota.SEMESTERLY_QUOTA', start[1]):
assert (get_quota(c, user) == UserQuota(
user, FAKE_DAILY_QUOTA + diff_daily,
FAKE_SEMESTERLY_QUOTA + diff_semesterly))
def assert_quota(c, user, diff_daily, diff_semesterly):
"""Assert the quota for a user is what we expect.
Typically, you want to pass a negative number for diff_daily and
diff_semesterly. This number is added to the start quota before assertion.
"""
start = 1000, 10000
with mock.patch('ocflib.printing.quota.daily_quota', return_value=start[0]), \
mock.patch('ocflib.printing.quota.SEMESTERLY_QUOTA', start[1]):
assert (
get_quota(c, user) ==
UserQuota(user, FAKE_DAILY_QUOTA + diff_daily, FAKE_SEMESTERLY_QUOTA + diff_semesterly)
)
def test_desk_staff_have_infinite_quota(mysql_connection):
assert (get_quota(mysql_connection,
'testopstaff') == UserQuota('testopstaff', 500, 500))
def test_non_existent_users_have_zero_quota(mysql_connection):
assert (get_quota(mysql_connection,
'nonexist') == UserQuota('nonexist', 0, 0))
def test_groups_have_zero_quota(mysql_connection):
assert (get_quota(mysql_connection, 'ggroup') == UserQuota('ggroup', 0, 0))
#!/usr/bin/env python3
"""Print the quota of a user."""
from ocflib.printing.quota import get_connection
from ocflib.printing.quota import get_quota
if __name__ == '__main__':
with get_connection() as c:
print(get_quota(c, input('user: ')))
def test_pubstaff_has_infinite_quota(mysql_connection):
assert (
get_quota(mysql_connection, 'pubstaff') ==
UserQuota('pubstaff', 500, 500)
)
def test_non_existent_users_have_zero_quota(mysql_connection):
assert (
get_quota(mysql_connection, 'nonexist') ==
UserQuota('nonexist', 0, 0)
)
def test_groups_have_zero_quota(mysql_connection):
assert (
get_quota(mysql_connection, 'ggroup') ==
UserQuota('ggroup', 0, 0)
)
def test_pubstaff_has_infinite_quota(mysql_connection):
assert (
get_quota(mysql_connection, 'pubstaff') ==
UserQuota('pubstaff', 500, 500)
)
#!/usr/bin/env python3
"""Print the quota of a user."""
from ocflib.printing.quota import get_connection
from ocflib.printing.quota import get_quota
if __name__ == "__main__":
with get_connection() as c:
print(get_quota(c, input("user: ")))
