feat: kotlin codegen

2025-08-06 13:51:20 +08:00 · 2023-10-01 20:15:21 +05:30
parent dda6e22dbf
commit d89e259b02
3 changed files with 597 additions and 145 deletions
--- a/test/codegen/kotlin_okhttp_codegen_test.dart
+++ b/test/codegen/kotlin_okhttp_codegen_test.dart
@ -3,124 +3,498 @@ import 'package:test/test.dart';
 import '../request_models.dart';

 void main() {
-  group('KotlinOkHttpCodeGen', () {
-    final kotlinOkHttpCodeGen = KotlinOkHttpCodeGen();
+  final kotlinOkHttpCodeGen = KotlinOkHttpCodeGen();

-    test('getCode returns valid code for GET request', () {
-      const expectedCode = r"""import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.MultipartBody
-import okhttp3.OkHttpClient
+  group('GET Request', () {
+    test('GET 1', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
 import okhttp3.Request
-import okhttp3.RequestBody.Companion.toRequestBody
-import okhttp3.RequestBody.Companion.asRequestBody
-import java.io.File
-import java.util.concurrent.TimeUnit

-val client = OkHttpClient()
-val request = Request.Builder()
-  .url("https://api.foss42.com")
-  .build()
-val response = client.newCall(request).execute()
+fun main() {
+    val client = OkHttpClient()

-println(response.body!!.string())
+    val url = "https://api.foss42.com"
+
+    val request = Request.Builder()
+        .url(url)
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
 """;
-      expect(kotlinOkHttpCodeGen.getCode(requestModelGet1), expectedCode);
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet1, "https"), expectedCode);
    });

-    test('getCode returns valid code for POST request', () {
-      const expectedCode = r"""import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.MultipartBody
-import okhttp3.OkHttpClient
+    test('GET 2', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
 import okhttp3.Request
-import okhttp3.RequestBody.Companion.toRequestBody
-import okhttp3.RequestBody.Companion.asRequestBody
-import java.io.File
-import java.util.concurrent.TimeUnit
+import okhttp3.HttpUrl.Companion.toHttpUrl

-val client = OkHttpClient()
-val mediaType = "application/json".toMediaType()
-val body = "{"text": "IS Upper"}".toRequestBody(mediaType)
-val request = Request.Builder()
-  .url("https://api.foss42.com/case/lower")
-  .post(body)
-  .build()
-val response = client.newCall(request).execute()
+fun main() {
+    val client = OkHttpClient()

-println(response.body!!.string())
+    val url = "https://api.foss42.com/country/data".toHttpUrl().newBuilder()
+        .addQueryParameter("code", "US")
+        .build()
+
+    val request = Request.Builder()
+        .url(url)
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
 """;
-      expect(kotlinOkHttpCodeGen.getCode(requestModelPost1), expectedCode);
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet2, "https"), expectedCode);
    });

-    test('getCode returns valid code for DELETE request', () {
-      const expectedCode = r"""import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.MultipartBody
-import okhttp3.OkHttpClient
+    test('GET 3', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
 import okhttp3.Request
-import okhttp3.RequestBody.Companion.toRequestBody
-import okhttp3.RequestBody.Companion.asRequestBody
-import java.io.File
-import java.util.concurrent.TimeUnit
+import okhttp3.HttpUrl.Companion.toHttpUrl

-val client = OkHttpClient()
-val mediaType = "application/json".toMediaType()
-val body = "{"title": "foo","body": "bar","userId": 1}".toRequestBody(mediaType)
-val request = Request.Builder()
-  .url("https://jsonplaceholder.typicode.com/posts/1")
-  .method("DELETE", body)
-  .build()
-val response = client.newCall(request).execute()
+fun main() {
+    val client = OkHttpClient()

-println(response.body!!.string())
+    val url = "https://api.foss42.com/country/data".toHttpUrl().newBuilder()
+        .addQueryParameter("code", "IND")
+        .build()
+
+    val request = Request.Builder()
+        .url(url)
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
 """;
-      expect(kotlinOkHttpCodeGen.getCode(requestModelDelete1), expectedCode);
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet3, "https"), expectedCode);
    });

