Coverage for packages/sql-smith/src/sql_smith/query/postgres/insert_query.py: 57%

1from sql_smith.capability import HasReturningMixin 

2from sql_smith.query import InsertQuery as BaseInsertQuery 

3 

4 

5class InsertQuery(HasReturningMixin, BaseInsertQuery): 

6 """Postgres InsertQuery that supports RETURNING.""" 

7 

8 def as_expression(self) -> "ExpressionInterface": 

9 query = super().as_expression() 

10 query = self._apply_returning(query) 

11 return query