diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2015-08-17 17:04:33 -0400 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2015-08-17 17:12:16 -0400 |
| commit | d14a4b480c3b43885707e4a6e2466589971ff4d5 (patch) | |
| tree | fab9267e749501c7b32c9e6e9f9f8429ef2b5644 /test/dialect/postgresql | |
| parent | ceeb033054f09db3eccbde3fad1941ec42919a54 (diff) | |
| download | sqlalchemy-d14a4b480c3b43885707e4a6e2466589971ff4d5.tar.gz | |
- merge of ticket_3514 None-handling branch
- Fixes to the ORM and to the postgresql JSON type regarding the
``None`` constant in conjunction with the Postgresql :class:`.JSON` type. When
the :paramref:`.JSON.none_as_null` flag is left at its default
value of ``False``, the ORM will now correctly insert the Json
"'null'" string into the column whenever the value on the ORM
object is set to the value ``None`` or when the value ``None``
is used with :meth:`.Session.bulk_insert_mappings`,
**including** if the column has a default or server default on it. This
makes use of a new type-level flag "evaluates_none" which is implemented
by the JSON type based on the none_as_null flag. fixes #3514
- Added a new constant :attr:`.postgresql.JSON.NULL`, indicating
that the JSON NULL value should be used for a value
regardless of other settings. part of fixes #3514
Diffstat (limited to 'test/dialect/postgresql')
| -rw-r--r-- | test/dialect/postgresql/test_types.py | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/test/dialect/postgresql/test_types.py b/test/dialect/postgresql/test_types.py index 9e0e5bcc6..00a2de2db 100644 --- a/test/dialect/postgresql/test_types.py +++ b/test/dialect/postgresql/test_types.py @@ -2398,6 +2398,15 @@ class JSONRoundTripTest(fixtures.TablesTest): ).fetchall() eq_([d for d, in data], [None]) + def _assert_column_is_JSON_NULL(self, column='data'): + col = self.tables.data_table.c[column] + + data = testing.db.execute( + select([col]). + where(cast(col, String) == "null") + ).fetchall() + eq_([d for d, in data], [None]) + def _test_insert(self, engine): engine.execute( self.tables.data_table.insert(), @@ -2419,6 +2428,13 @@ class JSONRoundTripTest(fixtures.TablesTest): ) self._assert_column_is_NULL(column='nulldata') + def _test_insert_nulljson_into_none_as_null(self, engine): + engine.execute( + self.tables.data_table.insert(), + {'name': 'r1', 'nulldata': JSON.NULL} + ) + self._assert_column_is_JSON_NULL(column='nulldata') + def _non_native_engine(self, json_serializer=None, json_deserializer=None): if json_serializer is not None or json_deserializer is not None: options = { @@ -2467,6 +2483,11 @@ class JSONRoundTripTest(fixtures.TablesTest): engine = testing.db self._test_insert_none_as_null(engine) + @testing.requires.psycopg2_native_json + def test_insert_native_nulljson_into_none_as_null(self): + engine = testing.db + self._test_insert_nulljson_into_none_as_null(engine) + def test_insert_python(self): engine = self._non_native_engine() self._test_insert(engine) @@ -2479,6 +2500,10 @@ class JSONRoundTripTest(fixtures.TablesTest): engine = self._non_native_engine() self._test_insert_none_as_null(engine) + def test_insert_python_nulljson_into_none_as_null(self): + engine = self._non_native_engine() + self._test_insert_nulljson_into_none_as_null(engine) + def _test_custom_serialize_deserialize(self, native): import json @@ -2645,6 +2670,36 @@ class JSONRoundTripTest(fixtures.TablesTest): engine = testing.db self._test_unicode_round_trip(engine) + def test_eval_none_flag_orm(self): + Base = declarative_base() + + class Data(Base): + __table__ = self.tables.data_table + + s = Session(testing.db) + + d1 = Data(name='d1', data=None, nulldata=None) + s.add(d1) + s.commit() + + s.bulk_insert_mappings( + Data, [{"name": "d2", "data": None, "nulldata": None}] + ) + eq_( + s.query( + cast(self.tables.data_table.c.data, String), + cast(self.tables.data_table.c.nulldata, String) + ).filter(self.tables.data_table.c.name == 'd1').first(), + ("null", None) + ) + eq_( + s.query( + cast(self.tables.data_table.c.data, String), + cast(self.tables.data_table.c.nulldata, String) + ).filter(self.tables.data_table.c.name == 'd2').first(), + ("null", None) + ) + class JSONBTest(JSONTest): |
