From ee767d8067780176bceaaf9b9069f76cd5a39ca2 Mon Sep 17 00:00:00 2001
From: Isaac Shoebottom <ishoebot@unb.ca>
Date: Fri, 21 Oct 2022 00:31:35 -0300
Subject: [PATCH] Make body function work with all tests

---
 assignments/A3/message.js | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/assignments/A3/message.js b/assignments/A3/message.js
index 9af0c5f..1b713ea 100644
--- a/assignments/A3/message.js
+++ b/assignments/A3/message.js
@@ -8,13 +8,16 @@ function match(message, headers) {
 }
 
 function body(message) {
-	if (message["body"][0]["content-type"] === "text/plain") {
-		return message["body"][0]["content"];
-	}
-	/* istanbul ignore else */
-	if (message["body"][0]["content-type"] === "multipart/signed") {
-		return message["body"][0]["content"][0]["content"];
+	if (message["body"] !== undefined) {
+		if (message["body"][0]["content-type"] === "text/plain") {
+			return message["body"][0]["content"];
+		}
+		/* istanbul ignore else */
+		if (message["body"][0]["content-type"] === "multipart/signed") {
+			return message["body"][0]["content"][0]["content"];
+		}
 	}
+	return undefined;
 }
 
 function string(message) {