Coverage for packages/sql-smith/src/sql_smith/capability/has_returning.py: 57%

1from sql_smith.functions import identify 

2 

3 

4class HasReturningMixin: 

5 def returning(self, column: str): 

6 self._returning = identify(column) 

7 return self 

8 

9 def _apply_returning(self, query: "ExpressionInterface") -> "ExpressionInterface": 

10 return ( 

11 query.append("RETURNING {}", self._returning) if self._returning else query 

12 )