diff options
| author | mike bayer <mike_mp@zzzcomputing.com> | 2017-02-01 18:13:06 -0500 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@awstats.zzzcomputing.com> | 2017-02-01 18:13:06 -0500 |
| commit | d71f4b47186972c5248c94ee2d04364da95a0965 (patch) | |
| tree | 5f558d187ad8f8c587077092699c823909067983 /test | |
| parent | 8342ca6c6932b8a30f4ea9650721244eeda35d51 (diff) | |
| parent | 388d8db68d0db6f38e72d60386023c9eef263034 (diff) | |
| download | sqlalchemy-d71f4b47186972c5248c94ee2d04364da95a0965.tar.gz | |
Merge "Accept FetchedValue, text() for column "default" value"
Diffstat (limited to 'test')
| -rw-r--r-- | test/engine/test_reflection.py | 53 |
1 files changed, 50 insertions, 3 deletions
diff --git a/test/engine/test_reflection.py b/test/engine/test_reflection.py index 0bc5b111e..9ddfe940f 100644 --- a/test/engine/test_reflection.py +++ b/test/engine/test_reflection.py @@ -1,13 +1,14 @@ import unicodedata import sqlalchemy as sa -from sqlalchemy import schema, inspect +from sqlalchemy import schema, inspect, sql from sqlalchemy import MetaData, Integer, String, Index, ForeignKey, \ - UniqueConstraint + UniqueConstraint, FetchedValue, DefaultClause from sqlalchemy.testing import ( ComparesTables, engines, AssertsCompiledSQL, fixtures, skip) from sqlalchemy.testing.schema import Table, Column -from sqlalchemy.testing import eq_, assert_raises, assert_raises_message +from sqlalchemy.testing import eq_, is_true, assert_raises, \ + assert_raises_message from sqlalchemy import testing from sqlalchemy.util import ue from sqlalchemy.testing import config @@ -1770,3 +1771,49 @@ class ColumnEventsTest(fixtures.RemovesEvents, fixtures.TestBase): "x", {"info": {"a": "b"}}, lambda table: eq_(table.c.x.info, {"a": "b"}) ) + + def test_override_server_default_fetchedvalue(self): + my_default = FetchedValue() + self._do_test( + "x", {"default": my_default}, + lambda table: eq_(table.c.x.server_default, my_default) + ) + + def test_override_server_default_default_clause(self): + my_default = DefaultClause("1") + self._do_test( + "x", {"default": my_default}, + lambda table: eq_(table.c.x.server_default, my_default) + ) + + def test_override_server_default_plain_text(self): + my_default = "1" + + def assert_text_of_one(table): + is_true( + isinstance( + table.c.x.server_default.arg, sql.elements.TextClause) + ) + eq_( + str(table.c.x.server_default.arg), "1" + ) + self._do_test( + "x", {"default": my_default}, + assert_text_of_one + ) + + def test_override_server_default_textclause(self): + my_default = sa.text("1") + + def assert_text_of_one(table): + is_true( + isinstance( + table.c.x.server_default.arg, sql.elements.TextClause) + ) + eq_( + str(table.c.x.server_default.arg), "1" + ) + self._do_test( + "x", {"default": my_default}, + assert_text_of_one + ) |
