def test_run_check(self, mocker, destination: Destination, tmp_path):
file_path = tmp_path / "config.json"
dummy_config = {"user": "******"}
write_file(file_path, dummy_config)
args = {"command": "check", "config": file_path}
parsed_args = argparse.Namespace(**args)
destination.run_cmd(parsed_args)
spec_msg = ConnectorSpecification(connectionSpecification={})
mocker.patch.object(destination, "spec", return_value=spec_msg)
validate_mock = mocker.patch(
"airbyte_cdk.destinations.destination.check_config_against_spec_or_exit"
)
expected_check_result = AirbyteConnectionStatus(
status=Status.SUCCEEDED)
mocker.patch.object(destination,
"check",
return_value=expected_check_result,
autospec=True)
returned_check_result = next(iter(destination.run_cmd(parsed_args)))
# verify method call with the correct params
# Affirm to Mypy that this is indeed a method on this mock
destination.check.assert_called_once() # type: ignore
# Affirm to Mypy that this is indeed a method on this mock
destination.check.assert_called_with(
logger=ANY, config=dummy_config) # type: ignore
# Check if config validation has been called
validate_mock.assert_called_with(dummy_config, spec_msg,
destination.logger)
# verify output was correct
assert _wrapped(expected_check_result) == returned_check_result
def test_run_spec(self, mocker, destination: Destination):
args = {"command": "spec"}
parsed_args = argparse.Namespace(**args)
expected_spec = ConnectorSpecification(
connectionSpecification={"json_schema": {
"prop": "value"
}})
mocker.patch.object(destination,
"spec",
return_value=expected_spec,
autospec=True)
spec_message = next(iter(destination.run_cmd(parsed_args)))
# Mypy doesn't understand magicmock so it thinks spec doesn't have assert_called_once attr
destination.spec.assert_called_once() # type: ignore
# verify the output of spec was returned
assert _wrapped(expected_spec) == spec_message
def test_run_cmd_with_incorrect_args_fails(self, args,
destination: Destination):
with pytest.raises(Exception):
list(destination.run_cmd(parsed_args=argparse.Namespace(**args)))
def test_run_write(self, mocker, destination: Destination, tmp_path,
monkeypatch):
config_path, dummy_config = tmp_path / "config.json", {
"user": "******"
}
write_file(config_path, dummy_config)
dummy_catalog = ConfiguredAirbyteCatalog(streams=[
ConfiguredAirbyteStream(
stream=AirbyteStream(name="mystream",
json_schema={"type": "object"}),
sync_mode=SyncMode.full_refresh,
destination_sync_mode=DestinationSyncMode.overwrite,
)
])
catalog_path = tmp_path / "catalog.json"
write_file(catalog_path, dummy_catalog.json(exclude_unset=True))
args = {
"command": "write",
"config": config_path,
"catalog": catalog_path
}
parsed_args = argparse.Namespace(**args)
expected_write_result = [
_wrapped(_state({"k1": "v1"})),
_wrapped(_state({"k2": "v2"}))
]
mocker.patch.object(
destination,
"write",
return_value=iter(expected_write_result),
autospec=True # convert to iterator to mimic real usage
)
spec_msg = ConnectorSpecification(connectionSpecification={})
mocker.patch.object(destination, "spec", return_value=spec_msg)
validate_mock = mocker.patch(
"airbyte_cdk.destinations.destination.check_config_against_spec_or_exit"
)
# mock input is a record followed by some state messages
mocked_input: List[AirbyteMessage] = [
_wrapped(_record("s1", {"k1": "v1"})), *expected_write_result
]
mocked_stdin_string = "\n".join(
[record.json(exclude_unset=True) for record in mocked_input])
mocked_stdin_string += "\n add this non-serializable string to verify the destination does not break on malformed input"
mocked_stdin = io.TextIOWrapper(
io.BytesIO(bytes(mocked_stdin_string, "utf-8")))
monkeypatch.setattr("sys.stdin", mocked_stdin)
returned_write_result = list(destination.run_cmd(parsed_args))
# verify method call with the correct params
# Affirm to Mypy that call_count is indeed a method on this mock
destination.write.assert_called_once() # type: ignore
# Affirm to Mypy that call_count is indeed a method on this mock
destination.write.assert_called_with( # type: ignore
config=dummy_config,
configured_catalog=dummy_catalog,
# Stdin is internally consumed as a generator so we use a custom matcher
# that iterates over two iterables to check equality
input_messages=OrderedIterableMatcher(mocked_input),
)
# Check if config validation has been called
validate_mock.assert_called_with(dummy_config, spec_msg,
destination.logger)
# verify output was correct
assert expected_write_result == returned_write_result