sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-03-14 00:58:38 +03:00
Code

s4/tests/samba-tool drs showrepl: test NO_COLOR and --color variants

Browse Source 
"--color variants" meaning --color=always instead of --color=yes, etc.

Signed-off-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Joseph Sutton <josephsutton@catalyst.net.nz>
This commit is contained in:
Douglas Bagnall 2022-08-17 14:28:15 +12:00 committed by Douglas Bagnall
parent 7d178ab95b
commit c61e8cdefc
1 changed files with 30 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
source4/torture/drs/python/samba_tool_drs_showrepl.py
View File

@ -21,6 +21,7 @@ import samba.tests
import drs_base
from samba.dcerpc import drsuapi
from samba import drs_utils
import os
import re
import json
import ldb
@ -229,6 +230,7 @@ class SambaToolDrsShowReplTests(drs_base.DrsBaseTestCase):
self._enable_all_repl(self.dc1)
self._force_all_reps(samdb1, self.dc1, 'inbound')
self._force_all_reps(samdb1, self.dc1, 'outbound')
old_no_color = os.environ.get('NO_COLOR')
all_good_green = "\033[1;32m[ALL GOOD]\033[0m\n"
all_good = "[ALL GOOD]\n"
@ -254,13 +256,40 @@ class SambaToolDrsShowReplTests(drs_base.DrsBaseTestCase):
self.assertStringsEqual(out, all_good)
out = self.check_output("samba-tool drs showrepl --pull-summary -v "
"--color=yes %s %s" %
"--color=always %s %s" %
(self.dc1, self.cmdline_creds))
out = get_string(out)
self.assertStringsEqual(out, all_good_green)
out = self.check_output("samba-tool drs showrepl --pull-summary -v "
"--color=never %s %s" %
(self.dc1, self.cmdline_creds))
out = get_string(out)
self.assertStringsEqual(out, all_good)
os.environ['NO_COLOR'] = 'bean'
out = self.check_output("samba-tool drs showrepl --pull-summary -v "
"--color=auto %s %s" %
(self.dc1, self.cmdline_creds))
out = get_string(out)
self.assertStringsEqual(out, all_good)
os.environ['NO_COLOR'] = ''
out = self.check_output("samba-tool drs showrepl --pull-summary -v "
"--color=auto %s %s" %
(self.dc1, self.cmdline_creds))
out = get_string(out)
self.assertStringsEqual(out, all_good_green)
except samba.tests.BlackboxProcessError as e:
self.fail(str(e))
finally:
if old_no_color is None:
os.environ.pop('NO_COLOR', None)
else:
os.environ['NO_COLOR'] = old_no_color
def test_samba_tool_showrepl_summary_forced_failure(self):