def test_process_protocol(self):
trans = HtmlTransformer()
protocol = [
ApiCall('request', Scheme({'id': int})),
ApiCall('response', Scheme({'status': AnyOf('ok', 'error')})),
]
res = trans.process_protocol(protocol)
self.assertEquals('<table class="api_protocol">'
'<tr class="api_call_request">'
'<td class="api_call_name">request</td>'
'<td class="api_call_scheme">'
'<table class="api_dict">'
'<tr><td colspan="2">{</td></tr>'
'<tr>'
'<td class="api_dict_key">id</td>'
'<td class="api_dict_value">int</td>'
'</tr>'
'<tr><td colspan="2">}</td></tr>'
'</table>'
'</td>'
'<td class="api_call_description"></td>'
'</tr>'
'<tr class="api_call_response">'
'<td class="api_call_name">response</td>'
'<td class="api_call_scheme">'
'<table class="api_dict">'
'<tr><td colspan="2">{</td></tr>'
'<tr>'
'<td class="api_dict_key">status</td>'
'<td class="api_dict_value">'
'<table class="api_any_of">'
'<tr class="api_any_of_case"><td>ok</td></tr>'
'<tr class="api_any_of_separator"><td>OR</td></tr>'
'<tr class="api_any_of_case"><td>error</td></tr>'
'</table>'
'</td>'
'</tr>'
'<tr><td colspan="2">}</td></tr>'
'</table>'
'</td>'
'<td class="api_call_description"></td>'
'</tr>'
'</table>', res)
def get_api_scheme(self, data, req_info, session, curs=None):
trans = HtmlTransformer()
return response_ok(scheme=trans.process_protocol(protocol))
