def set_data_ruletype(notify_type='1', maddr='*****@*****.**'): """ テストデータの作成 """ # ルール種別 ruletype = RuleType( rule_type_id=9999, rule_type_name='pytest_name', summary=None, rule_table_name='pytest_table', generation_limit=5, group_id='pytest_com', artifact_id='pytest_oase', container_id_prefix_staging='test', container_id_prefix_product='prod', current_container_id_staging=None, current_container_id_product=None, label_count=1, unknown_event_notification=notify_type, mail_address=maddr, disuse_flag='0', last_update_timestamp=datetime.datetime(2020, 6, 4, 12, 0, 0, tzinfo=datetime.timezone.utc), last_update_user='******') ruletype.save(force_insert=True) return ruletype
def set_data(): """ テストに必要なデータをDBに登録 """ now = datetime.datetime.now(pytz.timezone('UTC')) module = import_module('web_app.models.ZABBIX_monitoring_models') ZabbixAdapter = getattr(module, 'ZabbixAdapter') ZabbixMatchInfo = getattr(module, 'ZabbixMatchInfo') try: with transaction.atomic(): # テスト用DT作成 rule_type = RuleType( rule_type_name = 'ZABBIX_TEST用', summary = '', rule_table_name = 'zabbixtest', generation_limit = 3, group_id = 'com.oase', artifact_id = 'zabbix', container_id_prefix_staging = 'testzabbix', container_id_prefix_product = 'prodzabbix', label_count = '2', last_update_timestamp = now, last_update_user = '******' ) rule_type.save(force_insert=True) # テスト用ラベル作衛 label0 = DataObject( rule_type_id = rule_type.rule_type_id, conditional_name = 'トリガーID', label = 'label0', conditional_expression_id = 1, last_update_timestamp = now, last_update_user = '******' ) label0.save(force_insert=True) label1 = DataObject( rule_type_id = rule_type.rule_type_id, conditional_name = '説明', label = 'label1', conditional_expression_id = 2, last_update_timestamp = now, last_update_user = '******' ) label1.save(force_insert=True) # テスト用ZABBIX監視アダプタ作成 zabbix = ZabbixAdapter( zabbix_disp_name = 'ZABBIX', hostname = 'pytest-host', username = '******', password = '******', protocol = 'http', port = 80, rule_type_id = rule_type.rule_type_id, last_update_timestamp = now, last_update_user = '******' ) zabbix.save(force_insert=True) # テスト用ZABBIX突合情報作成 match1 = ZabbixMatchInfo( zabbix_adapter_id = zabbix.zabbix_adapter_id, data_object_id = label0.data_object_id, zabbix_response_key = 'triggerid', last_update_timestamp = now, last_update_user = '******' ) match1.save(force_insert=True) match2 = ZabbixMatchInfo( zabbix_adapter_id = zabbix.zabbix_adapter_id, data_object_id = label1.data_object_id, zabbix_response_key = 'description', last_update_timestamp = now, last_update_user = '******' ) match2.save(force_insert=True) return rule_type.rule_type_id, zabbix.zabbix_adapter_id except Exception as e: print(e)