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) )
#!/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: ")))