two version of R2R are hereHEADmaster

1 files changed, 27 insertions, 0 deletions

diff --git a/.venv/lib/python3.12/site-packages/test/unit/test_sendgrid.py b/.venv/lib/python3.12/site-packages/test/unit/test_sendgrid.py
new file mode 100644
index 00000000..328d978a
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/test/unit/test_sendgrid.py
@@ -0,0 +1,27 @@
+import unittest
+import sendgrid
+
+class UnitTests(unittest.TestCase):
+    def test_host_with_no_region(self):
+        sg = sendgrid.SendGridAPIClient(api_key='MY_API_KEY')
+        self.assertEqual("https://api.sendgrid.com",sg.client.host)
+
+    def test_host_with_eu_region(self):
+        sg = sendgrid.SendGridAPIClient(api_key='MY_API_KEY')
+        sg.set_sendgrid_data_residency("eu")
+        self.assertEqual("https://api.eu.sendgrid.com",sg.client.host)
+
+    def test_host_with_global_region(self):
+        sg = sendgrid.SendGridAPIClient(api_key='MY_API_KEY')
+        sg.set_sendgrid_data_residency("global")
+        self.assertEqual("https://api.sendgrid.com",sg.client.host)
+
+    def test_with_region_is_none(self):
+        sg = sendgrid.SendGridAPIClient(api_key='MY_API_KEY')
+        with self.assertRaises(ValueError):
+            sg.set_sendgrid_data_residency(None)
+
+    def test_with_region_is_invalid(self):
+        sg = sendgrid.SendGridAPIClient(api_key='MY_API_KEY')
+        with self.assertRaises(ValueError):
+            sg.set_sendgrid_data_residency("abc")
\ No newline at end of file