X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=test%2Fenablecfg.py;h=811d9ba6f86e339c6635542bb21e38d9c3445740;hb=fb5863ca8d0db3ff2e84721f7c902b031157ebb0;hp=22f11dde6b989017e74525a966550cbcb631ba07;hpb=695d2dfa32c5b1ac92c13238323b5041ff567b58;p=bertos.git

diff --git a/test/enablecfg.py b/test/enablecfg.py
index 22f11dde..811d9ba6 100755
--- a/test/enablecfg.py
+++ b/test/enablecfg.py
@@ -46,9 +46,24 @@ whitelist = [
 	'CONFIG_KERN_HEAP',
 	'CONFIG_KERN_PREEMPT',
 	'RAMP_USE_FLOATING_POINT',
+	'CONFIG_SER_HWHANDSHAKE',
+	'CONFIG_KDEBUG_PORT',
 ]
 
-pattern = r'\s*#define\s+(\w+)\s+(0)\W?'
+tests = [
+	'#define T_0 0\n',
+	'#define T_1 0\n',
+	'    #define    	T_2    	0		\n',
+	'#define T_3 0    /* */\n',
+	'#define T_4 0/* */\n',
+]
+no_tests = [
+	'#define T_1 0A\n',
+	'#define T_1 0x0A/* */\n',
+	'#define T_1 0UL /* */\n',
+	'#define T_1 0UL/* */\n',
+]
+pattern = r'\s*#define\s+(\w+)\s+(0)(?:\s+|/|$)'
 
 def f(match):
 	if match.group(1) in whitelist:
@@ -59,7 +74,14 @@ def f(match):
 		ex = match.end(2) - match.end(0)
 		return data[:sx] + '1' + data[ex:]
 
-data = file(sys.argv[1]).read()
-data, count = re.subn(pattern, f, data)
-if count:
-	file(sys.argv[1], 'w').write(data)
+if len(sys.argv) == 1:
+	for t in tests:
+		print t, re.subn(pattern, f, t)[0]
+	print '-' * 42
+	for t in no_tests:
+		print t, re.subn(pattern, f, t)[0]
+else:
+	data = file(sys.argv[1]).read()
+	data, count = re.subn(pattern, f, data)
+	if count:
+		file(sys.argv[1], 'w').write(data)