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

1from sql_smith.query import InsertQuery as BaseInsertQuery 

2 

3 

4class InsertQuery(BaseInsertQuery): 

5 """MySql InsertQuery that supports IGNORE.""" 

6 

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

8 super().__init__(engine) 

9 self._ignore = False 

10 

11 def ignore(self, status: bool) -> "InsertQuery": 

12 self._ignore = status 

13 return self 

14 

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

16 query = super().start_expression() 

17 if self._ignore: 

18 query = query.append("IGNORE") 

19 return query