Added relationships module in tests

Author: jedymatt

tests/models.py

@@ -1,7 +1,6 @@
 from sqlalchemy import Column, ForeignKey, Integer, String
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship
-from sqlalchemy.util.langhelpers import generic_repr
 from sqlalchemyseed.util import generate_repr
 
 Base = declarative_base()
@@ -115,14 +114,15 @@ class Single(Base):
     value = Column(String(45))
 
     def __repr__(self) -> str:
-        return generic_repr(self)
+        return generate_repr(self)
 
 
 class One(Base):
     """
     One class with no other Parent that relates to its child.
     """
     __tablename__ = 'single_parent'
+
     id = Column(Integer, primary_key=True)
     value = Column(String(45))
 
@@ -137,7 +137,10 @@ class OneToOne(Base):
     __tablename__ = 'one_to_one'
 
     id = Column(Integer, primary_key=True)
+    value = Column(String(45))
 
+    def __repr__(self) -> str:
+        return generate_repr(self)
 
 class OneToMany(Base):
     """
@@ -148,7 +151,7 @@ class OneToMany(Base):
     id = Column(Integer, primary_key=True)
 
     def __repr__(self) -> str:
-        return generic_repr(self)
+        return generate_repr(self)
 
 
 class ManyToOne(Base):
@@ -169,7 +172,7 @@ class ManyToMany(Base):
     id = Column(Integer, primary_key=True)
 
     def __repr__(self) -> str:
-        return generic_repr(self,)
+        return generate_repr(self,)
 
 
 val = Single(value='str value')

tests/relationships/__init__.py

@@ -0,0 +1 @@
+from . import association_object, many_to_many, one_to_many, one_to_one

tests/relationships/association_object.py

@@ -0,0 +1,28 @@
+from sqlalchemy import Column, ForeignKey, Integer, String
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+
+Base = declarative_base()
+
+
+class Association(Base):
+    __tablename__ = 'association'
+    left_id = Column(ForeignKey('left.id'), primary_key=True)
+    right_id = Column(ForeignKey('right.id'), primary_key=True)
+    extra_data = Column(String(45))
+    child = relationship("Child", back_populates="parents")
+    parent = relationship("Parent", back_populates="children")
+
+
+class Parent(Base):
+    __tablename__ = 'left'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    children = relationship("Association", back_populates="parent")
+
+
+class Child(Base):
+    __tablename__ = 'right'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    parents = relationship("Association", back_populates="child")

tests/relationships/many_to_many.py

@@ -0,0 +1,34 @@
+from sqlalchemy import Column, ForeignKey, Integer, Table
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql.sqltypes import String
+
+Base = declarative_base()
+
+
+association_table = Table(
+    'association',
+    Base.metadata,
+    Column('left_id', ForeignKey('left.id'), primary_key=True),
+    Column('right_id', ForeignKey('right.id'), primary_key=True),
+)
+
+
+class Parent(Base):
+    __tablename__ = 'left'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    children = relationship(
+        "Child",
+        secondary=association_table,
+        back_populates="parents")
+
+
+class Child(Base):
+    __tablename__ = 'right'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    parents = relationship(
+        "Parent",
+        secondary=association_table,
+        back_populates="children")

tests/relationships/many_to_one.py

@@ -0,0 +1,21 @@
+from sqlalchemy import Column, ForeignKey, Integer
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql.sqltypes import String
+
+Base = declarative_base()
+
+
+class Parent(Base):
+    __tablename__ = 'parent'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    child_id = Column(Integer, ForeignKey('child.id'))
+    child = relationship("Child", back_populates="parents")
+
+
+class Child(Base):
+    __tablename__ = 'child'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    parents = relationship("Parent", back_populates="child")

tests/relationships/one_to_many.py

@@ -0,0 +1,20 @@
+from sqlalchemy import Column, ForeignKey, Integer, String
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+
+Base = declarative_base()
+
+
+class Parent(Base):
+    __tablename__ = 'parent'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    children = relationship("Child", back_populates="parent")
+
+
+class Child(Base):
+    __tablename__ = 'child'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    parent_id = Column(Integer, ForeignKey('parent.id'))
+    parent = relationship("Parent", back_populates="children")

tests/relationships/one_to_one.py

@@ -0,0 +1,23 @@
+from sqlalchemy import Column, ForeignKey, Integer
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import relationship
+from sqlalchemy.sql.sqltypes import String
+
+Base = declarative_base()
+
+
+class Parent(Base):
+    __tablename__ = 'parent'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+    
+    child = relationship("Child", back_populates="parent", uselist=False)
+
+
+class Child(Base):
+    __tablename__ = 'child'
+    id = Column(Integer, primary_key=True)
+    value = Column(String(45))
+
+    parent_id = Column(Integer, ForeignKey('parent.id'))
+    parent = relationship("Parent", back_populates="child")

tests/test_seeder.py

@@ -1,3 +1,4 @@
+from typing import List
 import unittest
 
 from sqlalchemy import create_engine
@@ -6,26 +7,74 @@
 from sqlalchemyseed import HybridSeeder, errors
 from sqlalchemyseed import Seeder
 from tests.models import Base, Company
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy import Column, Integer, ForeignKey
+from sqlalchemy.orm import relationship
+
 
 from tests import instances as ins
+from tests import relationships as rel
+
+
+class TestSeederRelationship(unittest.TestCase):
+    """
+    TestSeederRelationship class for testing Seeder class dealing with relationships.
+    """
+
+    def setUp(self) -> None:
+        self.engine = create_engine('sqlite://')
+        Session = sessionmaker(bind=self.engine)
+        session = Session()
+        self.seeder = Seeder(session)
+        self.Base = None
+
+    def tearDown(self) -> None:
+
+        self.Base.metadata.drop_all(self.engine)
+        self.Base = None
+
+    def test_seed_one_to_many(self):
+        """
+        Test seed one to many relationship
+        """
+        # assign classes to remove module
+        Parent = rel.one_to_many.Parent
+        Child = rel.one_to_many.Child
+
+        self.Base = rel.one_to_many.Base
+        self.Base.metadata.create_all(self.engine)
+        json = {
+            'model': 'tests.relationships.one_to_many.Parent',
+            'data': {
+                'value': 'parent',
+                '!children': [
+                    {
+                        'data': {
+                            'value': 'child',
+                        },
+                    },
+                    {
+                        'data': {
+                            'value': 'child',
+                        },
+                    },
+                ],
+            },
+        }
+        self.seeder.seed(json)
+
+        # seeder.instances should only contain the first level entities
+        self.assertEqual(len(self.seeder.instances), 1)
+
+        parent: Parent = self.seeder.instances[0]
+        children: List[Child] = parent.children
+
+        self.assertEqual(parent.value, 'parent')
+        for child in children:
+            self.assertEqual(child.value, 'child')
+            self.assertEqual(child.parent, parent)
 
 
-# class TestSeeder(unittest.TestCase):
-#     """
-#     TestSeeder class for testing Seeder class.
-#     """
-#     def setUp(self) -> None:
-#         self.engine = create_engine('sqlite://')
-#         Session = sessionmaker(bind=self.engine)
-#         session = Session()
-#         Base.metadata.create_all(self.engine)
-#         self.seeder = Seeder(session)
-
-#     def tearDown(self) -> None:
-#         Base.metadata.drop_all(self.engine)
-    
-#     def test_single(self):
-        
 class TestSeeder(unittest.TestCase):
     """
     Test class for Seeder class