def test_override_block(self): def_block = loader._get_block_by_name( "get_token($user_agent, $device_sn, $os_platform, $app_version)", "def-api", self.project_mapping) test_block = { "name": "override block", "variables": [{ "var": 123 }], 'request': { 'url': '/api/get-token', 'method': 'POST', 'headers': { 'user_agent': '$user_agent', 'device_sn': '$device_sn', 'os_platform': '$os_platform', 'app_version': '$app_version' }, 'json': { 'sign': '${get_sign($user_agent, $device_sn, $os_platform, $app_version)}' } }, 'validate': [{ 'eq': ['status_code', 201] }, { 'len_eq': ['content.token', 32] }] } loader._extend_block(test_block, def_block) self.assertEqual(test_block["name"], "override block") self.assertIn( { 'check': 'status_code', 'expect': 201, 'comparator': 'eq' }, test_block["validate"]) self.assertIn( { 'check': 'content.token', 'comparator': 'len_eq', 'expect': 32 }, test_block["validate"])
def test_override_block(self): def_block = loader._get_block_by_name( "get_token($user_agent,$device_sn,$os_platform,$app_version)", "def-api") test_block = { 'name': 'override block', 'variables': [{ 'var': 123 }], 'request': { 'url': '/api/get-token', 'method': 'POST', 'headers': { 'user_agent': '$user_agent', 'device_sn': '$device_sn', 'os_platform': '$os_platform', 'app_version': '$app_version' }, 'json': { 'sign': '${get_sign($user_agent,$device_sn,$os_platform,$app_version)}' } }, 'validate': [{ 'eq': ['status_code', 201] }, { 'len_eq': ['content.token', 32] }] } loader._extend_block(test_block, def_block) assert test_block['name'] == 'override block' assert { 'check': 'status_code', 'expect': 201, 'comparator': 'eq' } in test_block['validate'] assert { 'check': 'content.token', 'expect': 32, 'comparator': 'len_eq' } in test_block['validate']