[ADD]Odoo SHELL helpers on GOLEM too

odoo/golem/shell_helpers.py

@@ -0,0 +1,46 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2021 Fabien Bourgeois <fabien@yaltik.com>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+""" SHELL helpers """
+
+from imp import reload
+import unittest
+
+def run_test(module, test_names='ALL', test_class_name='DEFAULT'):
+    """ Helper to allow testing of single method or all from TestCase
+    Takes module as python module, and names as strings
+    module is automatically reloaded for recent updates in shell
+    test_names can be only one string or an array of strings
+    test_class_name can be automatically found from module name """
+    module = reload(module)
+    suite = unittest.TestSuite()
+    if test_class_name == 'DEFAULT':
+        module_name = module.__name__.split('.')[-1]
+        test_class_name = ''.join(map(str.capitalize, module_name.split('_')))
+    if test_class_name not in dir(module):
+        raise Exception('Generated class name (%s) not found in module, please '
+                        'specify' % test_class_name)
+    if test_names == 'ALL':
+        testCase = getattr(module, test_class_name)
+        tests = unittest.TestLoader().loadTestsFromTestCase(testCase)
+        suite.addTests(tests)
+    else:
+        if isinstance(test_names, str):
+            test_names = [test_names]
+        add = lambda test_name: suite.addTest(getattr(module, test_class_name)(test_name))
+        map(add, test_names)
+    unittest.TextTestRunner().run(suite)