Skip to content

Commit 3534735

Browse files
committed
Update test seeder, added test case
1 parent 5625c57 commit 3534735

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

tests/test_seeder.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,31 @@ def test_seed_no_model(self):
6464
seeder.seed(instance)
6565
self.assertEqual(len(session.new), 3)
6666

67+
def test_seed_multiple_data(self):
68+
instance = {
69+
'model': 'tests.models.Company',
70+
'data': [
71+
{
72+
'name': 'MyCompany',
73+
'!employees': {
74+
'model': 'tests.models.Employee',
75+
'data': {
76+
'name': 'John Smith'
77+
}
78+
79+
}
80+
},
81+
{
82+
'name': 'MySecondCompany'
83+
},
84+
]
85+
}
86+
87+
with self.Session() as session:
88+
seeder = Seeder(session=session)
89+
seeder.seed(instance)
90+
self.assertEqual(len(session.new), 3)
91+
6792
def test_seed_no_relationship(self):
6893
instance = {
6994
'model': 'tests.models.Company',

0 commit comments

Comments
 (0)