def setUpClass(cls):
super(TestTimersHolder, cls).setUpClass()
loader = FixturesLoader()
TestTimersHolder.MODELS = loader.load_fixtures(
fixtures_pack=PACK, fixtures_dict=FIXTURES)['triggers']
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=FIXTURES)
def setUpClass(cls):
super(TestTimersHolder, cls).setUpClass()
loader = FixturesLoader()
TestTimersHolder.MODELS = loader.load_fixtures(
fixtures_pack=PACK, fixtures_dict=FIXTURES)['triggers']
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=FIXTURES)
def setUpClass(cls):
EventletTestCase.setUpClass()
DbTestCase.setUpClass()
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
EventletTestCase.setUpClass()
DbTestCase.setUpClass()
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
super(BackstopRuleMatcherTestCase, cls).setUpClass()
fixturesloader = FixturesLoader()
# Create TriggerTypes before creation of Rule to avoid failure. Rule requires the
# Trigger and therefore TriggerType to be created prior to rule creation.
cls.models = fixturesloader.save_fixtures_to_db(
fixtures_pack=PACK, fixtures_dict=FIXTURES_TRIGGERS)
cls.models.update(fixturesloader.save_fixtures_to_db(
fixtures_pack=PACK, fixtures_dict=FIXTURES_RULES))
def setUpClass(cls):
EventletTestCase.setUpClass()
DbTestCase.setUpClass()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
EventletTestCase.setUpClass()
DbTestCase.setUpClass()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
super(BackstopRuleMatcherTest, cls).setUpClass()
fixturesloader = FixturesLoader()
# Create TriggerTypes before creation of Rule to avoid failure. Rule requires the
# Trigger and therefore TriggerType to be created prior to rule creation.
cls.models = fixturesloader.save_fixtures_to_db(
fixtures_pack=PACK, fixtures_dict=FIXTURES_TRIGGERS)
cls.models.update(
fixturesloader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=FIXTURES_RULES))
def setUpClass(cls):
ExecutionDbTestCase.setUpClass()
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
ExecutionDbTestCase.setUpClass()
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def test_create_or_update_trigger_db_simple_triggers(self):
test_fixtures = {"triggertypes": ["triggertype1.yaml"]}
loader = FixturesLoader()
fixtures = loader.save_fixtures_to_db(fixtures_pack="generic",
fixtures_dict=test_fixtures)
triggertypes = fixtures["triggertypes"]
trigger_type_ref = ResourceReference.to_string_reference(
name=triggertypes["triggertype1.yaml"]["name"],
pack=triggertypes["triggertype1.yaml"]["pack"],
)
trigger = {
"name": triggertypes["triggertype1.yaml"]["name"],
"pack": triggertypes["triggertype1.yaml"]["pack"],
"type": trigger_type_ref,
}
trigger_service.create_or_update_trigger_db(trigger)
triggers = Trigger.get_all()
self.assertTrue(
len(triggers) == 1, "Only one trigger should be created.")
self.assertTrue(
triggers[0]["name"] == triggertypes["triggertype1.yaml"]["name"])
# Try adding duplicate
trigger_service.create_or_update_trigger_db(trigger)
triggers = Trigger.get_all()
self.assertTrue(
len(triggers) == 1, "Only one trigger should be present.")
self.assertTrue(
triggers[0]["name"] == triggertypes["triggertype1.yaml"]["name"])
def test_create_or_update_trigger_db_simple_triggers(self):
test_fixtures = {
'triggertypes': ['triggertype1.yaml']
}
loader = FixturesLoader()
fixtures = loader.save_fixtures_to_db(fixtures_pack='generic', fixtures_dict=test_fixtures)
triggertypes = fixtures['triggertypes']
trigger_type_ref = ResourceReference.to_string_reference(
name=triggertypes['triggertype1.yaml']['name'],
pack=triggertypes['triggertype1.yaml']['pack'])
trigger = {
'name': triggertypes['triggertype1.yaml']['name'],
'pack': triggertypes['triggertype1.yaml']['pack'],
'type': trigger_type_ref
}
trigger_service.create_or_update_trigger_db(trigger)
triggers = Trigger.get_all()
self.assertTrue(len(triggers) == 1, 'Only one trigger should be created.')
self.assertTrue(triggers[0]['name'] == triggertypes['triggertype1.yaml']['name'])
# Try adding duplicate
trigger_service.create_or_update_trigger_db(trigger)
triggers = Trigger.get_all()
self.assertTrue(len(triggers) == 1, 'Only one trigger should be present.')
self.assertTrue(triggers[0]['name'] == triggertypes['triggertype1.yaml']['name'])
def setUpClass(cls):
EventletTestCase.setUpClass()
ExecutionDbTestCase.setUpClass()
# Override the coordinator to use the noop driver otherwise the tests will be blocked.
tests_config.parse_args(coordinator_noop=True)
coordination.COORDINATOR = None
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=TEST_FIXTURES)
def test_exception_thrown_when_rule_creation_no_trigger_yes_triggertype(self):
test_fixtures = {
'triggertypes': ['triggertype1.yaml']
}
loader = FixturesLoader()
fixtures = loader.save_fixtures_to_db(fixtures_pack='generic', fixtures_dict=test_fixtures)
triggertypes = fixtures['triggertypes']
trigger_type_ref = ResourceReference.to_string_reference(
name=triggertypes['triggertype1.yaml']['name'],
pack=triggertypes['triggertype1.yaml']['pack'])
rule = {
'name': 'fancyrule',
'trigger': {
'type': trigger_type_ref
},
'criteria': {
},
'action': {
'ref': 'core.local',
'parameters': {
'cmd': 'date'
}
}
}
rule_api = RuleAPI(**rule)
self.assertRaises(TriggerDoesNotExistException,
trigger_service.create_trigger_db_from_rule, rule_api)
def test_exception_thrown_when_rule_creation_no_trigger_yes_triggertype(
self):
test_fixtures = {"triggertypes": ["triggertype1.yaml"]}
loader = FixturesLoader()
fixtures = loader.save_fixtures_to_db(fixtures_pack="generic",
fixtures_dict=test_fixtures)
triggertypes = fixtures["triggertypes"]
trigger_type_ref = ResourceReference.to_string_reference(
name=triggertypes["triggertype1.yaml"]["name"],
pack=triggertypes["triggertype1.yaml"]["pack"],
)
rule = {
"name": "fancyrule",
"trigger": {
"type": trigger_type_ref
},
"criteria": {},
"action": {
"ref": "core.local",
"parameters": {
"cmd": "date"
}
},
}
rule_api = RuleAPI(**rule)
self.assertRaises(
TriggerDoesNotExistException,
trigger_service.create_trigger_db_from_rule,
rule_api,
)
def setUpClass(cls):
EventletTestCase.setUpClass()
DbTestCase.setUpClass()
# Override the coordinator to use the noop driver otherwise the tests will be blocked.
tests_config.parse_args(coordinator_noop=True)
coordination.COORDINATOR = None
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def setUp(self):
super(SchedulerPoliciesTest, self).setUp()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES_2)
# Policy with "post_run" application
self.policy_db = models['policies']['policy_1.yaml']
def setUp(self):
super(SchedulerPoliciesTest, self).setUp()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES_2)
# Policy with "post_run" application
self.policy_db = models['policies']['policy_1.yaml']
def setUp(self):
super(NotifierPoliciesTestCase, self).setUp()
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2common)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES_1)
# Policy with "post_run" application
self.policy_db = models['policies']['policy_4.yaml']
def setUp(self):
super(RetryPolicyTestCase, self).setUp()
# Register common policy types
register_policy_types(st2actions)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
# Instantiate policy applicator we will use in the tests
policy_db = models['policies']['policy_4.yaml']
retry_on = policy_db.parameters['retry_on']
max_retry_count = policy_db.parameters['max_retry_count']
self.policy = ExecutionRetryPolicyApplicator(policy_ref='test_policy',
policy_type='action.retry',
retry_on=retry_on,
max_retry_count=max_retry_count,
delay=0)
def setUp(self):
super(RetryPolicyTestCase, self).setUp()
# Register common policy types
register_policy_types(st2actions)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
# Instantiate policy applicator we will use in the tests
policy_db = models['policies']['policy_4.yaml']
retry_on = policy_db.parameters['retry_on']
max_retry_count = policy_db.parameters['max_retry_count']
self.policy = ExecutionRetryPolicyApplicator(policy_ref='test_policy',
policy_type='action.retry',
retry_on=retry_on,
max_retry_count=max_retry_count,
delay=0)
def setUp(self):
super(RetryPolicyTestCase, self).setUp()
# Register runners
runners_registrar.register_runners()
# Register common policy types
register_policy_types(st2actions)
loader = FixturesLoader()
models = loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
# Instantiate policy applicator we will use in the tests
policy_db = models["policies"]["policy_4.yaml"]
retry_on = policy_db.parameters["retry_on"]
max_retry_count = policy_db.parameters["max_retry_count"]
self.policy = ExecutionRetryPolicyApplicator(
policy_ref="test_policy",
policy_type="action.retry",
retry_on=retry_on,
max_retry_count=max_retry_count,
delay=0,
)
def setUpClass(cls):
super(TestExportWorker, cls).setUpClass()
fixtures_loader = FixturesLoader()
loaded_fixtures = fixtures_loader.save_fixtures_to_db(fixtures_pack=DESCENDANTS_PACK,
fixtures_dict=DESCENDANTS_FIXTURES)
TestExportWorker.saved_executions = loaded_fixtures['executions']
def setUpClass(cls):
super(PolicyTest, cls).setUpClass()
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
super(TestTimersController, cls).setUpClass()
loader = FixturesLoader()
TestTimersController.MODELS = loader.save_fixtures_to_db(
fixtures_pack=PACK, fixtures_dict=FIXTURES)["triggers"]
def setUpClass(cls):
super(TestExportWorker, cls).setUpClass()
fixtures_loader = FixturesLoader()
loaded_fixtures = fixtures_loader.save_fixtures_to_db(fixtures_pack=DESCENDANTS_PACK,
fixtures_dict=DESCENDANTS_FIXTURES)
TestExportWorker.saved_executions = loaded_fixtures['executions']
def setUpClass(cls):
super(PolicyTestCase, cls).setUpClass()
loader = FixturesLoader()
loader.save_fixtures_to_db(fixtures_pack=PACK,
fixtures_dict=TEST_FIXTURES)
def setUpClass(cls):
super(TestTimersController, cls).setUpClass()
loader = FixturesLoader()
TestTimersController.MODELS = loader.save_fixtures_to_db(fixtures_pack=PACK, fixtures_dict=FIXTURES)["triggers"]
