mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-18 06:32:08 -04:00
Khairi Hafsham
772374735d
tested using pycodestyle version 2.2.0 Fixes: #3885 Change-Id: I5df43adc3aefe318f9eeab72a078247a548ec566 Pull-request: https://github.com/zzzeek/sqlalchemy/pull/343
132 lines
5.0 KiB
Python
132 lines
5.0 KiB
Python
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing import AssertsCompiledSQL, assert_raises_message
|
|
from sqlalchemy.sql import select, func
|
|
from sqlalchemy.engine import default
|
|
from sqlalchemy import exc
|
|
from sqlalchemy import Table, Integer, String, ForeignKey, Column, true
|
|
from sqlalchemy import lateral, outerjoin, join
|
|
|
|
|
|
class LateralTest(fixtures.TablesTest, AssertsCompiledSQL):
|
|
__dialect__ = default.DefaultDialect(supports_native_boolean=True)
|
|
|
|
run_setup_bind = None
|
|
|
|
run_create_tables = None
|
|
|
|
@classmethod
|
|
def define_tables(cls, metadata):
|
|
Table('people', metadata,
|
|
Column('people_id', Integer, primary_key=True),
|
|
Column('age', Integer),
|
|
Column('name', String(30)))
|
|
Table('bookcases', metadata,
|
|
Column('bookcase_id', Integer, primary_key=True),
|
|
Column(
|
|
'bookcase_owner_id',
|
|
Integer, ForeignKey('people.people_id')),
|
|
Column('bookcase_shelves', Integer),
|
|
Column('bookcase_width', Integer))
|
|
Table('books', metadata,
|
|
Column('book_id', Integer, primary_key=True),
|
|
Column(
|
|
'bookcase_id', Integer, ForeignKey('bookcases.bookcase_id')),
|
|
Column('book_owner_id', Integer, ForeignKey('people.people_id')),
|
|
Column('book_weight', Integer))
|
|
|
|
def test_standalone(self):
|
|
table1 = self.tables.people
|
|
subq = select([table1.c.people_id])
|
|
|
|
# alias name is not rendered because subquery is not
|
|
# in the context of a FROM clause
|
|
self.assert_compile(
|
|
lateral(subq, name="alias"),
|
|
'LATERAL (SELECT people.people_id FROM people)'
|
|
)
|
|
|
|
self.assert_compile(
|
|
subq.lateral(name="alias"),
|
|
'LATERAL (SELECT people.people_id FROM people)'
|
|
)
|
|
|
|
def test_select_from(self):
|
|
table1 = self.tables.people
|
|
subq = select([table1.c.people_id])
|
|
|
|
# in a FROM context, now you get "AS alias" and column labeling
|
|
self.assert_compile(
|
|
select([subq.lateral(name='alias')]),
|
|
'SELECT alias.people_id FROM LATERAL '
|
|
'(SELECT people.people_id AS people_id FROM people) AS alias'
|
|
)
|
|
|
|
def test_plain_join(self):
|
|
table1 = self.tables.people
|
|
table2 = self.tables.books
|
|
subq = select([table2.c.book_id]).\
|
|
where(table2.c.book_owner_id == table1.c.people_id)
|
|
|
|
# FROM books, people? isn't this wrong? No! Because
|
|
# this is only a fragment, books isn't in any other FROM clause
|
|
self.assert_compile(
|
|
join(table1, lateral(subq, name="alias"), true()),
|
|
"people JOIN LATERAL (SELECT books.book_id AS book_id "
|
|
"FROM books, people WHERE books.book_owner_id = people.people_id) "
|
|
"AS alias ON true"
|
|
)
|
|
|
|
# put it in correct context, implicit correlation works fine
|
|
self.assert_compile(
|
|
select([table1]).select_from(
|
|
join(table1, lateral(subq, name="alias"), true())
|
|
),
|
|
"SELECT people.people_id, people.age, people.name "
|
|
"FROM people JOIN LATERAL (SELECT books.book_id AS book_id "
|
|
"FROM books WHERE books.book_owner_id = people.people_id) "
|
|
"AS alias ON true"
|
|
)
|
|
|
|
# explicit correlation
|
|
subq = subq.correlate(table1)
|
|
self.assert_compile(
|
|
select([table1]).select_from(
|
|
join(table1, lateral(subq, name="alias"), true())
|
|
),
|
|
"SELECT people.people_id, people.age, people.name "
|
|
"FROM people JOIN LATERAL (SELECT books.book_id AS book_id "
|
|
"FROM books WHERE books.book_owner_id = people.people_id) "
|
|
"AS alias ON true"
|
|
)
|
|
|
|
def test_join_lateral_w_select_subquery(self):
|
|
table1 = self.tables.people
|
|
table2 = self.tables.books
|
|
|
|
subq = select([table2.c.book_id]).\
|
|
correlate(table1).\
|
|
where(table1.c.people_id == table2.c.book_owner_id).lateral()
|
|
stmt = select([table1, subq.c.book_id]).\
|
|
select_from(table1.join(subq, true()))
|
|
|
|
self.assert_compile(
|
|
stmt,
|
|
"SELECT people.people_id, people.age, people.name, anon_1.book_id "
|
|
"FROM people JOIN LATERAL (SELECT books.book_id AS book_id "
|
|
"FROM books "
|
|
"WHERE people.people_id = books.book_owner_id) AS anon_1 ON true"
|
|
)
|
|
|
|
def test_from_function(self):
|
|
bookcases = self.tables.bookcases
|
|
srf = lateral(func.generate_series(1, bookcases.c.bookcase_shelves))
|
|
|
|
self.assert_compile(
|
|
select([bookcases]).select_from(bookcases.join(srf, true())),
|
|
"SELECT bookcases.bookcase_id, bookcases.bookcase_owner_id, "
|
|
"bookcases.bookcase_shelves, bookcases.bookcase_width "
|
|
"FROM bookcases JOIN "
|
|
"LATERAL generate_series(:generate_series_1, "
|
|
"bookcases.bookcase_shelves) AS anon_1 ON true"
|
|
)
|