def test_read_local_zones(self, mopen):
seed_zone = """zone "255.in-addr.arpa" {
type master;
file "/etc/bind/db.255";
};"""
mopen.return_value.__enter__ = lambda s: s
mopen.return_value.__exit__ = Mock()
mopen.return_value.readlines.return_value = seed_zone.split('\n')
zp = ZoneParser('example.com')
self.assertEqual(zp.read_local_zones(), seed_zone.split('\n'))
def test_read_local_zones(self, mopen):
seed_zone = """zone "255.in-addr.arpa" {
type master;
file "/etc/bind/db.255";
};"""
mopen.return_value.__enter__ = lambda s: s
mopen.return_value.__exit__ = Mock()
mopen.return_value.readlines.return_value = seed_zone.split('\n')
zp = ZoneParser('example.com')
self.assertEqual(zp.read_local_zones(), seed_zone.split('\n'))
