From 5a8858aab9df94c4b2ab2117e911ca93bfda9bc0 Mon Sep 17 00:00:00 2001
From: Henddher Pedroza <henddher.pedroza@gmail.com>
Date: Sat, 27 Oct 2018 18:20:54 -0500
Subject: [PATCH] Tests for protfunc parser and _RE_DBREF

---
 evennia/prototypes/tests.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/evennia/prototypes/tests.py b/evennia/prototypes/tests.py
index 103a5d9404..fe75ccf4ec 100644
--- a/evennia/prototypes/tests.py
+++ b/evennia/prototypes/tests.py
@@ -262,6 +262,25 @@ class TestProtFuncs(EvenniaTest):
                      "prototype_desc": "testing prot",
                      "key": "ExampleObj"}
 
+    def test_RE_DBREF(self):
+        def check_RE_DBREF(value, expected_value):
+            try:
+                result = (
+                    protlib._RE_DBREF.match(value),
+                    protlib._RE_DBREF.search(value),
+                    protlib._RE_DBREF.sub("$obj(\\1)", value)
+                )
+                assert expected_value == result[2]
+            except Exception:
+                self.fail()
+                pass
+
+        check_RE_DBREF('#1234', '#1234')
+        check_RE_DBREF('(#1234)', '(#1234)')
+        check_RE_DBREF('obj(#1234)', 'obj(#1234)')
+        check_RE_DBREF('$obj(#1234)', '$obj(#1234)')
+        check_RE_DBREF('obj($obj(#1234))', 'obj($obj(#1234))')
+
     @mock.patch("evennia.prototypes.protfuncs.base_random", new=mock.MagicMock(return_value=0.5))
     @mock.patch("evennia.prototypes.protfuncs.base_randint", new=mock.MagicMock(return_value=5))
     def test_protfuncs(self):
@@ -315,10 +334,15 @@ class TestProtFuncs(EvenniaTest):
         self.assertEqual(protlib.protfunc_parser("$obj(#1)", session=self.session), '#1')
         self.assertEqual(protlib.protfunc_parser("stone(#12345)", session=self.session), 'stone(#12345)')
         self.assertEqual(protlib.protfunc_parser("#1", session=self.session), '#1')
+        self.assertEqual(protlib.protfunc_parser("#12345", session=self.session), '#12345')
+        self.assertEqual(protlib.protfunc_parser("nothing(#1)", session=self.session), 'nothing(#1)')
+        self.assertEqual(protlib.protfunc_parser("(#12345)", session=self.session), '(#12345)')
         self.assertEqual(protlib.protfunc_parser("$obj(Char)", session=self.session), '#6')
         self.assertEqual(protlib.protfunc_parser("$obj(Char)", session=self.session), '#6')
         self.assertEqual(protlib.protfunc_parser("$objlist(#1)", session=self.session), ['#1'])
 
+
+
         self.assertEqual(protlib.value_to_obj(
             protlib.protfunc_parser("#6", session=self.session)), self.char1)
         self.assertEqual(protlib.value_to_obj_or_any(