Skip to content

Commit

Permalink
Fixed unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
ckunki committed Oct 9, 2024
1 parent b4ee260 commit 6114c09
Showing 1 changed file with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -18,15 +18,15 @@

class TestJSONQueryHandler(JSONQueryHandler):
__test__ = False
def __init__(self, parameter: JSONType, query_handler_context: ScopeQueryHandlerContext):
super().__init__(parameter, query_handler_context)
self._parameter = parameter
def __init__(self, parameters: JSONType, query_handler_context: ScopeQueryHandlerContext):
super().__init__(parameters, query_handler_context)
self._parameters = parameters

def start(self) -> Union[Continue, Finish[JSONType]]:
return Finish[JSONType](self._parameter)
return Finish[JSONType](self._parameters)

def handle_query_result(self, query_result: QueryResult) -> Union[Continue, Finish[JSONType]]:
return Finish[JSONType](self._parameter)
return Finish[JSONType](self._parameters)


class TestJsonUDFQueryHandlerFactory(JsonUDFQueryHandlerFactory):
Expand Down

0 comments on commit 6114c09

Please sign in to comment.