@debug__pydev_pd_arg: apply where missing

1 files changed, 11 insertions, 11 deletions

diff --git a/src/server-tests/test_rzdoc.py b/src/server-tests/test_rzdoc.py
index 89b6c9f5..01e15400 100644
--- a/src/server-tests/test_rzdoc.py
+++ b/src/server-tests/test_rzdoc.py
@@ -1,9 +1,12 @@
 import logging
 import unittest
-from rz_server import Config
+
 import db_controller
-from rz_kernel import RZ_Kernel
 import neo4j_test_util
+from rz_kernel import RZ_Kernel
+from rz_server import Config
+from test_util__pydev import debug__pydev_pd_arg
+
 
 class TestRZDoc(unittest.TestCase):
 
@@ -24,7 +27,7 @@ class TestRZDoc(unittest.TestCase):
         rzdoc_name = test_label
         lookup_ret = self.kernel.rzdoc__lookup_by_name(rzdoc_name)
         self.assertIsNone(lookup_ret)
-        
+
         ret_create = self.kernel.rzdoc__create(rzdoc_name)
         lookup_ret = self.kernel.rzdoc__lookup_by_name(ret_create.name)
         self.assertTrue(None != lookup_ret and lookup_ret.name == rzdoc_name)
@@ -35,12 +38,9 @@ class TestRZDoc(unittest.TestCase):
 
     def tearDown(self): pass
 
-if __name__ == "__main__":
-
-    try:  # enable pydev remote debugging
-        import pydevd
-        pydevd.settrace()
-    except ImportError:
-        pass
-
+@debug__pydev_pd_arg
+def main():
     unittest.main(defaultTest='TestRZDoc.test_rzdoc_lifcycle', verbosity=2)
+
+if __name__ == "__main__":
+    main()