def test_add_record(self, zps, zpm):
bp = Provider('example.com')
bp.reload_config = Mock()
bp.add_record({'rr': 'A', 'alias': 'foo', 'addr': '127.0.0.1'})
zps.assert_called_once_with()
zpm.assert_called_once_with({
'alias': 'foo',
'addr': '127.0.0.1',
'rr': 'A'
})
bp.reload_config.assert_called_once_with()
def test_first_setup(self, ugm, spcom):
ugm.return_value = '10.0.0.1'
bp = Provider('example.com')
parser = MagicMock()
bp.first_setup(parser)
ugm.assert_called_once_with('public-address')
parser.dict_to_zone.assert_called_with({
'rr': 'A',
'alias': 'ns',
'addr': '10.0.0.1',
'ttl': 300
})
