# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from st2common.exceptions.triggers import TriggerDoesNotExistException from st2common.models.api.rule import RuleAPI from st2common.models.system.common import ResourceReference from st2common.models.db.trigger import TriggerDB from st2common.persistence.trigger import (Trigger, TriggerType) import st2common.services.triggers as trigger_service from st2tests.base import CleanDbTestCase from st2tests.fixturesloader import FixturesLoader MOCK_TRIGGER = TriggerDB() MOCK_TRIGGER.id = 'trigger-test.id' MOCK_TRIGGER.name = 'trigger-test.name' MOCK_TRIGGER.pack = 'dummy_pack_1' MOCK_TRIGGER.parameters = {} MOCK_TRIGGER.type = 'dummy_pack_1.trigger-type-test.name' class TriggerServiceTests(CleanDbTestCase): def test_create_trigger_db_from_rule(self): test_fixtures = { 'rules': ['cron_timer_rule_1.yaml', 'cron_timer_rule_3.yaml'] } loader = FixturesLoader() fixtures = loader.load_fixtures(fixtures_pack='generic', fixtures_dict=test_fixtures) rules = fixtures['rules']
# limitations under the License. import bson import copy import mock from st2common.models.db.action import ActionDB from st2common.models.db.rule import RuleDB, ActionExecutionSpecDB from st2common.models.db.trigger import TriggerDB, TriggerInstanceDB from st2common.util import reference from st2common.util import date as date_utils from st2reactor.rules.filter import RuleFilter from st2tests import DbTestCase MOCK_TRIGGER = TriggerDB() MOCK_TRIGGER.id = bson.ObjectId() MOCK_TRIGGER.name = 'trigger-test.name' MOCK_TRIGGER.pack = 'dummy_pack_1' MOCK_TRIGGER.type = 'system.test' MOCK_TRIGGER_INSTANCE = TriggerInstanceDB() MOCK_TRIGGER_INSTANCE.id = bson.ObjectId() MOCK_TRIGGER_INSTANCE.trigger = MOCK_TRIGGER.get_reference().ref MOCK_TRIGGER_INSTANCE.payload = { 'p1': 'v1', 'p2': 'preYYYpost', 'bool': True, 'int': 1, 'float': 0.8 } MOCK_TRIGGER_INSTANCE.occurrence_time = date_utils.get_datetime_utc_now()
import mock from st2common.models.db.trigger import TriggerDB from st2common.transport.publishers import PoolPublisher import st2reactor.container.utils as container_utils from st2tests.base import CleanDbTestCase MOCK_TRIGGER_TYPE = {} MOCK_TRIGGER_TYPE['id'] = 'trigger-type-test.id' MOCK_TRIGGER_TYPE['name'] = 'trigger-type-test.name' MOCK_TRIGGER_TYPE['pack'] = 'dummy_pack_1' MOCK_TRIGGER_TYPE['parameters_schema'] = {} MOCK_TRIGGER_TYPE['payload_schema'] = {} MOCK_TRIGGER = TriggerDB() MOCK_TRIGGER.id = 'trigger-test.id' MOCK_TRIGGER.name = 'trigger-test.name' MOCK_TRIGGER.pack = 'dummy_pack_1' MOCK_TRIGGER.parameters = {} MOCK_TRIGGER.type = 'dummy_pack_1.trigger-type-test.name' @mock.patch.object(PoolPublisher, 'publish', mock.MagicMock()) class ContainerUtilsTest(CleanDbTestCase): def test_create_trigger_instance_invalid_trigger(self): trigger_instance = 'dummy_pack.footrigger' instance = container_utils.create_trigger_instance(trigger_instance, {}, None) self.assertTrue(instance is None)
import bson import copy import mock from st2common.models.db.action import ActionDB from st2common.models.db.rule import RuleDB, ActionExecutionSpecDB from st2common.models.db.trigger import TriggerDB, TriggerInstanceDB from st2common.util import reference from st2common.util import date as date_utils from st2reactor.rules.filter import RuleFilter from st2tests import DbTestCase MOCK_TRIGGER = TriggerDB() MOCK_TRIGGER.id = bson.ObjectId() MOCK_TRIGGER.name = "trigger-test.name" MOCK_TRIGGER.pack = "dummy_pack_1" MOCK_TRIGGER.type = "system.test" MOCK_TRIGGER_INSTANCE = TriggerInstanceDB() MOCK_TRIGGER_INSTANCE.id = bson.ObjectId() MOCK_TRIGGER_INSTANCE.trigger = MOCK_TRIGGER.get_reference().ref MOCK_TRIGGER_INSTANCE.payload = {"p1": "v1", "p2": "preYYYpost", "bool": True, "int": 1, "float": 0.8} MOCK_TRIGGER_INSTANCE.occurrence_time = date_utils.get_datetime_utc_now() MOCK_ACTION = ActionDB() MOCK_ACTION.id = bson.ObjectId() MOCK_ACTION.name = "action-test-1.name" MOCK_RULE_1 = RuleDB()