From: Simon Glass <sjg@chromium.org>
Date: Sat, 18 Dec 2021 15:09:45 +0000 (-0700)
Subject: moveconfig: Allow adding unit tests
X-Git-Url: http://git.dujemihanovic.xyz/?a=commitdiff_plain;h=84067a5890f186a925dcf89f8402abaf12f09b45;p=u-boot.git

moveconfig: Allow adding unit tests

Add a -t option to run unit tests in this program. So far, there is none.

Signed-off-by: Simon Glass <sjg@chromium.org>
---

diff --git a/tools/moveconfig.py b/tools/moveconfig.py
index 369589727c..71a7736ca6 100755
--- a/tools/moveconfig.py
+++ b/tools/moveconfig.py
@@ -14,6 +14,7 @@ import asteval
 import collections
 import copy
 import difflib
+import doctest
 import filecmp
 import fnmatch
 import glob
@@ -28,6 +29,7 @@ import sys
 import tempfile
 import threading
 import time
+import unittest
 
 from buildman import bsettings
 from buildman import kconfiglib
@@ -1621,6 +1623,13 @@ def main():
 
     (options, configs) = parser.parse_args()
 
+    if options.test:
+        sys.argv = [sys.argv[0]]
+        fail, count = doctest.testmod()
+        if fail:
+            return 1
+        unittest.main()
+
     if len(configs) == 0 and not any((options.force_sync, options.build_db,
                                       options.imply)):
         parser.print_usage()