Fix barricade processor, so it should create mock responses for all p…

…rovided endpoints
mutualmobile · Jun 22, 2022 · 7b56659 · 7b56659
1 parent 978da26
commit 7b56659
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 3 deletions.
diff --git a/.idea/.name b/.idea/.name
diff --git a/app/src/main/assets/barricade/categories/success.json b/app/src/main/assets/barricade/categories/success.json
@@ -0,0 +1,5 @@
+[
+  "Test",
+  "Test2",
+  "Test3"
+]
diff --git a/app/src/main/java/com/mutualmobile/barricadeSample/data/ApiService.kt b/app/src/main/java/com/mutualmobile/barricadeSample/data/ApiService.kt
@@ -8,6 +8,7 @@ import retrofit2.http.GET
 interface ApiService {
     @GET("/jokes/random")
     @Barricade(
+        endpoint = "random",
         responses = [
             com.mutualmobile.barricade.annotation.Response(
                 fileName = "success.json",
@@ -22,5 +23,14 @@ interface ApiService {
     suspend fun getRandomJoke(): Response<JokeResponse>
 
     @GET("/jokes/categories")
+    @Barricade(
+        endpoint = "categories",
+        responses = [
+            com.mutualmobile.barricade.annotation.Response(
+                fileName = "success.json",
+                isDefault = true
+            )
+        ]
+    )
     suspend fun getJokeCategories(): Response<List<String>>
 }
diff --git a/barricade-compiler/src/main/java/com/mutualmobile/barricade/compiler/BarricadeProcessor.kt b/barricade-compiler/src/main/java/com/mutualmobile/barricade/compiler/BarricadeProcessor.kt
@@ -64,11 +64,9 @@ class BarricadeProcessor(
 
         file writeToFile "\tprivate val configs = hashMapOf<String, $PACKAGE_NAME.response.BarricadeResponseSet>()\n\n"
         file writeToFile "\tinit {\n"
-        file writeToFile "\t\tval barricadeResponsesForRandom = mutableListOf<com.mutualmobile.barricade.response.BarricadeResponse>()\n"
         symbols.forEach { symbol ->
             symbol.accept(BarricadeVisitor(file, logger), Unit)
         }
-        file writeToFile "\t\tconfigs.put(\"random\", com.mutualmobile.barricade.response.BarricadeResponseSet(barricadeResponsesForRandom, 0))\n"
         file writeToFile "\t}\n\n"
         file writeToFile "\toverride fun getConfigs(): HashMap<String, $PACKAGE_NAME.response.BarricadeResponseSet> = configs\n"
         file writeToFile "\toverride fun getResponseForEndpoint(endpoint: String): $PACKAGE_NAME.response.BarricadeResponse? {\n"

diff --git a/...e-compiler/src/main/java/com/mutualmobile/barricade/compiler/visitors/BarricadeVisitor.kt b/...e-compiler/src/main/java/com/mutualmobile/barricade/compiler/visitors/BarricadeVisitor.kt
@@ -34,13 +34,17 @@ class BarricadeVisitor(
             (it as KSAnnotation).arguments
         }
 
+        file writeToFile "\t\tval barricadeResponsesFor$endpoint = mutableListOf<com.mutualmobile.barricade.response.BarricadeResponse>()\n"
+
         responses.forEach { response ->
             val fileName = response.first { it.name?.asString() == Response::fileName.name }.value as String
             val isDefault = response.first { it.name?.asString() == Response::isDefault.name }.value as Boolean
             val statusCode = response.first { it.name?.asString() == Response::statusCode.name }.value as Int
             val type = response.first { it.name?.asString() == Response::type.name }.value as String
-            file writeToFile "\t\tbarricadeResponsesForRandom.add(${BarricadeProcessor.PACKAGE_NAME}.response.BarricadeResponse($statusCode, \"$fileName\", \"$type\"))\n"
+            file writeToFile "\t\tbarricadeResponsesFor$endpoint.add(${BarricadeProcessor.PACKAGE_NAME}.response.BarricadeResponse($statusCode, \"$fileName\", \"$type\"))\n"
             logger.info("Found args: $fileName, $isDefault, $statusCode, $type")
         }
+        file writeToFile "\t\tconfigs.put(\"$endpoint\", com.mutualmobile.barricade.response.BarricadeResponseSet(barricadeResponsesFor$endpoint, 0))\n"
+
     }
 }