def test_route1(self): r = Router("a.gbl.", default_gateway="b.gbl.") r.add_route("com.", "d.gbl.") r.add_route("se.", "d.gbl.") self.assertEquals("d.gbl.", r.get_route("test.com.")) self.assertEquals("d.gbl.", r.get_route("test.se."))