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

1from sql_smith.capability import HasReturningMixin 

2from sql_smith.query import UpdateQuery as BaseUpdateQuery 

3 

4 

5class UpdateQuery(HasReturningMixin, BaseUpdateQuery): 

6 """Postgres UpdateQuery that supports RETURNING.""" 

7 

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

9 query = super().as_expression() 

10 query = self._apply_returning(query) 

11 return query