flutter/apidash
1
0
Fork 0
mirror of https://github.com/foss42/apidash.git synced 2025-05-26 02:36:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add test for python (request) codegen

Browse Source
This commit is contained in:
Mixel2004
2023-07-04 22:34:46 +05:30
parent ac4a2160bf
commit 1b143af680
1 changed files with 188 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

188
test/codegen/python_request_codegen_test.dart Normal file
View File

@ -0,0 +1,188 @@
import 'package:apidash/codegen/python/pkg_request.dart';
import 'package:apidash/models/models.dart' show KVRow, RequestModel;
import 'package:test/test.dart';
import 'package:apidash/consts.dart';
void main() {
group('PythonRequestCodeGen', () {
final pythonRequestCodeGen = PythonRequestCodeGen();
test('getCode returns valid code for GET request', () {
const requestModel = RequestModel(
url: 'https://jsonplaceholder.typicode.com/todos/1',
method: HTTPVerb.get,
id: '',
);
const expectedCode = """import requests
import json
def main():
url = 'https://jsonplaceholder.typicode.com/todos/1'
response = requests.get(
url
)
status_code = response.status_code
if 200 <= status_code < 300:
print('Status Code:', status_code)
print('Response Body:', response.text)
else:
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main()""";
expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
});
test('getCode returns valid code for POST request', () {
const requestModel = RequestModel(
url: 'https://jsonplaceholder.typicode.com/posts',
method: HTTPVerb.post,
requestBody: '{"title": "foo","body": "bar","userId": 1}',
requestBodyContentType: ContentType.json,
id: '1',
);
const expectedCode = """import requests
import json
def main():
url = 'https://jsonplaceholder.typicode.com/posts'
data = '''{"title": "foo","body": "bar","userId": 1}'''
data = json.loads(data)
headers = {
"content-type": "application/json"
}
response = requests.post(
url, headers=headers, data=data
)
status_code = response.status_code
if 200 <= status_code < 300:
print('Status Code:', status_code)
print('Response Body:', response.text)
else:
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main()""";
expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
});
test('getCode returns valid code for DELETE request', () {
const requestModel = RequestModel(
url: 'https://jsonplaceholder.typicode.com/posts/1',
method: HTTPVerb.delete,
requestBody: '{"title": "foo","body": "bar","userId": 1}',
requestBodyContentType: ContentType.json,
id: '1',
);
const expectedCode = """import requests
import json
def main():
url = 'https://jsonplaceholder.typicode.com/posts/1'
data = '''{"title": "foo","body": "bar","userId": 1}'''
data = json.loads(data)
headers = {
"content-type": "application/json"
}
response = requests.delete(
url, headers=headers, data=data
)
status_code = response.status_code
if 200 <= status_code < 300:
print('Status Code:', status_code)
print('Response Body:', response.text)
else:
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main()""";
expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
});
test('getCode returns valid code for HEAD request', () {
const requestModel = RequestModel(
url: 'https://jsonplaceholder.typicode.com/posts/1',
method: HTTPVerb.head,
id: '1',
);
const expectedCode = """import requests
import json
def main():
url = 'https://jsonplaceholder.typicode.com/posts/1'
response = requests.head(
url
)
status_code = response.status_code
if 200 <= status_code < 300:
print('Status Code:', status_code)
print('Response Body:', response.text)
else:
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main()""";
expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
});
test(
'getCode returns valid code for requests with headers and query parameters',
() {
const requestModel = RequestModel(
url: 'https://jsonplaceholder.typicode.com/posts',
method: HTTPVerb.get,
requestParams: [
KVRow('userId', 1),
],
requestHeaders: [
KVRow('Custom-Header-1', 'Value-1'),
KVRow('Custom-Header-2', 'Value-2')
],
id: '1',
);
const expectedCode = """import requests
import json
def main():
url = 'https://jsonplaceholder.typicode.com/posts'
params = {
"userId": "1",
}
headers = {
"Custom-Header-1": "Value-1",
"Custom-Header-2": "Value-2"
}
response = requests.get(
url, params=params, headers=headers
)
status_code = response.status_code
if 200 <= status_code < 300:
print('Status Code:', status_code)
print('Response Body:', response.text)
else:
print('Error Status Code:', status_code)
print('Error Response Body:', response.reason)
main()""";
expect(pythonRequestCodeGen.getCode(requestModel, 'https'), expectedCode);
});
});
}