summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--python/samba/subunit/__init__.py5
-rw-r--r--selftest/subunithelper.py2
2 files changed, 6 insertions, 1 deletions
diff --git a/python/samba/subunit/__init__.py b/python/samba/subunit/__init__.py
index ab55ace104..cdc593398a 100644
--- a/python/samba/subunit/__init__.py
+++ b/python/samba/subunit/__init__.py
@@ -17,6 +17,11 @@
"""Subunit test protocol."""
+import samba
+
+samba.ensure_third_party_module("iso8601", "pyiso8601")
+import iso8601
+
import unittest
diff --git a/selftest/subunithelper.py b/selftest/subunithelper.py
index a1d49fd51c..5308e99a24 100644
--- a/selftest/subunithelper.py
+++ b/selftest/subunithelper.py
@@ -21,7 +21,7 @@ import re
import sys
from samba import subunit
from samba.subunit.run import TestProtocolClient
-import iso8601
+from samba.subunit import iso8601
import unittest
VALID_RESULTS = ['success', 'successful', 'failure', 'fail', 'skip', 'knownfail', 'error', 'xfail', 'skip-testsuite', 'testsuite-failure', 'testsuite-xfail', 'testsuite-success', 'testsuite-error', 'uxsuccess', 'testsuite-uxsuccess']