test_sendgrid.py

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")