def test_autodiscover_path_1_2_3_4_invalid_srv(self, m): # Test steps 1 -> 2 -> 3 -> 4 -> invalid SRV URL d = Autodiscovery(email=self.account.primary_smtp_address, credentials=self.account.protocol.credentials) m.post(self.dummy_ad_endpoint, status_code=501) m.post(f"https://autodiscover.{self.domain}/Autodiscover/Autodiscover.xml", status_code=501) m.get(f"http://autodiscover.{self.domain}/Autodiscover/Autodiscover.xml", status_code=200) tmp = d._get_srv_records d._get_srv_records = Mock(return_value=[SrvRecord(1, 1, 443, get_random_hostname())]) try: with self.assertRaises(AutoDiscoverFailed): # Fails in step 4 with invalid response ad_response, _ = d.discover() finally: d._get_srv_records = tmp
def test_get_srv_records(self): from exchangelib.autodiscover.discovery import SrvRecord ad = Autodiscovery('*****@*****.**') # Unknown domain self.assertEqual(ad._get_srv_records('example.XXXXX'), []) # No SRV record self.assertEqual(ad._get_srv_records('example.com'), []) # Finding a real server that has a correct SRV record is not easy. Mock it _orig = dns.resolver.Resolver class _Mock1: @staticmethod def resolve(hostname, cat): class A: @staticmethod def to_text(): # Return a valid record return '1 2 3 example.com.' return [A()] dns.resolver.Resolver = _Mock1 del ad.resolver # Test a valid record self.assertEqual( ad._get_srv_records('example.com.'), [SrvRecord(priority=1, weight=2, port=3, srv='example.com')]) class _Mock2: @staticmethod def resolve(hostname, cat): class A: @staticmethod def to_text(): # Return malformed data return 'XXXXXXX' return [A()] dns.resolver.Resolver = _Mock2 del ad.resolver # Test an invalid record self.assertEqual(ad._get_srv_records('example.com'), []) dns.resolver.Resolver = _orig del ad.resolver
def test_autodiscover_path_1_2_3_4_valid_srv_valid_response(self, m): # Test steps 1 -> 2 -> 3 -> 4 -> 5 d = Autodiscovery(email=self.account.primary_smtp_address, credentials=self.account.protocol.credentials) redirect_srv = "httpbin.org" ews_url = f"https://{redirect_srv}/EWS/Exchange.asmx" redirect_email = f"john@redirected.{redirect_srv}" m.post(self.dummy_ad_endpoint, status_code=501) m.post(f"https://autodiscover.{self.domain}/Autodiscover/Autodiscover.xml", status_code=501) m.get(f"http://autodiscover.{self.domain}/Autodiscover/Autodiscover.xml", status_code=200) m.head(f"https://{redirect_srv}/Autodiscover/Autodiscover.xml", status_code=200) m.post( f"https://{redirect_srv}/Autodiscover/Autodiscover.xml", status_code=200, content=self.settings_xml(redirect_email, ews_url), ) tmp = d._get_srv_records d._get_srv_records = Mock(return_value=[SrvRecord(1, 1, 443, redirect_srv)]) try: ad_response, _ = d.discover() self.assertEqual(ad_response.autodiscover_smtp_address, redirect_email) self.assertEqual(ad_response.protocol.ews_url, ews_url) finally: d._get_srv_records = tmp