Obey settings

smithy-lang · Sep 30, 2024 · c20fec5 · c20fec5
1 parent 49292ec
commit c20fec5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/...serde/src/main/kotlin/software/amazon/smithy/rust/codegen/serde/SerializeImplGenerator.kt b/...serde/src/main/kotlin/software/amazon/smithy/rust/codegen/serde/SerializeImplGenerator.kt
@@ -234,6 +234,9 @@ class SerializeImplGenerator(private val codegenContext: CodegenContext) {
             implSerializeConfigured(symbolBuilder(shape, numericType).build()) {
                 rustTemplate(
                     """
+                    if !self.settings.out_of_range_floats_as_strings {
+                        return self.value.serialize(serializer)
+                    }
                     if self.value.is_nan() {
                         serializer.serialize_str("NaN")
                     } else if *self.value == #{ty}::INFINITY {

diff --git a/...gen-serde/src/test/kotlin/software/amazon/smithy/rust/codegen/serde/SerdeDecoratorTest.kt b/...gen-serde/src/test/kotlin/software/amazon/smithy/rust/codegen/serde/SerdeDecoratorTest.kt
@@ -381,6 +381,9 @@ class SerdeDecoratorTest {
                             settings.redact_sensitive_fields = true;
                             let serialized = #{serde_json}::to_string(&input.serialize_ref(&settings)).expect("failed to serialize");
                             assert_eq!(serialized, ${expectedRedacted.dq()});
+                            settings.out_of_range_floats_as_strings = false;
+                            let serialized = #{serde_json}::to_string(&input.serialize_ref(&settings)).expect("failed to serialize");
+                            assert_ne!(serialized, ${expectedRedacted.dq()});
                             """,
                             *codegenScope,
                         )