def __init__(self, producer_name, db_file_path, output_file_path): self.logger = get_logger() self.producer_name = producer_name self.spark = get_spark_session(self.logger) self.dir_name = os.path.dirname(__file__) self.db_file_path = db_file_path self.output_file_path = output_file_path
def __init__(self, producer_name, kafka_bootstrap_servers, kafka_input_topic_name, input_json_file, kafka_producer_sleep_time, db_file_path): self.logger = get_logger() self.producer_name = producer_name self.kafka_bootstrap_servers = kafka_bootstrap_servers self.kafka_input_topic_name = kafka_input_topic_name self.input_json_file = input_json_file self.kafka_producer_sleep_time = kafka_producer_sleep_time self.db_file_path = db_file_path
def __init__(self, consumer_name, kafka_bootstrap_servers, kafka_output_topic_name, kafka_window_time, output_file_dir): self.logger = get_logger() self.spark = get_spark_session(self.logger) self.grouped_schema = get_grouped_message_schema() self.consumer_name = consumer_name self.dir_name = os.path.dirname(__file__) self.kafka_bootstrap_servers = kafka_bootstrap_servers self.kafka_output_topic_name = kafka_output_topic_name self.kafka_window_time = kafka_window_time self.output_file_dir = output_file_dir
def __init__(self): self.logging = logging_util.get_logger() self.account_svc = ManageAccount() self.authorize_svc = AuthorizeTransaction()
""" This file is a unit test file which covers the unit tests for manage_account.py """ from pyscripts.business_rules.manage_account import ManageAccount from pyscripts.util import logging_util import json logging = logging_util.get_logger() mng_acc = ManageAccount() def test_generate_violations(): mng_acc.account_dict = {} violations_list = [] acc_json_data = json.loads( '{"account": {"active-card": true, "available-limit": 100}}') response_dict = mng_acc.generate_violations(violations_list, logging, acc_json_data) assert response_dict == '{"account": {"active-card": true, "available-limit": 100}, "violations": []}' mng_acc.account_dict = json.loads(response_dict) acc_json_data = json.loads( '{"account": {"active-card": true, "available-limit": 300}}') violations_list = ["account-already-initialized"] response_dict2 = mng_acc.generate_violations(violations_list, logging, acc_json_data) assert response_dict2 == '{"account": {"active-card": true, "available-limit": 100}, "violations": ["account-already-initialized"]}' def test_create_account():