Пример #1
0
 def test_save(self):
     os.mkdir("savedir")
     self.a = RuntimeAuthority("foo", "savedir")
     self.a.set_record("baz", "A", {'address': "192.168.1.2"}, True)
     self.a.set_record("bar", "A", {'address': "192.168.1.1"}, True)
     self.assertEqual(json.load(open("savedir/foo")), [{
         "bar": {
             "type": "A",
             "address": "192.168.1.1",
         }
     }, {
         "baz": {
             "type": "A",
             "address": "192.168.1.2",
         }
     }])
Пример #2
0
 def test_save(self):
     os.mkdir("savedir")
     self.a = RuntimeAuthority("foo", "savedir")
     self.a.set_record("baz", "A", {'address': "192.168.1.2"}, True)
     self.a.set_record("bar", "A", {'address': "192.168.1.1"}, True)
     self.assertEqual(json.load(open("savedir/foo")), [
         {"bar": {
             "type": "A",
             "address": "192.168.1.1",
         }},
         {"baz": {
             "type": "A",
             "address": "192.168.1.2",
         }}
     ])
Пример #3
0
 def test_load(self):
     os.mkdir("savedir")
     json.dump((
         {
             "bar": {
                 "type": "A",
                 "address": "192.168.1.1",
             }
         },
         {
             "baz": {
                 "type": "A",
                 "address": "192.168.1.2",
             }
         },
     ), open("savedir/foo", "w"))
     self.a = RuntimeAuthority("foo", "savedir")
     self.assertEqual(
         self.a.records, {
             "bar.foo": [Record_A(address="192.168.1.1")],
             "baz.foo": [Record_A(address="192.168.1.2")],
         })
     os.unlink("savedir/foo")
     os.rmdir("savedir")
Пример #4
0
 def test_load(self):
     os.mkdir("savedir")
     json.dump((
         {"bar": {
             "type": "A",
             "address": "192.168.1.1",
         }},
         {"baz": {
             "type": "A",
             "address": "192.168.1.2",
         }},
     ), open("savedir/foo", "w"))
     self.a = RuntimeAuthority("foo", "savedir")
     self.assertEqual(self.a.records, {
         "bar.foo": [Record_A(address="192.168.1.1")],
         "baz.foo": [Record_A(address="192.168.1.2")],
     })
     os.unlink("savedir/foo")
     os.rmdir("savedir")
Пример #5
0
 def setUp(self):
     self.a = RuntimeAuthority("foo", None)
Пример #6
0
class TestRuntimeAuthority(unittest.TestCase):

    def setUp(self):
        self.a = RuntimeAuthority("foo", None)

    def test_a_records(self):
        foo_value = MagicMock(Record_A, autospec=True)
        bar_value = MagicMock(Record_A, autospec=True)
        foo_value.TYPE = 1
        bar_value.TYPE = 1
        foo_value.name = "foo"
        bar_value.name = "bar"
        foo_value.ttl = None
        bar_value.ttl = None
        foo_value.compareAttributes = ('address', 'ttl')
        bar_value.compareAttributes = ('address', 'ttl')
        foo_value.dottedQuad.return_value = "192.168.0.1"
        bar_value.dottedQuad.return_value = "192.168.0.2"
        self.a.records = {
            "foo.foo": [foo_value],
            "bar.foo": [bar_value],
        }
        rv = self.a.get_records_by_type("A")
        self.assertEqual(sorted(rv), [
            ("A", "bar.foo", ["192.168.0.2"]),
            ("A", "foo.foo", ["192.168.0.1"]),
        ])

    def test_txt_records(self):
        pass

    def test_cname_records(self):
        pass

    def test_conflict_rules(self):
        pass

    def test_load(self):
        os.mkdir("savedir")
        json.dump((
            {"bar": {
                "type": "A",
                "address": "192.168.1.1",
            }},
            {"baz": {
                "type": "A",
                "address": "192.168.1.2",
            }},
        ), open("savedir/foo", "w"))
        self.a = RuntimeAuthority("foo", "savedir")
        self.assertEqual(self.a.records, {
            "bar.foo": [Record_A(address="192.168.1.1")],
            "baz.foo": [Record_A(address="192.168.1.2")],
        })
        os.unlink("savedir/foo")
        os.rmdir("savedir")

    def test_save(self):
        os.mkdir("savedir")
        self.a = RuntimeAuthority("foo", "savedir")
        self.a.set_record("baz", "A", {'address': "192.168.1.2"}, True)
        self.a.set_record("bar", "A", {'address': "192.168.1.1"}, True)
        self.assertEqual(json.load(open("savedir/foo")), [
            {"bar": {
                "type": "A",
                "address": "192.168.1.1",
            }},
            {"baz": {
                "type": "A",
                "address": "192.168.1.2",
            }}
        ])
Пример #7
0
 def setUp(self):
     self.a = RuntimeAuthority("foo", None)
Пример #8
0
class TestRuntimeAuthority(unittest.TestCase):
    def setUp(self):
        self.a = RuntimeAuthority("foo", None)

    def test_a_records(self):
        foo_value = MagicMock(Record_A, autospec=True)
        bar_value = MagicMock(Record_A, autospec=True)
        foo_value.TYPE = 1
        bar_value.TYPE = 1
        foo_value.name = "foo"
        bar_value.name = "bar"
        foo_value.ttl = None
        bar_value.ttl = None
        foo_value.compareAttributes = ('address', 'ttl')
        bar_value.compareAttributes = ('address', 'ttl')
        foo_value.dottedQuad.return_value = "192.168.0.1"
        bar_value.dottedQuad.return_value = "192.168.0.2"
        self.a.records = {
            "foo.foo": [foo_value],
            "bar.foo": [bar_value],
        }
        rv = self.a.get_records_by_type("A")
        self.assertEqual(sorted(rv), [
            ("A", "bar.foo", ["192.168.0.2"]),
            ("A", "foo.foo", ["192.168.0.1"]),
        ])

    def test_txt_records(self):
        pass

    def test_cname_records(self):
        pass

    def test_conflict_rules(self):
        pass

    def test_load(self):
        os.mkdir("savedir")
        json.dump((
            {
                "bar": {
                    "type": "A",
                    "address": "192.168.1.1",
                }
            },
            {
                "baz": {
                    "type": "A",
                    "address": "192.168.1.2",
                }
            },
        ), open("savedir/foo", "w"))
        self.a = RuntimeAuthority("foo", "savedir")
        self.assertEqual(
            self.a.records, {
                "bar.foo": [Record_A(address="192.168.1.1")],
                "baz.foo": [Record_A(address="192.168.1.2")],
            })
        os.unlink("savedir/foo")
        os.rmdir("savedir")

    def test_save(self):
        os.mkdir("savedir")
        self.a = RuntimeAuthority("foo", "savedir")
        self.a.set_record("baz", "A", {'address': "192.168.1.2"}, True)
        self.a.set_record("bar", "A", {'address': "192.168.1.1"}, True)
        self.assertEqual(json.load(open("savedir/foo")), [{
            "bar": {
                "type": "A",
                "address": "192.168.1.1",
            }
        }, {
            "baz": {
                "type": "A",
                "address": "192.168.1.2",
            }
        }])