diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2019-12-01 17:24:27 -0500 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2020-05-24 11:54:08 -0400 |
| commit | dce8c7a125cb99fad62c76cd145752d5afefae36 (patch) | |
| tree | 352dfa2c38005207ca64f45170bbba2c0f8c927e /test/sql/test_select.py | |
| parent | 1502b5b3e4e4b93021eb927a6623f288ef006ba6 (diff) | |
| download | sqlalchemy-dce8c7a125cb99fad62c76cd145752d5afefae36.tar.gz | |
Unify Query and select() , move all processing to compile phase
Convert Query to do virtually all compile state computation
in the _compile_context() phase, and organize it all
such that a plain select() construct may also be used as the
source of information in order to generate ORM query state.
This makes it such that Query is not needed except for
its additional methods like from_self() which are all to
be deprecated.
The construction of ORM state will occur beyond the
caching boundary when the new execution model is integrated.
future select() gains a working join() and filter_by() method.
as we continue to rebase and merge each commit in the steps,
callcounts continue to bump around. will have to look at
the final result when it's all in.
References: #5159
References: #4705
References: #4639
References: #4871
References: #5010
Change-Id: I19e05b3424b07114cce6c439b05198ac47f7ac10
Diffstat (limited to 'test/sql/test_select.py')
| -rw-r--r-- | test/sql/test_select.py | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/test/sql/test_select.py b/test/sql/test_select.py new file mode 100644 index 000000000..7bac921a1 --- /dev/null +++ b/test/sql/test_select.py @@ -0,0 +1,163 @@ +from sqlalchemy import Column +from sqlalchemy import exc +from sqlalchemy import ForeignKey +from sqlalchemy import Integer +from sqlalchemy import MetaData +from sqlalchemy import String +from sqlalchemy import Table +from sqlalchemy.future import select as future_select +from sqlalchemy.sql import column +from sqlalchemy.sql import table +from sqlalchemy.testing import assert_raises_message +from sqlalchemy.testing import AssertsCompiledSQL +from sqlalchemy.testing import fixtures + +table1 = table( + "mytable", + column("myid", Integer), + column("name", String), + column("description", String), +) + +table2 = table( + "myothertable", column("otherid", Integer), column("othername", String) +) + +metadata = MetaData() + + +parent = Table( + "parent", + metadata, + Column("id", Integer, primary_key=True), + Column("data", String(50)), +) +child = Table( + "child", + metadata, + Column("id", Integer, primary_key=True), + Column("parent_id", ForeignKey("parent.id")), + Column("data", String(50)), +) + + +class FutureSelectTest(fixtures.TestBase, AssertsCompiledSQL): + __dialect__ = "default" + + def test_join_nofrom_implicit_left_side_explicit_onclause(self): + stmt = future_select(table1).join( + table2, table1.c.myid == table2.c.otherid + ) + + self.assert_compile( + stmt, + "SELECT mytable.myid, mytable.name, mytable.description " + "FROM mytable JOIN myothertable " + "ON mytable.myid = myothertable.otherid", + ) + + def test_join_nofrom_explicit_left_side_explicit_onclause(self): + stmt = future_select(table1).join_from( + table1, table2, table1.c.myid == table2.c.otherid + ) + + self.assert_compile( + stmt, + "SELECT mytable.myid, mytable.name, mytable.description " + "FROM mytable JOIN myothertable " + "ON mytable.myid = myothertable.otherid", + ) + + def test_join_nofrom_implicit_left_side_implicit_onclause(self): + stmt = future_select(parent).join(child) + + self.assert_compile( + stmt, + "SELECT parent.id, parent.data FROM parent JOIN child " + "ON parent.id = child.parent_id", + ) + + def test_join_nofrom_explicit_left_side_implicit_onclause(self): + stmt = future_select(parent).join_from(parent, child) + + self.assert_compile( + stmt, + "SELECT parent.id, parent.data FROM parent JOIN child " + "ON parent.id = child.parent_id", + ) + + def test_join_froms_implicit_left_side_explicit_onclause(self): + stmt = ( + future_select(table1) + .select_from(table1) + .join(table2, table1.c.myid == table2.c.otherid) + ) + + self.assert_compile( + stmt, + "SELECT mytable.myid, mytable.name, mytable.description " + "FROM mytable JOIN myothertable " + "ON mytable.myid = myothertable.otherid", + ) + + def test_join_froms_explicit_left_side_explicit_onclause(self): + stmt = ( + future_select(table1) + .select_from(table1) + .join_from(table1, table2, table1.c.myid == table2.c.otherid) + ) + + self.assert_compile( + stmt, + "SELECT mytable.myid, mytable.name, mytable.description " + "FROM mytable JOIN myothertable " + "ON mytable.myid = myothertable.otherid", + ) + + def test_join_froms_implicit_left_side_implicit_onclause(self): + stmt = future_select(parent).select_from(parent).join(child) + + self.assert_compile( + stmt, + "SELECT parent.id, parent.data FROM parent JOIN child " + "ON parent.id = child.parent_id", + ) + + def test_join_froms_explicit_left_side_implicit_onclause(self): + stmt = ( + future_select(parent).select_from(parent).join_from(parent, child) + ) + + self.assert_compile( + stmt, + "SELECT parent.id, parent.data FROM parent JOIN child " + "ON parent.id = child.parent_id", + ) + + def test_joins_w_filter_by(self): + stmt = ( + future_select(parent) + .filter_by(data="p1") + .join(child) + .filter_by(data="c1") + .join_from(table1, table2, table1.c.myid == table2.c.otherid) + .filter_by(otherid=5) + ) + + self.assert_compile( + stmt, + "SELECT parent.id, parent.data FROM parent JOIN child " + "ON parent.id = child.parent_id, mytable JOIN myothertable " + "ON mytable.myid = myothertable.otherid " + "WHERE parent.data = :data_1 AND child.data = :data_2 " + "AND myothertable.otherid = :otherid_1", + checkparams={"data_1": "p1", "data_2": "c1", "otherid_1": 5}, + ) + + def test_filter_by_no_property(self): + assert_raises_message( + exc.InvalidRequestError, + 'Entity namespace for "mytable" has no property "foo"', + future_select(table1).filter_by, + foo="bar", + ) |
