1 # -*- coding: utf-8 -*-
2 # Unix SMB/CIFS implementation. Tests for netbios py module
3 # Copyright (C) Noel Power <noel.power@suse.com> 2018
5 # This program is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
15 # You should have received a copy of the GNU General Public License
16 # along with this program. If not, see <http://www.gnu.org/licenses/>.
22 from samba import netbios
25 class NetBiosTests(samba.tests.TestCase):
27 super(NetBiosTests, self).setUp()
28 self.n = netbios.Node()
29 self.ifc = os.environ["SERVER_IP"]
30 self.dc = os.environ["DC_NETBIOSNAME"]
33 super(NetBiosTests, self).tearDown()
35 def test_query_name(self):
36 (reply_from, names, addresses) = self.n.query_name(self.dc, self.ifc, timeout=4)
37 assert reply_from == self.ifc
38 assert names[0] == self.dc
39 assert addresses[0] == self.ifc
41 def test_name_status(self):
42 (reply_from, name, name_list) = self.n.name_status(self.dc, self.ifc, timeout=4)
43 assert reply_from == self.ifc
44 assert name[0] == self.dc
45 assert len(name_list) > 0
47 def test_register_name(self):
49 (reply_from, name, reply_address, code) = self.n.register_name((self.dc, 0x20), address, self.ifc, multi_homed=False, timeout=4)
50 assert reply_from == self.ifc
51 assert name[0] == self.dc
52 assert reply_address == address
55 def disabled_test_refresh(self):
56 # can't get the below test to work, disabling
58 res = self.n.refresh_name((self.dc, 0x20), address, self.ifc, timeout=10)