From 9c109ed1f65a545b3a9056b01920fea088de32b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rodolphe=20Br=C3=A9ard?= <rodolphe@what.tf>
Date: Sat, 15 Apr 2023 22:07:27 +0200
Subject: [PATCH] Re-add the set_header_alg and set_body_alg functions, but
 only for the tests

---
 src/canonicalization.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/canonicalization.rs b/src/canonicalization.rs
index f548fce..aea5a1f 100644
--- a/src/canonicalization.rs
+++ b/src/canonicalization.rs
@@ -47,6 +47,18 @@ impl Canonicalization {
 			CanonicalizationType::Simple => body_simple(body),
 		}
 	}
+
+	#[cfg(test)]
+	pub fn set_header_alg(mut self, alg: CanonicalizationType) -> Self {
+		self.header_alg = alg;
+		self
+	}
+
+	#[cfg(test)]
+	pub fn set_body_alg(mut self, alg: CanonicalizationType) -> Self {
+		self.body_alg = alg;
+		self
+	}
 }
 
 impl Default for Canonicalization {