def test_to_file(self, tm, mopen):
mopen.return_value.__enter__ = lambda s: s
mopen.return_value.__exit__ = Mock()
mopen.return_value.write = Mock()
z = Zone()
z.read_template = Mock()
z.read_template.return_value = "hi {{data}}"
z.to_file()
z.read_template.assert_called_once()
mopen.assert_called_with('/etc/bind/db.example.com', 'w')
tm.assert_called_with(
data={
'SOA': [],
'AAAA': [],
'TXT': [],
'PTR': [],
'SPF': [],
'A': [],
'CERT': [],
'CNAME': [],
'SRV': [],
'CAA': [],
'NS': [],
'NAPTR': []
})
def test_to_file(self, tm, mopen):
mopen.return_value.__enter__ = lambda s: s
mopen.return_value.__exit__ = Mock()
mopen.return_value.write = Mock()
z = Zone()
z.read_template = Mock()
z.read_template.return_value = "hi {{data}}"
z.to_file()
z.read_template.assert_called_once()
mopen.assert_called_with('/etc/bind/db.example.com', 'w')
tm.assert_called_with(data={'SOA': [], 'AAAA': [], 'TXT': [],
'PTR': [], 'SPF': [], 'A': [], 'CERT': [], 'CNAME': [], 'SRV': [],
'CAA': [], 'NS': [], 'NAPTR': []})
def test_read_template(self, mopen):
mopen.return_value.__enter__ = lambda s: s
mopen.return_value.__exit__ = Mock()
mopen.return_value.read.return_value = "{{foo}}"
z = Zone()
self.assertEqual(z.read_template(), "{{foo}}")
