def test_no_parameters(self): files = file_download_feature.substitute_variable_values( [], ['/home/user/test.txt'], [], '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/user/test.txt'])
def test_replace_username_and_param(self): param1 = create_parameter_model('param1') files = file_download_feature.substitute_variable_values( [param1], ['/home/${auth.username}/${param1}.txt'], {'param1': 'val1'}, '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/user_X/val1.txt'])
def test_replace_audit_name(self): param1 = create_parameter_model('param1', no_value=True) files = file_download_feature.substitute_variable_values( [param1], ['/home/user/${auth.audit_name}.txt'], {'param1': 'val1'}, '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/user/127.0.0.1.txt'])
def test_skip_secure_replace(self): param1 = create_parameter_model('param1', secure=True) files = file_download_feature.substitute_variable_values( [param1], ['/home/user/${param1}.txt'], {'param1': 'val1'}, '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/user/${param1}.txt'])
def test_no_pattern_match(self): param1 = create_parameter_model('param1') files = file_download_feature.substitute_variable_values( [param1], ['/home/user/${paramX}.txt'], {'param1': 'val1'}, '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/user/${paramX}.txt'])
def test_two_replaces_in_two_files(self): parameters = [] parameters.append(create_parameter_model('param1', all_parameters=parameters)) parameters.append(create_parameter_model('param2', all_parameters=parameters)) files = file_download_feature.substitute_variable_values( parameters, ['/home/${param2}/${param1}.txt', '/tmp/${param2}.txt', '/${param1}'], {'param1': 'val1', 'param2': 'val2'}, '127.0.0.1', 'user_X') self.assertEqual(files, ['/home/val2/val1.txt', '/tmp/val2.txt', '/val1'])