- initial tests for bulk

2 files changed, 319 insertions, 1 deletions

diff --git a/lib/sqlalchemy/orm/session.py b/lib/sqlalchemy/orm/session.py
index ef911824c..7dd577230 100644
--- a/lib/sqlalchemy/orm/session.py
+++ b/lib/sqlalchemy/orm/session.py
@@ -2056,7 +2056,8 @@ class Session(_SessionClassMethods):
                 mapper, states, isupdate, True, return_defaults)
 
     def bulk_insert_mappings(self, mapper, mappings, return_defaults=False):
-        self._bulk_save_mappings(mapper, mappings, False, False, return_defaults)
+        self._bulk_save_mappings(
+            mapper, mappings, False, False, return_defaults)
 
     def bulk_update_mappings(self, mapper, mappings):
         self._bulk_save_mappings(mapper, mappings, True, False, False)
diff --git a/test/orm/test_bulk.py b/test/orm/test_bulk.py
new file mode 100644
index 000000000..4bcde2480
--- /dev/null
+++ b/test/orm/test_bulk.py
@@ -0,0 +1,317 @@
+from sqlalchemy import testing
+from sqlalchemy.testing import eq_
+from sqlalchemy.testing.schema import Table, Column
+from sqlalchemy.testing import fixtures
+from sqlalchemy import Integer, String, ForeignKey
+from sqlalchemy.orm import mapper, Session
+from sqlalchemy.testing.assertsql import CompiledSQL
+from test.orm import _fixtures
+
+
+class BulkTest(testing.AssertsExecutionResults):
+    run_inserts = None
+
+
+class BulkInsertTest(BulkTest, _fixtures.FixtureTest):
+
+    @classmethod
+    def setup_mappers(cls):
+        User, Address = cls.classes("User", "Address")
+        u, a = cls.tables("users", "addresses")
+
+        mapper(User, u)
+        mapper(Address, a)
+
+    def test_bulk_save_return_defaults(self):
+        User, = self.classes("User",)
+
+        s = Session()
+        objects = [
+            User(name="u1"),
+            User(name="u2"),
+            User(name="u3")
+        ]
+        assert 'id' not in objects[0].__dict__
+
+        with self.sql_execution_asserter() as asserter:
+            s.bulk_save_objects(objects, return_defaults=True)
+
+        asserter.assert_(
+            CompiledSQL(
+                "INSERT INTO users (name) VALUES (:name)",
+                [{'name': 'u1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO users (name) VALUES (:name)",
+                [{'name': 'u2'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO users (name) VALUES (:name)",
+                [{'name': 'u3'}]
+            ),
+        )
+        eq_(objects[0].__dict__['id'], 1)
+
+    def test_bulk_save_no_defaults(self):
+        User, = self.classes("User",)
+
+        s = Session()
+        objects = [
+            User(name="u1"),
+            User(name="u2"),
+            User(name="u3")
+        ]
+        assert 'id' not in objects[0].__dict__
+
+        with self.sql_execution_asserter() as asserter:
+            s.bulk_save_objects(objects)
+
+        asserter.assert_(
+            CompiledSQL(
+                "INSERT INTO users (name) VALUES (:name)",
+                [{'name': 'u1'}, {'name': 'u2'}, {'name': 'u3'}]
+            ),
+        )
+        assert 'id' not in objects[0].__dict__
+
+
+class BulkInheritanceTest(fixtures.MappedTest, BulkTest):
+    @classmethod
+    def define_tables(cls, metadata):
+        Table(
+            'people', metadata,
+            Column(
+                'person_id', Integer,
+                primary_key=True,
+                test_needs_autoincrement=True),
+            Column('name', String(50)),
+            Column('type', String(30)))
+
+        Table(
+            'engineers', metadata,
+            Column(
+                'person_id', Integer,
+                ForeignKey('people.person_id'),
+                primary_key=True),
+            Column('status', String(30)),
+            Column('primary_language', String(50)))
+
+        Table(
+            'managers', metadata,
+            Column(
+                'person_id', Integer,
+                ForeignKey('people.person_id'),
+                primary_key=True),
+            Column('status', String(30)),
+            Column('manager_name', String(50)))
+
+        Table(
+            'boss', metadata,
+            Column(
+                'boss_id', Integer,
+                ForeignKey('managers.person_id'),
+                primary_key=True),
+            Column('golf_swing', String(30)))
+
+    @classmethod
+    def setup_classes(cls):
+        class Base(cls.Comparable):
+            pass
+
+        class Person(Base):
+            pass
+
+        class Engineer(Person):
+            pass
+
+        class Manager(Person):
+            pass
+
+        class Boss(Manager):
+            pass
+
+    @classmethod
+    def setup_mappers(cls):
+        Person, Engineer, Manager, Boss = \
+            cls.classes('Person', 'Engineer', 'Manager', 'Boss')
+        p, e, m, b = cls.tables('people', 'engineers', 'managers', 'boss')
+
+        mapper(
+            Person, p, polymorphic_on=p.c.type,
+            polymorphic_identity='person')
+        mapper(Engineer, e, inherits=Person, polymorphic_identity='engineer')
+        mapper(Manager, m, inherits=Person, polymorphic_identity='manager')
+        mapper(Boss, b, inherits=Manager, polymorphic_identity='boss')
+
+    def test_bulk_save_joined_inh_return_defaults(self):
+        Person, Engineer, Manager, Boss = \
+            self.classes('Person', 'Engineer', 'Manager', 'Boss')
+
+        s = Session()
+        objects = [
+            Manager(name='m1', status='s1', manager_name='mn1'),
+            Engineer(name='e1', status='s2', primary_language='l1'),
+            Engineer(name='e2', status='s3', primary_language='l2'),
+            Boss(
+                name='b1', status='s3', manager_name='mn2',
+                golf_swing='g1')
+        ]
+        assert 'person_id' not in objects[0].__dict__
+
+        with self.sql_execution_asserter() as asserter:
+            s.bulk_save_objects(objects, return_defaults=True)
+
+        asserter.assert_(
+            CompiledSQL(
+                "INSERT INTO people (name, type) VALUES (:name, :type)",
+                [{'type': 'manager', 'name': 'm1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO managers (person_id, status, manager_name) "
+                "VALUES (:person_id, :status, :manager_name)",
+                [{'person_id': 1, 'status': 's1', 'manager_name': 'mn1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO people (name, type) VALUES (:name, :type)",
+                [{'type': 'engineer', 'name': 'e1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO people (name, type) VALUES (:name, :type)",
+                [{'type': 'engineer', 'name': 'e2'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO engineers (person_id, status, primary_language) "
+                "VALUES (:person_id, :status, :primary_language)",
+                [{'person_id': 2, 'status': 's2', 'primary_language': 'l1'},
+                 {'person_id': 3, 'status': 's3', 'primary_language': 'l2'}]
+
+            ),
+            CompiledSQL(
+                "INSERT INTO people (name, type) VALUES (:name, :type)",
+                [{'type': 'boss', 'name': 'b1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO managers (person_id, status, manager_name) "
+                "VALUES (:person_id, :status, :manager_name)",
+                [{'person_id': 4, 'status': 's3', 'manager_name': 'mn2'}]
+
+            ),
+            CompiledSQL(
+                "INSERT INTO boss (golf_swing) VALUES (:golf_swing)",
+                [{'golf_swing': 'g1'}]
+            )
+        )
+        eq_(objects[0].__dict__['person_id'], 1)
+
+    def test_bulk_save_joined_inh_no_defaults(self):
+        Person, Engineer, Manager, Boss = \
+            self.classes('Person', 'Engineer', 'Manager', 'Boss')
+
+        s = Session()
+        with self.sql_execution_asserter() as asserter:
+            s.bulk_save_objects([
+                Manager(
+                    person_id=1,
+                    name='m1', status='s1', manager_name='mn1'),
+                Engineer(
+                    person_id=2,
+                    name='e1', status='s2', primary_language='l1'),
+                Engineer(
+                    person_id=3,
+                    name='e2', status='s3', primary_language='l2'),
+                Boss(
+                    person_id=4,
+                    name='b1', status='s3', manager_name='mn2',
+                    golf_swing='g1')
+            ],
+
+            )
+
+        # the only difference here is that common classes are grouped together.
+        # at the moment it doesn't lump all the "people" tables from
+        # different classes together.
+        asserter.assert_(
+            CompiledSQL(
+                "INSERT INTO people (person_id, name, type) VALUES "
+                "(:person_id, :name, :type)",
+                [{'person_id': 1, 'type': 'manager', 'name': 'm1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO managers (person_id, status, manager_name) "
+                "VALUES (:person_id, :status, :manager_name)",
+                [{'status': 's1', 'person_id': 1, 'manager_name': 'mn1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO people (person_id, name, type) VALUES "
+                "(:person_id, :name, :type)",
+                [{'person_id': 2, 'type': 'engineer', 'name': 'e1'},
+                 {'person_id': 3, 'type': 'engineer', 'name': 'e2'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO engineers (person_id, status, primary_language) "
+                "VALUES (:person_id, :status, :primary_language)",
+                [{'person_id': 2, 'status': 's2', 'primary_language': 'l1'},
+                 {'person_id': 3, 'status': 's3', 'primary_language': 'l2'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO people (person_id, name, type) VALUES "
+                "(:person_id, :name, :type)",
+                [{'person_id': 4, 'type': 'boss', 'name': 'b1'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO managers (person_id, status, manager_name) "
+                "VALUES (:person_id, :status, :manager_name)",
+                [{'status': 's3', 'person_id': 4, 'manager_name': 'mn2'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO boss (golf_swing) VALUES (:golf_swing)",
+                [{'golf_swing': 'g1'}]
+            )
+        )
+
+    def test_bulk_insert_joined_inh_return_defaults(self):
+        Person, Engineer, Manager, Boss = \
+            self.classes('Person', 'Engineer', 'Manager', 'Boss')
+
+        s = Session()
+        with self.sql_execution_asserter() as asserter:
+            s.bulk_insert_mappings(
+                Boss,
+                [
+                    dict(
+                        name='b1', status='s1', manager_name='mn1',
+                        golf_swing='g1'
+                    ),
+                    dict(
+                        name='b2', status='s2', manager_name='mn2',
+                        golf_swing='g2'
+                    ),
+                    dict(
+                        name='b3', status='s3', manager_name='mn3',
+                        golf_swing='g3'
+                    ),
+                ]
+            )
+
+        # the only difference here is that common classes are grouped together.
+        # at the moment it doesn't lump all the "people" tables from
+        # different classes together.
+        asserter.assert_(
+            CompiledSQL(
+                "INSERT INTO people (name) VALUES (:name)",
+                [{'name': 'b1'}, {'name': 'b2'}, {'name': 'b3'}]
+            ),
+            CompiledSQL(
+                "INSERT INTO managers (status, manager_name) VALUES "
+                "(:status, :manager_name)",
+                [{'status': 's1', 'manager_name': 'mn1'},
+                 {'status': 's2', 'manager_name': 'mn2'},
+                 {'status': 's3', 'manager_name': 'mn3'}]
+
+            ),
+            CompiledSQL(
+                "INSERT INTO boss (golf_swing) VALUES (:golf_swing)",
+                [{'golf_swing': 'g1'},
+                 {'golf_swing': 'g2'}, {'golf_swing': 'g3'}]
+            )
+        )