def test_normalize_contents(self, mopen, opem, spm, idm): mopen.return_value.__enter__ = lambda s: s mopen.return_value.__exit__ = Mock() mopen.return_value.readlines.return_value = self.ez idm.return_value = 'ABC123' opem.return_value = True zp = ZoneParser('example.com') zp.normalize_contents() spm.assert_called_with(['named-checkzone', '-o', '/tmp/ABC123', 'example.com', '/etc/bind/db.example.com'])
def test_from_file_exception(self, mopen): mopen.return_value.__enter__ = lambda s: s mopen.return_value.__exit__ = Mock() mopen.return_value.readlines = Mock(side_effect=OSError('Intentional')) zp = ZoneParser('foo.com') zp.normalize_contents = Mock() zp.normalize_contents.return_value = self.ez self.assertEqual(zp.from_file(), [])
def test_from_file(self, mopen): mopen.return_value.__enter__ = lambda s: s mopen.return_value.__exit__ = Mock() mopen.return_value.readlines.return_value = self.ez zp = ZoneParser('ns') zp.normalize_contents = Mock() zp.normalize_contents.return_value = self.ez self.assertEqual(zp.from_file(), self.ez)