def test_dict_to_zone(self): zp = ZoneParser('example.com') zp.update_ns = Mock() zp.update_soa = Mock() zp.update_cname = Mock() zp.update_a = Mock() zp.dict_to_zone({'rr': 'NS'}) zp.update_ns.assert_called_with({'rr': 'NS'}) zp.dict_to_zone({'rr': 'SOA'}) zp.update_soa.assert_called_with({'rr': 'SOA'}) zp.dict_to_zone({'rr': 'CNAME'}) zp.update_cname.assert_called_with({'rr': 'CNAME'}) zp.dict_to_zone({'rr': 'A'}) zp.update_a.assert_called_with({'rr': 'A'}) zp.dict_to_zone({'rr': 'NOPE'})