find interface mappings

gemini-hlsw · Oct 31, 2023 · e782488 · e782488
1 parent 4aafc1a
commit e782488
Showing 1 changed file with 29 additions and 14 deletions.
diff --git a/modules/service/src/test/scala/lucuma/odb/graphql/Checklist.scala b/modules/service/src/test/scala/lucuma/odb/graphql/Checklist.scala
@@ -5,9 +5,10 @@ package lucuma.odb.graphql
 
 import cats.effect.IO
 import cats.syntax.all._
+import grackle.Field
 import grackle.ObjectType
 
-@munit.IgnoreSuite // comment this out if you want to run this. there's probably a better way to do this
+//@munit.IgnoreSuite // comment this out if you want to run this. there's probably a better way to do this
 class Checklist extends OdbSuite {
   val validUsers = Nil
 
@@ -56,26 +57,40 @@ class Checklist extends OdbSuite {
   def printObjectType(m: BaseMapping[IO], t: ObjectType): IO[Unit] =
     if JsonMappedTypes.contains(t.name) then
       IO.println(s"- [x] ${t.name} (via Json blob)")
-    else
+    else {
+
+      def definedInInterface(f: Field): Option[String] =
+        t.interfaces.find { nt =>
+          m.typeMapping(nt).exists {
+            case om: m.ObjectMapping =>
+               om.fieldMapping(f.name).isDefined
+            case sm: m.SwitchMapping =>
+               sm.lookup.exists { case (p, om) =>
+                 om.fieldMapping(f.name).isDefined
+               }
+            case _                   =>
+               false
+          }
+        }.map(_.name)
+
+      def printField(om: m.ObjectMapping, f: Field): IO[Unit] =
+        om.fieldMapping(f.name) match {
+          case None    => IO.println(s"  - ${definedInInterface(f).fold(s"[ ] ${f.name}")(n => s"[x] ${f.name} (via $n)")}")
+          case Some(_) => IO.println(s"  - [x] ${f.name}")
+        }
+
       m.typeMapping(t) match
-        case None    => IO.println(s"- [ ] ${t.name}")
+        case None                      =>
+          IO.println(s"- [ ] ${t.name}")
         case Some(om: m.ObjectMapping) =>
-          IO.println(s"- [x] ${t.name}") >>
-          t.fields.traverse_ { f =>
-            om.fieldMapping(f.name) match
-              case None => IO.println(s"  - [ ] ${f.name}")
-              case Some(_) => IO.println(s"  - [x] ${f.name}")
-          }
+          IO.println(s"- [x] ${t.name}") >> t.fields.traverse_ { f => printField(om, f) }
         case Some(sm: m.SwitchMapping) =>
           sm.lookup.traverse_ { (p, om) =>
             IO.println(s"- [x] ${t.name} (at ${p.path.mkString(s"${p.rootTpe}/", "/", "")})") >>
-            t.fields.traverse_ { f =>
-              om.fieldMapping(f.name) match
-                case None => IO.println(s"  - [ ] ${f.name}")
-                case Some(_) => IO.println(s"  - [x] ${f.name}")
-            }
+            t.fields.traverse_ { f => printField(om, f) }
           }
         case m => fail(s"Can't handle object mapping $m")
+  }
 
   def printChecklist(m: BaseMapping[IO]): IO[Unit] = {
     IO.println("\n\n# Schema Coverage") >>