Skip to content

Commit fbf21c1

Browse files
Merge pull request #231 from sanger/fix-issues-with-reading
Fixing issues with reader schema versions
2 parents 2e99b76 + 8483690 commit fbf21c1

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

lab_share_lib/processing/rabbit_message.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def decode(self, possible_encoders: List[AvroEncoderBase], reader_schema_version
3939
try:
4040
LOGGER.debug(f"Attempting to decode message with encoder class '{type(encoder).__name__}'.")
4141
self._decoded_list = list(
42-
encoder.decode(self.encoded_body, self.writer_schema_version, reader_schema_version)
42+
encoder.decode(self.encoded_body, reader_schema_version, self.writer_schema_version)
4343
)
4444
if encoder.encoder_type == ENCODER_TYPE_BINARY:
4545
MESSAGE_LOGGER.info(f"Decoded binary message body:\n{self._decoded_list}")

tests/processing/test_rabbit_message.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def test_schema_version_extracts_the_header_correctly(subject):
5252
def test_decode_populates_decoded_list(subject, valid_decoder):
5353
subject.decode([valid_decoder], "1")
5454

55-
valid_decoder.decode.assert_called_once_with(ENCODED_BODY, HEADERS[RABBITMQ_HEADER_KEY_VERSION], "1")
55+
valid_decoder.decode.assert_called_once_with(ENCODED_BODY, "1", HEADERS[RABBITMQ_HEADER_KEY_VERSION])
5656
assert subject._decoded_list == DECODED_LIST
5757

5858

@@ -71,8 +71,8 @@ def test_decode_returns_the_successful_second_decoder(subject, error_decoder, va
7171
def test_decode_successfully_decodes_if_second_decoder_works(subject, error_decoder, valid_decoder):
7272
subject.decode([error_decoder, valid_decoder], "1")
7373

74-
error_decoder.decode.assert_called_once_with(ENCODED_BODY, HEADERS[RABBITMQ_HEADER_KEY_VERSION], "1")
75-
valid_decoder.decode.assert_called_once_with(ENCODED_BODY, HEADERS[RABBITMQ_HEADER_KEY_VERSION], "1")
74+
error_decoder.decode.assert_called_once_with(ENCODED_BODY, "1", HEADERS[RABBITMQ_HEADER_KEY_VERSION])
75+
valid_decoder.decode.assert_called_once_with(ENCODED_BODY, "1", HEADERS[RABBITMQ_HEADER_KEY_VERSION])
7676
assert subject._decoded_list == DECODED_LIST
7777

7878

tests/rabbit/test_avro_encoder.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def json_writer(string_writer, schema, record_list):
137137
def test_decode_decodes_the_message(self, json_subject, fastavro, schema_version):
138138
fastavro.json_reader.return_value = SCHEMA_DICT
139139

140-
result = json_subject.decode(MESSAGE_BODY.encode(), schema_version, "1")
140+
result = json_subject.decode(MESSAGE_BODY.encode(), "1", schema_version)
141141

142142
fastavro.json_reader.assert_called_once_with(ANY, fastavro.parse_schema.return_value)
143143
string_reader = fastavro.json_reader.call_args.args[0]
@@ -150,7 +150,7 @@ def test_both_encode_and_decode_actions_work_together(self, json_subject, schema
150150
records = [MESSAGE_BODY]
151151

152152
message = json_subject.encode(records, schema_version)
153-
result = json_subject.decode(message.body, schema_version, "1")
153+
result = json_subject.decode(message.body, "1", schema_version)
154154

155155
assert list(result) == records
156156

@@ -179,7 +179,7 @@ def test_encode_encodes_the_message(self, binary_file_subject, schema_version):
179179
def test_decode_decodes_the_message(self, binary_file_subject, schema_version, file):
180180
records = [MESSAGE_BODY]
181181

182-
result = binary_file_subject.decode(file, schema_version, "1")
182+
result = binary_file_subject.decode(file, "1", schema_version)
183183

184184
assert list(result) == records
185185

@@ -188,7 +188,7 @@ def test_both_encode_and_decode_actions_work_together(self, binary_file_subject,
188188
records = [MESSAGE_BODY]
189189

190190
message = binary_file_subject.encode(records, schema_version)
191-
result = binary_file_subject.decode(message.body, schema_version, "1")
191+
result = binary_file_subject.decode(message.body, "1", schema_version)
192192

193193
assert list(result) == records
194194

@@ -224,7 +224,7 @@ def test_encode_single_object_encodes_the_message(self, binary_message_subject,
224224
def test_decode_decodes_the_message(self, binary_message_subject, schema_version, message):
225225
records = [MESSAGE_BODY]
226226

227-
result = binary_message_subject.decode(message, schema_version, "1")
227+
result = binary_message_subject.decode(message, "1", schema_version)
228228

229229
assert result == records
230230

@@ -233,7 +233,7 @@ def test_both_encode_and_decode_actions_work_together(self, binary_message_subje
233233
records = [MESSAGE_BODY]
234234

235235
message = binary_message_subject.encode(records, schema_version)
236-
result = binary_message_subject.decode(message.body, schema_version, "1")
236+
result = binary_message_subject.decode(message.body, "1", schema_version)
237237

238238
assert result == records
239239

@@ -242,7 +242,7 @@ def test_both_encode_single_object_and_decode_actions_work_together(self, binary
242242
records = [MESSAGE_BODY]
243243

244244
message = binary_message_subject.encode_single_object(records[0], schema_version)
245-
result = binary_message_subject.decode(message.body, schema_version, "1")
245+
result = binary_message_subject.decode(message.body, "1", schema_version)
246246

247247
assert result == records
248248

0 commit comments

Comments
 (0)