Coverage for packages/sql-smith/src/sql_smith/query/sql_server/delete_query.py: 50%

1from sql_smith.functions import literal 

2from sql_smith.query import DeleteQuery as BaseDeleteQuery 

3 

4 

5class DeleteQuery(BaseDeleteQuery): 

6 """Sql Server DeleteQuery to support LIMIT.""" 

7 

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

9 query = super().start_expression() 

10 if self._limit is None: 

11 return query 

12 return query.append("TOP({})", literal(self._limit)) 

13 

14 def _apply_limit(self, query: "ExpressionInterface") -> "ExpressionInterface": 

15 return query