@@ -21,16 +21,12 @@ def auto_enable_work_pools(enable_work_pools):
21
21
class TestCreate :
22
22
async def test_create_work_pool (self , orion_client ):
23
23
pool_name = "my-pool"
24
- infra_type = "process"
25
24
res = await run_sync_in_worker_thread (
26
25
invoke_and_assert ,
27
- f"work-pool create { pool_name } -t { infra_type } " ,
26
+ f"work-pool create { pool_name } " ,
28
27
)
29
28
assert res .exit_code == 0
30
- assert (
31
- f"Created work pool { pool_name !r} with infrastructure type { infra_type !r} "
32
- in res .output
33
- )
29
+ assert f"Created work pool { pool_name !r} " in res .output
34
30
client_res = await orion_client .read_work_pool (pool_name )
35
31
assert client_res .name == pool_name
36
32
assert isinstance (client_res , WorkPool )
@@ -39,7 +35,7 @@ async def test_default_template(self, orion_client):
39
35
pool_name = "my-pool"
40
36
res = await run_sync_in_worker_thread (
41
37
invoke_and_assert ,
42
- f"work-pool create { pool_name } -t process " ,
38
+ f"work-pool create { pool_name } " ,
43
39
)
44
40
assert res .exit_code == 0
45
41
client_res = await orion_client .read_work_pool (pool_name )
@@ -49,7 +45,7 @@ async def test_default_paused(self, orion_client):
49
45
pool_name = "my-pool"
50
46
res = await run_sync_in_worker_thread (
51
47
invoke_and_assert ,
52
- f"work-pool create { pool_name } -t process " ,
48
+ f"work-pool create { pool_name } " ,
53
49
)
54
50
assert res .exit_code == 0
55
51
client_res = await orion_client .read_work_pool (pool_name )
@@ -59,7 +55,7 @@ async def test_paused_true(self, orion_client):
59
55
pool_name = "my-pool"
60
56
res = await run_sync_in_worker_thread (
61
57
invoke_and_assert ,
62
- f"work-pool create { pool_name } --paused -t process " ,
58
+ f"work-pool create { pool_name } --paused" ,
63
59
)
64
60
assert res .exit_code == 0
65
61
client_res = await orion_client .read_work_pool (pool_name )
0 commit comments