Coverage for packages/sql-smith/src/sql_smith/query/mysql/select_query.py: 77%

1from sql_smith.query import SelectQuery as BaseSelectQuery 

2 

3 

4class SelectQuery(BaseSelectQuery): 

5 """Mysql SelectQuery that supports SQL_CALC_FOUND_ROWS.""" 

6 

7 def __init__(self, engine: "EngineInterface"): 

8 super().__init__(engine) 

9 self._calc_found_rows = False 

10 

11 def calc_found_rows(self, status: bool) -> "SelectQuery": 

12 self._calc_found_rows = status 

13 return self 

14 

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

16 query = super().start_expression() 

17 if self._calc_found_rows: 

18 query = query.append("SQL_CALC_FOUND_ROWS") 

19 return query