-    test('getCode returns valid code for HEAD request', () {
-      const expectedCode = """import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.MultipartBody
-import okhttp3.OkHttpClient
+    test('GET 4', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
 import okhttp3.Request
-import okhttp3.RequestBody.Companion.toRequestBody
-import okhttp3.RequestBody.Companion.asRequestBody
-import java.io.File
-import java.util.concurrent.TimeUnit
+import okhttp3.HttpUrl.Companion.toHttpUrl

-val client = OkHttpClient()
-val request = Request.Builder()
-  .url("https://jsonplaceholder.typicode.com/posts/1")
-  .head()
-  .build()
-val response = client.newCall(request).execute()
+fun main() {
+    val client = OkHttpClient()

-println(response.body!!.string())
+    val url = "https://api.foss42.com/humanize/social".toHttpUrl().newBuilder()
+        .addQueryParameter("num", "8700000")
+        .addQueryParameter("digits", "3")
+        .addQueryParameter("system", "SS")
+        .addQueryParameter("add_space", "true")
+        .addQueryParameter("trailing_zeros", "true")
+        .build()
+
+    val request = Request.Builder()
+        .url(url)
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
 """;
-      expect(kotlinOkHttpCodeGen.getCode(requestModelHead1), expectedCode);
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet4, "https"), expectedCode);
    });

