from unittest import TestCase
from yawast.scanner.plugins.dns import basic


class TestGetMx(TestCase):
    def test_get_mx(self):
        recs = basic.get_mx("adamcaudill.com")

        self.assertTrue(len(recs) > 0)

        for rec in recs:
            if rec[0].startswith("aspmx4"):
                self.assertEqual("aspmx4.googlemail.com.", rec[0])

from unittest import TestCase
from yawast.scanner.plugins.dns import basic


class TestGetMx(TestCase):
    def test_get_mx(self):
        recs = basic.get_mx("adamcaudill.com")

        self.assertTrue(len(recs) > 0)

        for rec in recs:
            if rec[0].startswith("aspmx4"):
                self.assertEqual("aspmx4.googlemail.com.", rec[0])
