From 8f4a77e44ee8f712363877b52976202a1d399d7c Mon Sep 17 00:00:00 2001
From: "seer-by-sentry[bot]"
 <157164994+seer-by-sentry[bot]@users.noreply.github.com>
Date: Thu, 16 Oct 2025 16:14:14 +0000
Subject: [PATCH] test: add unit tests for PR#1058

---
 flask/src/test/test_utils.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/flask/src/test/test_utils.py b/flask/src/test/test_utils.py
index 50b3717ae..a17a65677 100644
--- a/flask/src/test/test_utils.py
+++ b/flask/src/test/test_utils.py
@@ -2,7 +2,7 @@
 
 import unittest
 from unittest.mock import patch, Mock
-from ..utils import weighter, parseHeaders, get_iterator, yuval, chris
+from ..utils import weighter, parseHeaders, get_iterator, yuval, chris, get_subscription_plan
 from datetime import datetime
 from pytz import timezone
 
@@ -67,5 +67,25 @@ def test_chris(self):
         result = chris()
         self.assertEqual(result, '')
 
+    def test_get_subscription_plan_monthly(self):
+        result = get_subscription_plan('monthly')
+        self.assertEqual(result, 'monthly')
+
+    def test_get_subscription_plan_annual(self):
+        result = get_subscription_plan('annual')
+        self.assertEqual(result, 'annual')
+
+    def test_get_subscription_plan_invalid_string(self):
+        result = get_subscription_plan('quarterly')
+        self.assertEqual(result, 'annual')
+
+    def test_get_subscription_plan_none(self):
+        result = get_subscription_plan(None)
+        self.assertEqual(result, 'annual')
+
+    def test_get_subscription_plan_empty_string(self):
+        result = get_subscription_plan('')
+        self.assertEqual(result, 'annual')
+
 if __name__ == '__main__':
     unittest.main()