def test_check_positive_integer_passes_on_positive_int():
thrown = None
try:
validation.check_positive_integer(100)
except Exception as ex:
thrown = ex
assert thrown is None
async def update_global_agent_config(self,
sleep_min: int = None,
sleep_max: int = None,
watchdog: int = None,
untrusted_timer: int = None,
implant_name: str = None,
bootstrap_abilities: List[str] = None,
deadman_abilities=None):
set_config = self._config_interface.set_config
if sleep_min is not None:
validation.check_positive_integer(sleep_min, name='sleep_min')
set_config(name='agents', prop='sleep_min', value=sleep_min)
if sleep_max is not None:
validation.check_positive_integer(sleep_max, name='sleep_max')
set_config(name='agents', prop='sleep_max', value=sleep_max)
if untrusted_timer is not None:
validation.check_positive_integer(untrusted_timer,
name='untrusted_timer')
set_config(name='agents',
prop='untrusted_timer',
value=untrusted_timer)
if watchdog is not None:
validation.check_positive_integer(watchdog, name='watchdog')
set_config(name='agents', prop='watchdog', value=watchdog)
if implant_name is not None:
validation.check_not_empty_string(implant_name,
name='implant_name')
set_config(name='agents', prop='implant_name', value=implant_name)
if bootstrap_abilities is not None:
await self._update_agent_ability_list_property(
bootstrap_abilities, 'bootstrap_abilities')
if deadman_abilities is not None:
await self._update_agent_ability_list_property(
deadman_abilities, 'deadman_abilities')
def test_check_positive_integer_throws_on_non_int():
with pytest.raises(errors.DataValidationError):
validation.check_positive_integer(None)