def select_accounting_function(args, conn, output_file, parser):
if args.func == "view_user":
aclif.view_user(conn, args.username)
elif args.func == "add_user":
aclif.add_user(
conn,
args.username,
args.bank,
args.admin_level,
args.shares,
args.max_jobs,
args.max_wall_pj,
)
elif args.func == "delete_user":
aclif.delete_user(conn, args.username, args.bank)
elif args.func == "edit_user":
aclif.edit_user(conn, args.username, args.field, args.new_value)
elif args.func == "view_job_records":
jobs.view_job_records(
conn,
output_file,
jobid=args.jobid,
user=args.user,
before_end_time=args.before_end_time,
after_start_time=args.after_start_time,
)
elif args.func == "add_bank":
aclif.add_bank(conn, args.bank, args.shares, args.parent_bank)
elif args.func == "view_bank":
aclif.view_bank(conn, args.bank)
elif args.func == "delete_bank":
aclif.delete_bank(conn, args.bank)
elif args.func == "edit_bank":
aclif.edit_bank(conn, args.bank, args.shares)
elif args.func == "update_usage":
jobs_conn = establish_sqlite_connection(args.job_archive_db_path)
jobs.update_job_usage(conn, jobs_conn, args.priority_decay_half_life)
else:
print(parser.print_usage())
def test_10_no_options_passed(self):
my_dict = {}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 18)
def test_09_multiple_params(self):
my_dict = {"user": "******", "after_start_time": time.time()}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 1)
def test_08_by_user_success(self):
my_dict = {"user": "******"}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 2)
def test_07_by_user_failure(self):
my_dict = {"user": "******"}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 0)
def test_06_before_end_time_none(self):
my_dict = {"before_end_time": 0}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 0)
def test_05_before_end_time_all(self):
my_dict = {"before_end_time": time.time()}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 18)
def test_04_after_start_time_none(self):
my_dict = {"after_start_time": time.time()}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 0)
def test_03_after_start_time_all(self):
my_dict = {"after_start_time": 0}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 18)
def test_02_with_jobid_failure(self):
my_dict = {"jobid": 000}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 0)
def test_01_with_jobid_valid(self):
my_dict = {"jobid": 102}
job_records = jobs.view_job_records(jobs_conn, op, **my_dict)
self.assertEqual(len(job_records), 1)
