1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
import pytest
# Copyright (c) 2017-2019 David Steele <dsteele@gmail.com>
#
# SPDX-License-Identifier: GPL-2.0-or-later
# License-Filename: LICENSE
from comitup import mdns
from mock import Mock, patch
@pytest.fixture()
def avahi_fxt(monkeypatch, request):
monkeypatch.setattr("comitup.mdns.dbus.Interface", Mock())
monkeypatch.setattr('comitup.mdns.dbus.SystemBus', Mock())
monkeypatch.setattr('comitup.mdns.log', Mock())
save_group = mdns.group
mdns.group = Mock()
def fin():
mdns.group = save_group
request.addfinalizer(fin)
return None
def test_avahi_null(avahi_fxt):
pass
def test_avahi_establish_group(avahi_fxt):
old_group = mdns.group
mdns.establish_group()
assert mdns.group != old_group
def test_avahi_make_a_record(avahi_fxt):
mdns.make_a_record('host', '1', '1.2.3.4')
assert mdns.group.AddRecord.called
def test_avahi_add_service(avahi_fxt):
mdns.add_service('host', '1', '1.2.3.4')
assert mdns.group.AddService.called
@patch('comitup.mdns.establish_group', Mock())
def test_avahi_clear_entries(avahi_fxt):
isempty = Mock(return_value=False)
mdns.group = Mock()
mdns.group.IsEmpty = isempty
mdns.clear_entries()
assert isempty.called
assert mdns.group.Reset.called
assert mdns.establish_group.called
assert not mdns.log.called
@patch('comitup.nm.get_devices', Mock(return_value=[]))
def test_avahi_add_hosts(avahi_fxt):
mdns.add_hosts(['host1', 'host2'])
assert mdns.group.Commit.called
@pytest.mark.parametrize("dns_in, dns_out", (
("a.b.c", "a.b.c".encode()),
("A.B.C", "A.B.C".encode()),
("a..b", "a.b".encode()),
("a.b.", "a.b".encode()),
))
def test_avahi_encode_dns(dns_in, dns_out):
assert dns_out == mdns.encode_dns(dns_in)
@patch('comitup.mdns.log.warn')
def test_avahi_clear_fail(warn, avahi_fxt):
mdns.group = None
mdns.clear_entries()
|