Skip to content

Commit 05820f2

Browse files
fix: [SNOW-1890085] restore lost changes during rebasing
1 parent b6853c1 commit 05820f2

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/conftest.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from io import StringIO
2525
from logging import FileHandler
2626
from pathlib import Path
27-
from typing import Generator, List, NamedTuple, Optional, Union
27+
from typing import Any, Dict, Generator, List, NamedTuple, Optional, Union
2828
from unittest import mock
2929

3030
import pytest
@@ -276,6 +276,7 @@ def __init__(
276276
self._checkout_count = 0
277277
self._role = role
278278
self._warehouse = warehouse
279+
self.kwargs: List[Dict[str, Any]] = []
279280

280281
def get_query(self):
281282
return "\n".join(self.queries)
@@ -315,6 +316,7 @@ def execute_string(self, query: str, **kwargs):
315316
if self._checkout_count > 1:
316317
raise ProgrammingError("Checkout already exists")
317318
self.queries.append(query)
319+
self.kwargs.append(kwargs)
318320
return (self.cs,)
319321

320322
def execute_stream(self, query: StringIO, **kwargs):

0 commit comments

Comments
 (0)