Untitled

"""testing common_utils"""
import os
import sys
import unittest
from unittest.mock import patch, MagicMock


sys.path.append("../../src")
sys.modules["awswrangler"] = MagicMock()
sys.modules["data_reservoir"] = MagicMock()
sys.modules["data_reservoir.common_services"] = MagicMock()
sys.modules["data_reservoir.master_data"] = MagicMock()
sys.modules["data_reservoir.s3_services"] = MagicMock()
sys.modules["requests"] = MagicMock()
os.environ["env"] = "dev"
os.environ["AWS_DEFAULT_REGION"] = "us-east-1"
os.environ['reservoir_bucket'] = "test_bucket"
os.environ["to_emails"] = "test_to_emails"
os.environ["signature"] = "test_signature"
os.environ["cc_emails"] = "test_cc_emails"
from src import common_utils


class Response:
    def __init__(self):
        self.status_code=200


class LambdaTest(unittest.TestCase):
    """Class for passing lambda test portion of pipeline."""    
    def test_write_invmnt_trxn_data(self):
        """Test write_invmnt_trxn_data"""
        mock_common_utils_patcher = patch('src.common_utils.requests.post').start()
        mock_common_utils_patcher.return_value = Response()
        mock_response = common_utils.write_invmnt_trxn_data({"body": ["data"]}, "base_url", "token")
        self.assertEqual(200, mock_response.status_code)
        patch.stopall()