def setUp(self): curr_dir = os.path.dirname(__file__) file_path = 'assets/config.yaml' # get the yaml file with open(os.path.join(curr_dir, file_path), 'r') as stream: try: self.normal_content = yaml.safe_load(stream) except yaml.YAMLError as err: Logger.error('error load yaml file ' + str(err)) self.vault_data_key = {'db': {'password': '******'}} self.kv_sample = { 'appmode': 'development', 'db': { 'username': '******', 'host': 'localhost', 'port': 1234 }, 'elasticsearch': { 'host': 'localhost', 'port': 9200 } } Outfit.setup(path='tests/assets/config.yaml')
def test_setup_log_py(self): self.delete_all_log_files() Outfit.setup('./tests/assets/config-log-py.yaml') Logger.info('test_info') with open('tests/logs/info.log', 'r') as finfo: temp_info = finfo.readlines() last_line = temp_info[len(temp_info) - 1] self.assertTrue('test_info' in last_line) Logger.debug('test_debug') with open('tests/logs/debug.log', 'r') as fdebug: temp_debug = fdebug.readlines() last_line = temp_debug[len(temp_debug) - 1] self.assertTrue('test_debug' in last_line) Logger.error('test_error') with open('tests/logs/errors.log', 'r') as ferrors: temp_errors = ferrors.readlines() last_line = temp_errors[len(temp_errors) - 1] self.assertTrue('test_error' in last_line) Logger.critical('test_critical') with open('tests/logs/errors.log', 'r') as fcritical: temp_critical = fcritical.readlines() last_line = temp_critical[len(temp_critical) - 1] self.assertTrue('test_critical' in last_line)
def test_setup_log_consulkv_failed(self, mock_kv): self.delete_all_log_files() from outfit.utils.io import load_yaml mock_kv.return_value = logging Outfit.setup('./tests/assets/config-log-kv.yaml') Logger.info('test_info') with open('tests/test_logs/info.log', 'r') as finfo: temp_info = finfo.readlines() print(temp_info) print(len(temp_info)) last_line = temp_info[len(temp_info) - 1] self.assertTrue('test_info' in last_line) Logger.debug('test_debug') with open('tests/test_logs/debug.log', 'r') as fdebug: temp_debug = fdebug.readlines() last_line = temp_debug[len(temp_debug) - 1] self.assertTrue('test_debug' in last_line) Logger.error('test_error') with open('tests/test_logs/errors.log', 'r') as ferrors: temp_errors = ferrors.readlines() last_line = temp_errors[len(temp_errors) - 1] self.assertTrue('test_error' in last_line) Logger.critical('test_critical') with open('tests/test_logs/errors.log', 'r') as fcritical: temp_critical = fcritical.readlines() last_line = temp_critical[len(temp_critical) - 1] self.assertTrue('test_critical' in last_line)
def main(): tShirt = Tops( "637387222", "Tretch Muscle Fit Polo", "A comfortable, cotton-blend polo made with stretch, featuring an icon at left chest. Designed in a muscle-defining, super slim fit through the chest, waist and arms. Muscle Fit. Imported", "White", "Flat", "Cotton", "Male", "Hollister", "Classic", "N/A") short = Bottons("001987623", "Minimal pink short", "Minimalistic and moder short great for the summer", "Pink", "Flat", "Cotton", "Male", "Hollister", "Modern", "N/A") toms = Shoewear("908876789", "Ericsons", "Gret comfort toms ideal for hot climates", "White", "Flat", "Polyester", "Unisex", "Toms", "Modern", "N/A") acc = Accessories("099897789", "Leo Dimond ring", "2k white gold luxury ring", "Silver", "N/A", "N/A", "Male", "Leo Dimond", "Classic", "N/A") suit = Fullbody("N/A", "N/A", "N/A", "N/a", "N/A", "N/A", "N/A", "N/A", "N/A", "N/A") firstOutfit = Outfit(tShirt, short, toms, acc, suit) def getItemInList(): topDict = {} topDict["Product Number"] = tShirt.getProductNumber() topDict["Product Name"] = tShirt.getProductName() topDict["Product Description"] = tShirt.getProductDescription() topDict["Color"] = tShirt.getColor() topDict["Texture"] = tShirt.getTexture() topDict["Material"] = tShirt.getMaterial() topDict["Sex"] = tShirt.getSex() topDict["Brand"] = tShirt.getBrand() topDict["Style"] = tShirt.getStyle() topDict["Pattern"] = tShirt.getPattern() bottomDict = {} bottomDict["Product Number"] = short.getProductNumber() bottomDict["Product Name"] = short.getProductName() bottomDict["Product Description"] = short.getProductDescription() bottomDict["Color"] = short.getColor() bottomDict["Texture"] = short.getTexture() bottomDict["Material"] = short.getMaterial() bottomDict["Sex"] = short.getSex() bottomDict["Brand"] = short.getBrand() bottomDict["Style"] = short.getStyle() bottomDict["Pattern"] = short.getPattern() print(topDict) print(bottomDict) getItemInList() print(firstOutfit.getTop()) print(firstOutfit.getBottom()) print(firstOutfit.getShoeware()) print(firstOutfit.getFullbody()) print(firstOutfit.getAccesory()) print(firstOutfit.getOutfit())
def test_setup_log_consulkv(self, mock_kv): self.delete_all_log_files() from outfit.utils.io import load_yaml mock_kv.return_value = logging #result = convert_yaml(self.normal_content) Outfit.setup('./tests/assets/config-log-kv.yaml') m = minfo = mock_open( read_data='INFO:test_utilslogger.py(50)> test_info') mdebug = mock_open( read_data='ERROR:test_utilslogger.py(62)> test_debug') merror = mock_open( read_data='ERROR:test_utilslogger.py(62)> test_error') mcritical = mock_open( read_data='ERROR:test_utilslogger.py(62)> test_critical') m.side_effect = [ minfo.return_value, mdebug.return_value, merror.return_value, mcritical.return_value ] with patch('builtins.open', m): Logger.info('test_info') with open('tests/test_logs/info.log', 'r') as finfo: temp_info = finfo.read() last_line = temp_info[len(temp_info) - 1] self.assertTrue('test_info' in temp_info) Logger.debug('test_debug') with open('tests/test_logs/debug.log', 'r') as fdebug: temp_debug = fdebug.readlines() last_line = temp_debug[len(temp_debug) - 1] self.assertTrue('test_debug' in last_line) Logger.error('test_error') with open('tests/test_logs/errors.log', 'r') as ferrors: temp_errors = ferrors.readlines() last_line = temp_errors[len(temp_errors) - 1] self.assertTrue('test_error' in last_line) Logger.critical('test_critical') with open('tests/test_logs/errors.log', 'r') as fcritical: temp_critical = fcritical.readlines() last_line = temp_critical[len(temp_critical) - 1] self.assertTrue('test_critical' in last_line)
def setUp(self): curr_dir = os.path.dirname(__file__) file_path = 'assets/config.yaml' # get the yaml file with open(os.path.join(curr_dir, file_path), 'r') as stream: try: self.normal_content = yaml.safe_load(stream) except yaml.YAMLError as err: Logger.error('error load yaml file ' + str(err)) self.vault_data_key = {'db': {'password': '******'}} self.kv_sample = { 'appmode': 'development', 'db': { 'username': '******', 'host': 'localhost', 'port': 1234 }, 'elasticsearch': { 'host': 'localhost', 'port': 9200 } } Outfit.setup(path='tests/assets/config.yaml') os.environ['host_vault'] = 'localhost' os.environ['port_vault'] = '12345' os.environ['scheme_vault'] = 'http' os.environ['token_vault'] = 'qwerty123' os.environ['path_vault'] = 'application' os.environ['host_consul'] = 'localhost' os.environ['port_consul'] = '12345' os.environ['scheme_consul'] = 'http' os.environ['token_consul'] = 'qwerty123' os.environ['path_consul'] = 'application' os.environ['LOG_MODE'] = 'development' os.environ['LOG_LOCATION'] = './tests/assets/logging.yaml' os.environ['LOG_TYPE'] = 'yaml_file' os.environ['DEFAULT_TYPE'] = 'yaml_file' os.environ['DEFAULT_LOCATION'] = './tests/assets/logging.yaml'
def __init__(self, outfit_name: str, guild_id: int, member_role_id: int, *args, **kwargs): super().__init__(*args, **kwargs) self.guild_id = guild_id self.member_role_id = member_role_id self.outfit = Outfit(name=outfit_name) self.database = Database(tag=self.outfit.alias)() self.Member = Member(database=self.database.members) self.Ops = Ops self.ops_list = dict() self.add_commands()
def test_env_var(self, mock_consul_kv, mock_vault_kv): mock_consul_kv.return_value = [ None, { 'Value': json.dumps(self.kv_sample).encode() } ] mock_vault_kv.return_value = {'data': secret_kv} Outfit.setup('tests/assets/config-env.yaml') vault = VaultCon() consul = ConsulCon() result_vault = vault.get_secret_kv() result_consul = consul.get_kv() result = merge_dict(result_vault, result_consul) self.assertEqual(result['big_query']['client_id'], '1234567', ' client id from vault not match') self.assertEqual(result['appmode'], 'development', ' appmode from consul not match')