From 390d6273096fe0e435a0d145e6a785f38372ba67 Mon Sep 17 00:00:00 2001
From: fujiwarat <takao.fujiwara1@gmail.com>
Date: Thu, 2 Jul 2020 14:47:39 +0900
Subject: tests: Enable pycotap

---
 tests/anthytest.py | 20 +++++++++++++++++---
 1 file changed, 17 insertions(+), 3 deletions(-)

diff --git a/tests/anthytest.py b/tests/anthytest.py
index 11d0e2b..03a7c31 100755
--- a/tests/anthytest.py
+++ b/tests/anthytest.py
@@ -20,10 +20,23 @@ import sys
 import subprocess
 import unittest
 
+TAP_MODULE_NONE, \
+TAP_MODULE_TAPPY, \
+TAP_MODULE_PYCOTAP = list(range(3))
+
+tap_module = TAP_MODULE_NONE
+
 try:
     from tap import TAPTestRunner
-except ModuleNotFoundError as err:
-    print('Ignore tap module: %s' % str(err))
+    tap_module = TAP_MODULE_TAPPY
+    print('Load tappy')
+except ModuleNotFoundError:
+    try:
+        from pycotap import TAPTestRunner
+        tap_module = TAP_MODULE_PYCOTAP
+        print('Load pycotap')
+    except ModuleNotFoundError as err:
+        print('Ignore tap module: %s' % str(err))
 
 PY3K = sys.version_info >= (3, 0)
 DONE_EXIT = True
@@ -344,7 +357,8 @@ def main():
     if args.tap:
         loader = unittest.TestLoader()
         runner = TAPTestRunner()
-        runner.set_stream(True)
+        if tap_module == TAP_MODULE_TAPPY:
+            runner.set_stream(True)
         unittest.main(testRunner=runner, testLoader=loader)
     else:
         unittest.main()
-- 
cgit v1.2.1