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']