fixed tests and made attempt

src/masoniteorm/collection/Collection.py

@@ -40,6 +40,8 @@ def first(self, callback=None):
         if callback:
             filtered = self.filter(callback)
         response = None
+
+        # print(filtered._items)
         if filtered:
             response = filtered[0]
         return response
@@ -343,7 +345,7 @@ def _serialize(item):
 
     def add_relation(self, result=None):
         for model in self._items:
-            model.add_relations(result or {})
+            model.add_relation(result or {})
 
         return self
 
@@ -532,8 +534,13 @@ def __eq__(self, other):
     def __getitem__(self, item):
         if isinstance(item, slice):
             return self.__class__(self._items[item])
+        if isinstance(item, dict):
+            return self._items.get(item, None)
 
-        return self._items[item]
+        try:
+            return self._items[item]
+        except KeyError:
+            return None
 
     def __setitem__(self, key, value):
         self._items[key] = value

src/masoniteorm/query/QueryBuilder.py

@@ -1951,14 +1951,16 @@ def _register_relationships_to_model(
         Returns:
             self
         """
-        map_related = related_result.group_by(related.foreign_key)
-        if isinstance(hydrated_model, Collection):
+        if related_result and isinstance(hydrated_model, Collection):
+            map_related = related_result.group_by(related.foreign_key)
             for model in hydrated_model:
                 if isinstance(related_result, Collection):
+                    print("aa", related)
                     related.register_related(relation_key, model, map_related)
                 else:
                     model.add_relation({relation_key: related_result or None})
         else:
+            print("heeee", related_result)
             hydrated_model.add_relation({relation_key: related_result or None})
         return self
 

src/masoniteorm/relationships/BelongsTo.py

@@ -63,8 +63,7 @@ def get_related(self, query, relation, eagers=(), callback=None):
             ).first()
 
     def register_related(self, key, model, collection):
-        related = collection.where(
-            self.foreign_key, getattr(model, self.local_key)
-        ).first()
+        print('ooooo', collection._items, "tttt", model.serialize(), model, getattr(model, self.local_key))
+        related = collection.get(getattr(model, self.local_key), None)
 
-        model.add_relation({key: related or None})
+        model.add_relation({key: related[0] if related else None})

src/masoniteorm/relationships/HasMany.py

@@ -27,6 +27,7 @@ def set_keys(self, owner, attribute):
         return self
 
     def register_related(self, key, model, collection):
+        print("zzzzzz", getattr(model, self.local_key), collection._items)
         model.add_relation(
             {key: collection.get(getattr(model, self.local_key))}
         )

tests/User.py

@@ -8,7 +8,7 @@ class User(Model):
 
     __fillable__ = ["name", "email", "password"]
 
-    __connection__ = "mysql"
+    __connection__ = "t"
 
     __auth__ = "email"
 

tests/integrations/config/database.py

@@ -38,7 +38,7 @@
         "log_queries": True,
         "propagate": False,
     },
-    "t": {"driver": "sqlite", "database": "ormtestreg.sqlite3", "log_queries": True},
+    "t": {"driver": "sqlite", "database": "orm.sqlite3", "log_queries": True},
     "devprod": {
         "driver": "mysql",
         "host": os.getenv("MYSQL_DATABASE_HOST"),

tests/sqlite/builder/test_sqlite_query_builder_eager_loading.py

@@ -96,4 +96,5 @@ def test_with_multiple_per_same_relation(self):
         builder = self.get_builder()
         result = User.with_("articles", "articles.logo").where("id", 1).first()
         self.assertTrue(result.serialize()["articles"])
+        print("pppppp", result.serialize()["articles"][0])
         self.assertTrue(result.serialize()["articles"][0]["logo"])