import samba.tests
from samba.compat import StringIO
+from samba.compat import get_string
from samba.netcmd.main import cmd_sambatool
from samba.credentials import Credentials
from samba.auth import system_session
super(SambaDnsUpdateTests, self).setUp()
try:
out = self.check_output("samba_dnsupdate --verbose")
- self.assertTrue("Looking for DNS entry" in out, out)
+ self.assertTrue(b"Looking for DNS entry" in out, out)
except samba.tests.BlackboxProcessError:
pass
out = self.check_output("samba_dnsupdate --verbose")
except samba.tests.BlackboxProcessError as e:
self.fail("Error calling samba_dnsupdate: %s" % e)
- self.assertTrue("No DNS updates needed" in out, out)
+ self.assertTrue(b"No DNS updates needed" in out, out)
def test_samba_dnsupate_set_ip(self):
try:
out = self.check_output("samba_dnsupdate --verbose --current-ip=10.0.0.1")
- self.assertTrue(" DNS updates and" in out, out)
- self.assertTrue(" DNS deletes needed" in out, out)
+ self.assertTrue(b" DNS updates and" in out, out)
+ self.assertTrue(b" DNS deletes needed" in out, out)
except samba.tests.BlackboxProcessError:
pass
except samba.tests.BlackboxProcessError as e:
self.fail("Error calling samba_dnsupdate: %s" % e)
- self.assertTrue("No DNS updates needed" in out, out)
+ self.assertTrue(b"No DNS updates needed" in out, out)
try:
rpc_out = self.check_output("samba_dnsupdate --verbose --use-samba-tool --rpc-server-ip=%s" % self.server_ip)
except samba.tests.BlackboxProcessError as e:
self.fail("Error calling samba_dnsupdate: %s" % e)
- self.assertTrue(" DNS updates and" in rpc_out, rpc_out)
- self.assertTrue(" DNS deletes needed" in rpc_out, rpc_out)
+ self.assertTrue(b" DNS updates and" in rpc_out, rpc_out)
+ self.assertTrue(b" DNS deletes needed" in rpc_out, rpc_out)
out = self.check_output("samba_dnsupdate --verbose")
- self.assertTrue("No DNS updates needed" in out, out + rpc_out)
+ self.assertTrue(b"No DNS updates needed" in out, out + rpc_out)
def test_add_new_uncovered_site(self):
name = 'sites'
ldb.FLAG_MOD_ADD, "siteList")
dns_c = "samba_dnsupdate --verbose --use-file={0}".format(tmp_uc)
- out = self.check_output(dns_c)
+ out = get_string(self.check_output(dns_c))
self.assertFalse(site_name.lower() in out, out)
self.samdb.modify(m)
shutil.copyfile(uc_fn, tmp_uc)
- out = self.check_output(dns_c)
+ out = get_string(self.check_output(dns_c))
self.assertFalse("No DNS updates needed" in out, out)
self.assertTrue(site_name.lower() in out, out)