def test_save(self, osrm, fwm): zp = ZoneParser('example.com') zp.passes_validation = Mock() zp.passes_validation.return_value = True zp.add_to_local_zones = Mock() zp.save() osrm.assert_called_with('/etc/bind/db.example.com.proposed') fwm.assert_called_with('/etc/bind/db.example.com') zp.add_to_local_zones.assert_called_once()
def test_save(self, fwm, mopen): mopen.return_value.__enter__ = lambda s: s mopen.return_value.__exit__ = Mock() mopen.return_value.readlines.return_value = """ zone "localhost" { type master; file "/etc/bind/db.local"; }; zone "127.in-addr.arpa" { type master; file "/etc/bind/db.127"; };""".split('\n') zp = ZoneParser('example.com') zp.save() fwm.assert_called_with('/etc/bind/db.example.com')