WIP: adding dbt in score

src/main/java/org/schemata/app/SchemaScoreApp.java

@@ -11,8 +11,8 @@
 
 public class SchemaScoreApp implements Callable<Integer> {
 
-  private List<Schema> schemaList;
-  private String schemaName;
+  private final List<Schema> schemaList;
+  private final String schemaName;
 
   public SchemaScoreApp(List<Schema> schemaList, String schemaName) {
     this.schemaList = schemaList;

src/main/java/org/schemata/graph/SchemaGraph.java

@@ -103,6 +103,10 @@ public Set<Schema> getAllEntityVertex() {
     return graph.vertexSet().stream().filter(f -> "ENTITY".equalsIgnoreCase(f.type())).collect(Collectors.toSet());
   }
 
+  public Set<Schema> getAllFactVertex() {
+    return graph.vertexSet().stream().filter(f -> "MODEL".equalsIgnoreCase(f.type()) && "FACT".equalsIgnoreCase(f.modelType()) ).collect(Collectors.toSet());
+  }
+
   public Double getVertexPageRankScore(String vertex) {
     return pageRank.getVertexScore(getSchema(vertex));
   }
@@ -112,6 +116,7 @@ public Double getSchemataScore(String vertex) {
     double score = switch (schema.type().toUpperCase()) {
       case "ENTITY" -> computeEntityScore(vertex);
       case "EVENT" -> computeEventScore(vertex, schema.eventType());
+      case "FACT" -> computeFactScore(vertex);
       default -> 0.0;
     };
     return roundUp(score);
@@ -136,6 +141,11 @@ private double computeEventScore(String vertex, String eventType) {
     return score;
   }
 
+  public double computeFactScore(String vertex) {
+    System.out.println(getAllFactVertex());
+    return 0.0;
+  }
+
   private double computeNonLifecycleScore(String vertex) {
     Set<Schema> referenceVertex =
         outgoingEntityVertexOf(vertex).stream().map(v -> outgoingEntityVertexOf(v.name())).flatMap(Collection::stream)

src/test/java/org/schemata/graph/DerivedSchemaGraphTest.java

@@ -0,0 +1,28 @@
+package org.schemata.graph;
+
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.schemata.ResourceLoader;
+import org.schemata.provider.dbt.DbtSchemaParser;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+
+public class DerivedSchemaGraphTest {
+
+  static SchemaGraph graph;
+
+  @BeforeAll
+  static void setUp() {
+    DbtSchemaParser parser = new DbtSchemaParser();
+    var schemaList = parser.getSchemaList(ResourceLoader.getDbtBasePath());
+    graph = new SchemaGraph(schemaList);
+  }
+
+  @Test
+  public void testGetAllFaceVertex() {
+    assertEquals(1, graph.getAllFactVertex().size());
+    System.out.println(graph.getAllFactVertex());
+  }
+
+}

src/test/java/org/schemata/graph/SchemaGraphTest.java → src/test/java/org/schemata/graph/EventSchemaGraphTest.java

@@ -22,7 +22,7 @@
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 
-public class SchemaGraphTest {
+public class EventSchemaGraphTest {
 
   static SchemaGraph graph;
 

src/test/java/org/schemata/provider/dbt/DbtSchemaParserTest.java

@@ -3,12 +3,16 @@
 import org.junit.jupiter.api.Test;
 import org.schemata.ResourceLoader;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
 
 public class DbtSchemaParserTest {
 
   @Test
-  public void testMe() {
+  public void testSchemataList() {
     DbtSchemaParser parser = new DbtSchemaParser();
-    parser.getSchemaList(ResourceLoader.getDbtBasePath());
+    var schemaList = parser.getSchemaList(ResourceLoader.getDbtBasePath());
+    var expectedSize = 7;
+    assertEquals(expectedSize, schemaList.size());
   }
 }

src/test/resources/dbt/manifest.json

@@ -241,6 +241,8 @@
         "tags": [],
         "meta": {},
         "materialized": "incremental",
+        "domain": "core",
+        "model_type": "fact",
         "persist_docs": {},
         "quoting": {},
         "column_types": {},