-    test(
-        'getCode returns valid code for requests with headers and query parameters',
-        () {
-      const expectedCode = """import okhttp3.MediaType.Companion.toMediaType
-import okhttp3.MultipartBody
-import okhttp3.OkHttpClient
+    test('GET 5', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.github.com/repos/foss42/apidash"
+
+    val request = Request.Builder()
+        .url(url)
+        .addHeader("User-Agent", "Test Agent")
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet5, "https"), expectedCode);
+    });
+
+    test('GET 6', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.HttpUrl.Companion.toHttpUrl
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.github.com/repos/foss42/apidash".toHttpUrl().newBuilder()
+        .addQueryParameter("raw", "true")
+        .build()
+
+    val request = Request.Builder()
+        .url(url)
+        .addHeader("User-Agent", "Test Agent")
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet6, "https"), expectedCode);
+    });
+
+    test('GET 7', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.foss42.com"
+
+    val request = Request.Builder()
+        .url(url)
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet7, "https"), expectedCode);
+    });
+
+    test('GET 8', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.HttpUrl.Companion.toHttpUrl
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.github.com/repos/foss42/apidash".toHttpUrl().newBuilder()
+        .addQueryParameter("raw", "true")
+        .build()
+
+    val request = Request.Builder()
+        .url(url)
+        .addHeader("User-Agent", "Test Agent")
+        .get()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelGet8, "https"), expectedCode);
+    });
+  });
+
+  group('HEAD Request', () {
+    test('HEAD 1', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.foss42.com"
+
+    val request = Request.Builder()
+        .url(url)
+        .head()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(kotlinOkHttpCodeGen.getCode(requestModelHead1, "https"),
+          expectedCode);
+    });
+
+    test('HEAD 2', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "http://api.foss42.com"
+
+    val request = Request.Builder()
+        .url(url)
+        .head()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+""";
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelHead2, "http"), expectedCode);
+    });
+  });
+
+  group('POST Request', () {
+    test('POST 1', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
 import okhttp3.Request
 import okhttp3.RequestBody.Companion.toRequestBody
-import okhttp3.RequestBody.Companion.asRequestBody
-import java.io.File
-import java.util.concurrent.TimeUnit
+import okhttp3.MediaType.Companion.toMediaType

-val client = OkHttpClient()
-val request = Request.Builder()
-  .url("https://jsonplaceholder.typicode.com/posts")
-  .addQueryParameter("userId", "1")
-  .addHeader("Custom-Header-1", "Value-1")
-  .addHeader("Custom-Header-2", "Value-2")
-  .build()
-val response = client.newCall(request).execute()
+fun main() {
+    val client = OkHttpClient()

-println(response.body!!.string())
+    val url = "https://api.foss42.com/case/lower"
+
+    val mediaType = "text/plain".toMediaType()
+
+    val body = """{
+"text": "I LOVE Flutter"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .post(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(kotlinOkHttpCodeGen.getCode(requestModelPost1, "https"),
+          expectedCode);
+    });
+
+    test('POST 2', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody.Companion.toRequestBody
+import okhttp3.MediaType.Companion.toMediaType
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.foss42.com/case/lower"
+
+    val mediaType = "application/json".toMediaType()
+
+    val body = """{
+"text": "I LOVE Flutter"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .post(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(kotlinOkHttpCodeGen.getCode(requestModelPost2, "https"),
+          expectedCode);
+    });
+
+    test('POST 3', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody.Companion.toRequestBody
+import okhttp3.MediaType.Companion.toMediaType
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://api.foss42.com/case/lower"
+
+    val mediaType = "application/json".toMediaType()
+
+    val body = """{
+"text": "I LOVE Flutter"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .addHeader("User-Agent", "Test Agent")
+        .post(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(kotlinOkHttpCodeGen.getCode(requestModelPost3, "https"),
+          expectedCode);
+    });
+  });
+  group('PUT Request', () {
+    test('PUT 1', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody.Companion.toRequestBody
+import okhttp3.MediaType.Companion.toMediaType
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://reqres.in/api/users/2"
+
+    val mediaType = "application/json".toMediaType()
+
+    val body = """{
+"name": "morpheus",
+"job": "zion resident"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .put(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(
+          kotlinOkHttpCodeGen.getCode(requestModelPut1, "https"), expectedCode);
+    });
+  });
+
+  group('PATCH Request', () {
+    test('PATCH 1', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody.Companion.toRequestBody
+import okhttp3.MediaType.Companion.toMediaType
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://reqres.in/api/users/2"
+
+    val mediaType = "application/json".toMediaType()
+
+    val body = """{
+"name": "marfeus",
+"job": "accountant"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .patch(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(kotlinOkHttpCodeGen.getCode(requestModelPatch1, "https"),
+          expectedCode);
+    });
+  });
+
+  group('DELETE Request', () {
+    test('DELETE 1', () {
+      const expectedCode = r"""import okhttp3.OkHttpClient
+import okhttp3.Request
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://reqres.in/api/users/2"
+
+    val request = Request.Builder()
+        .url(url)
+        .delete()
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
 """;
-      expect(kotlinOkHttpCodeGen.getCode(requestModelGet2), expectedCode);
+      expect(kotlinOkHttpCodeGen.getCode(requestModelDelete1, "https"),
+          expectedCode);
+    });
+
+    test('DELETE 2', () {
+      const expectedCode = r'''import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody.Companion.toRequestBody
+import okhttp3.MediaType.Companion.toMediaType
+
+fun main() {
+    val client = OkHttpClient()
+
+    val url = "https://reqres.in/api/users/2"
+
+    val mediaType = "application/json".toMediaType()
+
+    val body = """{
+"name": "marfeus",
+"job": "accountant"
+}""".toRequestBody(mediaType)
+
+    val request = Request.Builder()
+        .url(url)
+        .delete(body)
+        .build()
+
+    val response = client.newCall(request).execute()
+
+    println(response.code)
+    println(response.body?.string())
+}
+''';
+      expect(kotlinOkHttpCodeGen.getCode(requestModelDelete2, "https"),
+          expectedCode);
    });
  });
 }