From b2c24fab4f2a7ad5ef00a345ee25b2a1f6226e18 Mon Sep 17 00:00:00 2001
From: Steve Faulkner <471400+southpolesteve@users.noreply.github.com>
Date: Tue, 29 Sep 2020 17:31:58 -0500
Subject: [PATCH] More test cases

---
 src/Utils/MessageValidation.test.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Utils/MessageValidation.test.ts b/src/Utils/MessageValidation.test.ts
index bc205241f..b55473fa9 100644
--- a/src/Utils/MessageValidation.test.ts
+++ b/src/Utils/MessageValidation.test.ts
@@ -16,6 +16,8 @@ test.each`
   ${"https://main.documentdb.ext.microsoftazure.de"} | ${false}
   ${"https://random.domain"}                         | ${true}
   ${"https://malicious.cloudapp.azure.com"}          | ${true}
+  ${"https://maliciousazure.com"}                    | ${true}
+  ${"https://maliciousportalsazure.com"}             | ${true}
 `("returns $expected when called with $domain", ({ domain, expected }) => {
   expect(isInvalidParentFrameOrigin({ origin: domain } as MessageEvent)).toBe(expected);
 });