← Back to branch summary

~ubuntu-branches/debian/jessie/sqlalchemy/jessie

« back to all changes in this revision

Viewing changes to lib/sqlalchemy/testing/suite/test_insert.py

  • Committer: Package Import Robot
  • Author(s): Piotr Ożarowski, Jakub Wilk, Piotr Ożarowski
  • Date: 2013-07-06 20:53:52 UTC
  • mfrom: (1.4.23) (16.1.17 experimental)
  • Revision ID: package-import@ubuntu.com-20130706205352-ryppl1eto3illd79
Tags: 0.8.2-1
[ Jakub Wilk ]
* Use canonical URIs for Vcs-* fields.

[ Piotr Ożarowski ]
* New upstream release
* Upload to unstable
* Build depend on python3-all instead of -dev, extensions are not built for
  Python 3.X 

Show diffs side-by-side

added added

removed removed

Lines of Context:
lib/sqlalchemy/testing/suite/test_insert.py
 
1
from .. import fixtures, config
 
2
from ..config import requirements
 
3
from .. import exclusions
 
4
from ..assertions import eq_
 
5
from .. import engines
 
6
 
 
7
from sqlalchemy import Integer, String, select, util
 
8
 
 
9
from ..schema import Table, Column
 
10
 
 
11
 
 
12
class LastrowidTest(fixtures.TablesTest):
 
13
    run_deletes = 'each'
 
14
 
 
15
    __requires__ = 'implements_get_lastrowid', 'autoincrement_insert'
 
16
 
 
17
    __engine_options__ = {"implicit_returning": False}
 
18
 
 
19
    @classmethod
 
20
    def define_tables(cls, metadata):
 
21
        Table('autoinc_pk', metadata,
 
22
                Column('id', Integer, primary_key=True,
 
23
                                    test_needs_autoincrement=True),
 
24
                Column('data', String(50))
 
25
            )
 
26
 
 
27
        Table('manual_pk', metadata,
 
28
                Column('id', Integer, primary_key=True, autoincrement=False),
 
29
                Column('data', String(50))
 
30
            )
 
31
 
 
32
    def _assert_round_trip(self, table, conn):
 
33
        row = conn.execute(table.select()).first()
 
34
        eq_(
 
35
            row,
 
36
            (config.db.dialect.default_sequence_base, "some data")
 
37
        )
 
38
 
 
39
    def test_autoincrement_on_insert(self):
 
40
 
 
41
        config.db.execute(
 
42
            self.tables.autoinc_pk.insert(),
 
43
            data="some data"
 
44
        )
 
45
        self._assert_round_trip(self.tables.autoinc_pk, config.db)
 
46
 
 
47
    def test_last_inserted_id(self):
 
48
 
 
49
        r = config.db.execute(
 
50
            self.tables.autoinc_pk.insert(),
 
51
            data="some data"
 
52
        )
 
53
        pk = config.db.scalar(select([self.tables.autoinc_pk.c.id]))
 
54
        eq_(
 
55
            r.inserted_primary_key,
 
56
            [pk]
 
57
        )
 
58
 
 
59
    @exclusions.fails_if(lambda: util.pypy, "lastrowid not maintained after "
 
60
                            "connection close")
 
61
    @requirements.dbapi_lastrowid
 
62
    def test_native_lastrowid_autoinc(self):
 
63
        r = config.db.execute(
 
64
            self.tables.autoinc_pk.insert(),
 
65
            data="some data"
 
66
        )
 
67
        lastrowid = r.lastrowid
 
68
        pk = config.db.scalar(select([self.tables.autoinc_pk.c.id]))
 
69
        eq_(
 
70
            lastrowid, pk
 
71
        )
 
72
 
 
73
 
 
74
class InsertBehaviorTest(fixtures.TablesTest):
 
75
    run_deletes = 'each'
 
76
 
 
77
    @classmethod
 
78
    def define_tables(cls, metadata):
 
79
        Table('autoinc_pk', metadata,
 
80
                Column('id', Integer, primary_key=True, \
 
81
                                test_needs_autoincrement=True),
 
82
                Column('data', String(50))
 
83
            )
 
84
 
 
85
    def test_autoclose_on_insert(self):
 
86
        if requirements.returning.enabled:
 
87
            engine = engines.testing_engine(
 
88
                            options={'implicit_returning': False})
 
89
        else:
 
90
            engine = config.db
 
91
 
 
92
        r = engine.execute(
 
93
            self.tables.autoinc_pk.insert(),
 
94
            data="some data"
 
95
        )
 
96
        assert r.closed
 
97
        assert r.is_insert
 
98
        assert not r.returns_rows
 
99
 
 
100
    @requirements.returning
 
101
    def test_autoclose_on_insert_implicit_returning(self):
 
102
        r = config.db.execute(
 
103
            self.tables.autoinc_pk.insert(),
 
104
            data="some data"
 
105
        )
 
106
        assert r.closed
 
107
        assert r.is_insert
 
108
        assert not r.returns_rows
 
109
 
 
110
    @requirements.empty_inserts
 
111
    def test_empty_insert(self):
 
112
        r = config.db.execute(
 
113
            self.tables.autoinc_pk.insert(),
 
114
            )
 
115
        assert r.closed
 
116
 
 
117
        r = config.db.execute(
 
118
            self.tables.autoinc_pk.select().\
 
119
                    where(self.tables.autoinc_pk.c.id != None)
 
120
        )
 
121
 
 
122
        assert len(r.fetchall())
 
123
 
 
124
 
 
125
class ReturningTest(fixtures.TablesTest):
 
126
    run_deletes = 'each'
 
127
    __requires__ = 'returning', 'autoincrement_insert'
 
128
 
 
129
    __engine_options__ = {"implicit_returning": True}
 
130
 
 
131
    def _assert_round_trip(self, table, conn):
 
132
        row = conn.execute(table.select()).first()
 
133
        eq_(
 
134
            row,
 
135
            (config.db.dialect.default_sequence_base, "some data")
 
136
        )
 
137
 
 
138
    @classmethod
 
139
    def define_tables(cls, metadata):
 
140
        Table('autoinc_pk', metadata,
 
141
                Column('id', Integer, primary_key=True, \
 
142
                                test_needs_autoincrement=True),
 
143
                Column('data', String(50))
 
144
            )
 
145
 
 
146
    def test_explicit_returning_pk(self):
 
147
        engine = config.db
 
148
        table = self.tables.autoinc_pk
 
149
        r = engine.execute(
 
150
            table.insert().returning(
 
151
                            table.c.id),
 
152
            data="some data"
 
153
        )
 
154
        pk = r.first()[0]
 
155
        fetched_pk = config.db.scalar(select([table.c.id]))
 
156
        eq_(fetched_pk, pk)
 
157
 
 
158
    def test_autoincrement_on_insert_implcit_returning(self):
 
159
 
 
160
        config.db.execute(
 
161
            self.tables.autoinc_pk.insert(),
 
162
            data="some data"
 
163
        )
 
164
        self._assert_round_trip(self.tables.autoinc_pk, config.db)
 
165
 
 
166
    def test_last_inserted_id_implicit_returning(self):
 
167
 
 
168
        r = config.db.execute(
 
169
            self.tables.autoinc_pk.insert(),
 
170
            data="some data"
 
171
        )
 
172
        pk = config.db.scalar(select([self.tables.autoinc_pk.c.id]))
 
173
        eq_(
 
174
            r.inserted_primary_key,
 
175
            [pk]
 
176
        )
 
177
 
 
178
 
 
179
__all__ = ('LastrowidTest', 'InsertBehaviorTest', 'ReturningTest')