11import unittest
22from unittest .mock import Mock , patch
3- from dapi .jobs import jobs
3+ from dapi .jobs import generate_job_request
44from datetime import datetime
55
66
@@ -20,10 +20,10 @@ def setUp(self):
2020 self .app_info_mock .jobAttributes .execSystemLogicalQueue = "normal"
2121 self .t_mock .apps .getAppLatestVersion .return_value = self .app_info_mock
2222
23- @patch ("dapi.jobs.jobs. datetime" )
23+ @patch ("dapi.jobs.datetime" )
2424 def test_generate_job_info_default (self , mock_datetime ):
2525 mock_datetime .now .return_value = datetime (2023 , 5 , 1 , 12 , 0 , 0 )
26- result = jobs . generate_job_info (
26+ result = generate_job_request (
2727 self .t_mock , self .app_name , self .input_uri , self .input_file
2828 )
2929 self .assertEqual (result ["name" ], f"{ self .app_name } _20230501_120000" )
@@ -52,11 +52,10 @@ def test_generate_job_info_custom(self):
5252 custom_cores_per_node = 4
5353 custom_queue = "high-priority"
5454 custom_allocation = "project123"
55- result = jobs . generate_job_info (
55+ result = generate_job_request (
5656 self .t_mock ,
5757 self .app_name ,
5858 self .input_uri ,
59- self .input_file ,
6059 job_name = custom_job_name ,
6160 max_minutes = custom_max_minutes ,
6261 node_count = custom_node_count ,
@@ -77,15 +76,11 @@ def test_generate_job_info_custom(self):
7776 def test_generate_job_info_invalid_app (self ):
7877 self .t_mock .apps .getAppLatestVersion .side_effect = Exception ("Invalid app" )
7978 with self .assertRaises (Exception ):
80- jobs .generate_job_info (
81- self .t_mock , "invalid-app" , self .input_uri , self .input_file
82- )
79+ generate_job_request (self .t_mock , "invalid-app" , self .input_uri )
8380
8481 def test_generate_job_info_opensees (self ):
8582 opensees_app_name = "opensees-express"
86- result = jobs .generate_job_info (
87- self .t_mock , opensees_app_name , self .input_uri , self .input_file
88- )
83+ result = generate_job_request (self .t_mock , opensees_app_name , self .input_uri )
8984 self .assertIn ("parameterSet" , result )
9085 self .assertIn ("envVariables" , result ["parameterSet" ])
9186 self .assertEqual (
0 commit comments