fix: tests

src/masoniteorm/collection/Collection.py

@@ -41,7 +41,6 @@ def first(self, callback=None):
             filtered = self.filter(callback)
         response = None
 
-        # print(filtered._items)
         if filtered:
             response = filtered[0]
         return response

src/masoniteorm/query/QueryBuilder.py

@@ -1952,18 +1952,22 @@ def _register_relationships_to_model(
             self
         """
         if related_result and isinstance(hydrated_model, Collection):
-            map_related = related_result.group_by(related.foreign_key)
+            map_related = self._map_related(related_result, related)
             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
 
+    def _map_related(self, related_result, related):
+        if related.__class__.__name__ == 'MorphTo':
+            return related_result
+
+        return related_result.group_by(related.get_foreign_key())
+
     def all(self, selects=[], query=False):
         """Returns all records from the table.
 

src/masoniteorm/relationships/BelongsTo.py

@@ -63,7 +63,6 @@ def get_related(self, query, relation, eagers=(), callback=None):
             ).first()
 
     def register_related(self, key, model, collection):
-        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[0] if related else None})

src/masoniteorm/relationships/HasMany.py

@@ -27,7 +27,6 @@ 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))}
+            {key: collection.get(getattr(model, self.local_key)) or Collection()}
         )

src/masoniteorm/relationships/MorphTo.py

@@ -50,7 +50,7 @@ def __get__(self, instance, owner):
 
     def __getattr__(self, attribute):
         relationship = self.fn(self)()
-        return getattr(relationship.builder, attribute)
+        return getattr(relationship._related_builder, attribute)
 
     def apply_query(self, builder, instance):
         """Apply the query and return a dictionary to be hydrated

tests/sqlite/relationships/test_sqlite_relationships.py

@@ -1,11 +1,8 @@
-import os
 import unittest
-
 from src.masoniteorm.models import Model
 from src.masoniteorm.relationships import belongs_to, has_many, has_one, belongs_to_many
 from tests.integrations.config.database import DB
 
-
 class Profile(Model):
     __table__ = "profiles"
     __connection__ = "dev"