def METHOD_NAME(self):
    state = AirbyteStateMessage(data={})
    self.source.read.return_value = [
        AirbyteMessage(type=Type.LOG, log=AirbyteLogMessage(level=Level.INFO, message="test")),
        AirbyteMessage(type=Type.RECORD, record=AirbyteRecordMessage(stream="test", data={"test": 1}, emitted_at=1)),
        AirbyteMessage(type=Type.STATE, state=state),
    ]
    result = list(self.integration._load_data("test", None))
    self.assertEqual(
        result,
        [
            {"data": {"test": 1}, "id": "1"},
        ],
    )
    self.integration.last_state = state