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():