From 1b143af680dd108dc1fe0f90505fce2646c483b1 Mon Sep 17 00:00:00 2001
From: Mixel2004 <84668201+Mixel2004@users.noreply.github.com>
Date: Tue, 4 Jul 2023 22:34:46 +0530
Subject: [PATCH] Add test for python (request) codegen

---
 test/codegen/python_request_codegen_test.dart | 188 ++++++++++++++++++
 1 file changed, 188 insertions(+)
 create mode 100644 test/codegen/python_request_codegen_test.dart

diff --git a/test/codegen/python_request_codegen_test.dart b/test/codegen/python_request_codegen_test.dart
new file mode 100644
index 00000000..a4c86096
--- /dev/null
+++ b/test/codegen/python_request_codegen_test.dart
@@ -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);
+    });
+  });
+}