mirror of
https://github.com/foss42/apidash.git
synced 2025-07-30 14:55:56 +08:00
Update python requests codes
This commit is contained in:
Ashita Prasad
@ -1,10 +1,9 @@
|
||||
import 'package:apidash/codegen/kotlin/pkg_okhttp.dart';
|
||||
import 'python/pkg_http_client.dart';
|
||||
import 'package:apidash/codegen/python/pkg_request.dart';
|
||||
import 'package:apidash/consts.dart';
|
||||
|
||||
import 'package:apidash/models/models.dart' show RequestModel;
|
||||
import 'package:apidash/consts.dart';
|
||||
import 'dart/pkg_http.dart';
|
||||
import 'kotlin/pkg_okhttp.dart';
|
||||
import 'python/pkg_http_client.dart';
|
||||
import 'python/pkg_requests.dart';
|
||||
|
||||
class Codegen {
|
||||
String? getCode(
|
||||
@ -20,7 +19,7 @@ class Codegen {
|
||||
case CodegenLanguage.pythonHttpClient:
|
||||
return PythonHttpClient().getCode(requestModel, defaultUriScheme);
|
||||
case CodegenLanguage.pythonRequests:
|
||||
return PythonRequestCodeGen().getCode(requestModel, defaultUriScheme);
|
||||
return PythonRequestsCodeGen().getCode(requestModel, defaultUriScheme);
|
||||
default:
|
||||
throw ArgumentError('Invalid codegenLanguage');
|
||||
}
|
||||
|
||||
@ -6,7 +6,7 @@ import 'package:apidash/utils/utils.dart'
|
||||
show getValidRequestUri, padMultilineString, rowsToMap;
|
||||
import 'package:apidash/models/models.dart' show RequestModel;
|
||||
|
||||
class PythonRequestCodeGen {
|
||||
class PythonRequestsCodeGen {
|
||||
final String kTemplateStart = """import requests
|
||||
|
||||
url = '{{url}}'
|
||||
@ -1,179 +0,0 @@
|
||||
import 'package:apidash/codegen/python/pkg_request.dart';
|
||||
import 'package:apidash/models/models.dart';
|
||||
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
|
||||
|
||||
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
|
||||
|
||||
def main():
|
||||
url = 'https://jsonplaceholder.typicode.com/posts'
|
||||
|
||||
data = '''{"title": "foo","body": "bar","userId": 1}'''
|
||||
|
||||
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
|
||||
|
||||
def main():
|
||||
url = 'https://jsonplaceholder.typicode.com/posts/1'
|
||||
|
||||
data = '''{"title": "foo","body": "bar","userId": 1}'''
|
||||
|
||||
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
|
||||
|
||||
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: [
|
||||
NameValueModel(name: 'userId', value: 1),
|
||||
],
|
||||
requestHeaders: [
|
||||
NameValueModel(name: 'Custom-Header-1', value: 'Value-1'),
|
||||
NameValueModel(name: 'Custom-Header-2', value: 'Value-2')
|
||||
],
|
||||
id: '1',
|
||||
);
|
||||
const expectedCode = """import requests
|
||||
|
||||
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);
|
||||
});
|
||||
});
|
||||
}
|
||||
327
test/codegen/python_requests_codegen_test.dart
Normal file
327
test/codegen/python_requests_codegen_test.dart
Normal file
@ -0,0 +1,327 @@
|
||||
import 'package:apidash/codegen/python/pkg_requests.dart';
|
||||
import '../request_models.dart';
|
||||
import 'package:test/test.dart';
|
||||
|
||||
void main() {
|
||||
final pythonRequestsClient = PythonRequestsCodeGen();
|
||||
|
||||
group('GET Request', () {
|
||||
test('GET 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com'
|
||||
|
||||
response = requests.get(url)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 2', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/country/data'
|
||||
|
||||
params = {
|
||||
"code": "US"
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 3', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/country/data'
|
||||
|
||||
params = {
|
||||
"code": "IND"
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet3, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 4', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/humanize/social'
|
||||
|
||||
params = {
|
||||
"num": "8700000",
|
||||
"digits": "3",
|
||||
"system": "SS",
|
||||
"add_space": "true",
|
||||
"trailing_zeros": "true"
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet4, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 5', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.github.com/repos/foss42/apidash'
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Test Agent"
|
||||
}
|
||||
|
||||
response = requests.get(url, headers=headers)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet5, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 6', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.github.com/repos/foss42/apidash'
|
||||
|
||||
params = {
|
||||
"raw": "true"
|
||||
}
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Test Agent"
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params, headers=headers)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet6, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 7', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com'
|
||||
|
||||
response = requests.get(url)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet7, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('GET 8', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.github.com/repos/foss42/apidash'
|
||||
|
||||
params = {
|
||||
"raw": "true"
|
||||
}
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Test Agent"
|
||||
}
|
||||
|
||||
response = requests.get(url, params=params, headers=headers)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelGet8, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('HEAD Request', () {
|
||||
test('HEAD 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com'
|
||||
|
||||
response = requests.head(url)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelHead1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('HEAD 2', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'http://api.foss42.com'
|
||||
|
||||
response = requests.head(url)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelHead2, "http"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('POST Request', () {
|
||||
test('POST 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/case/lower'
|
||||
|
||||
payload = r'''{
|
||||
"text": "I LOVE Flutter"
|
||||
}'''
|
||||
|
||||
headers = {
|
||||
"content-type": "text/plain"
|
||||
}
|
||||
|
||||
response = requests.post(url, data=payload, headers=headers)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelPost1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('POST 2', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/case/lower'
|
||||
|
||||
payload = {
|
||||
"text": "I LOVE Flutter"
|
||||
}
|
||||
|
||||
response = requests.post(url, json=payload)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelPost2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('POST 3', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://api.foss42.com/case/lower'
|
||||
|
||||
payload = {
|
||||
"text": "I LOVE Flutter"
|
||||
}
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Test Agent"
|
||||
}
|
||||
|
||||
response = requests.post(url, json=payload, headers=headers)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelPost3, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
group('PUT Request', () {
|
||||
test('PUT 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://reqres.in/api/users/2'
|
||||
|
||||
payload = {
|
||||
"name": "morpheus",
|
||||
"job": "zion resident"
|
||||
}
|
||||
|
||||
response = requests.put(url, json=payload)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelPut1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('PATCH Request', () {
|
||||
test('PATCH 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://reqres.in/api/users/2'
|
||||
|
||||
payload = {
|
||||
"name": "marfeus",
|
||||
"job": "accountant"
|
||||
}
|
||||
|
||||
response = requests.patch(url, json=payload)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelPatch1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
|
||||
group('DELETE Request', () {
|
||||
test('DELETE 1', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://reqres.in/api/users/2'
|
||||
|
||||
response = requests.delete(url)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelDelete1, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
|
||||
test('DELETE 2', () {
|
||||
const expectedCode = r"""import requests
|
||||
|
||||
url = 'https://reqres.in/api/users/2'
|
||||
|
||||
payload = {
|
||||
"name": "marfeus",
|
||||
"job": "accountant"
|
||||
}
|
||||
|
||||
response = requests.delete(url, json=payload)
|
||||
|
||||
print('Status Code:', response.status_code)
|
||||
print('Response Body:', response.text)
|
||||
""";
|
||||
expect(pythonRequestsClient.getCode(requestModelDelete2, "https"),
|
||||
expectedCode);
|
||||
});
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user