def test_disable_partition_policy():
hwc = HELLO_WORLD(
production=True,
priority=200,
service=True,
cron_collision_policy='RUN_OVERLAP',
partition_policy=PystachioPartitionPolicy(reschedule=False),
constraints={
'dedicated': 'root',
'cpu': 'x86_64'
},
environment='prod')
job = convert_pystachio_to_thrift(hwc)
assert job.taskConfig.partitionPolicy == PartitionPolicy(False, 0)
def test_config_with_options():
hwc = HELLO_WORLD(production=True,
priority=200,
service=True,
cron_collision_policy='RUN_OVERLAP',
partition_policy=PystachioPartitionPolicy(delay_secs=10),
constraints={
'dedicated': 'root',
'cpu': 'x86_64'
},
environment='prod')
job = convert_pystachio_to_thrift(hwc)
assert job.instanceCount == 1
tti = job.taskConfig
assert tti.production
assert tti.priority == 200
assert tti.isService
assert job.cronCollisionPolicy == CronCollisionPolicy.RUN_OVERLAP
assert len(tti.constraints) == 2
assert job.key.environment == 'prod'
assert tti.partitionPolicy == PartitionPolicy(True, 10)