From 321d4005fc697f763c7f796d98ca7be037953ec2 Mon Sep 17 00:00:00 2001
From: Tegiminis <dragonmaw@gmail.com>
Date: Sat, 12 Feb 2022 16:51:27 -0800
Subject: [PATCH] static trait unit tests

---
 evennia/contrib/rpg/traits/tests.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/evennia/contrib/rpg/traits/tests.py b/evennia/contrib/rpg/traits/tests.py
index bb31bbfc13..cadf777627 100644
--- a/evennia/contrib/rpg/traits/tests.py
+++ b/evennia/contrib/rpg/traits/tests.py
@@ -274,13 +274,14 @@ class TestTraitStatic(_TraitHandlerBase):
             trait_type="static",
             base=1,
             mod=2,
+            mult=2.0,
             extra_val1="xvalue1",
             extra_val2="xvalue2",
         )
         self.trait = self.traithandler.get("test1")
 
     def _get_values(self):
-        return self.trait.base, self.trait.mod, self.trait.value
+        return self.trait.base, self.trait.mod, self.trait.mult, self.trait.value
 
     def test_init(self):
         self.assertEqual(
@@ -290,25 +291,33 @@ class TestTraitStatic(_TraitHandlerBase):
                 "trait_type": "static",
                 "base": 1,
                 "mod": 2,
+                "mult": 2.0,
                 "extra_val1": "xvalue1",
                 "extra_val2": "xvalue2",
             },
         )
 
     def test_value(self):
-        """value is base + mod"""
-        self.assertEqual(self._get_values(), (1, 2, 3))
+        """value is (base + mod) * mult"""
+        self.assertEqual(self._get_values(), (1, 2, 2.0, 6))
         self.trait.base += 4
-        self.assertEqual(self._get_values(), (5, 2, 7))
+        self.assertEqual(self._get_values(), (5, 2, 2.0, 14))
         self.trait.mod -= 1
-        self.assertEqual(self._get_values(), (5, 1, 6))
+        self.assertEqual(self._get_values(), (5, 1, 2.0, 12))
+        self.trait.mult += 1.0
+        self.assertEqual(self._get_values(), (5, 1, 3.0, 18))
+        self.trait.mult = 0.75
+        self.assertEqual(self._get_values(), (5, 1, 0.75, 4.5))
+
 
     def test_delete(self):
         """Deleting resets to default."""
         del self.trait.base
-        self.assertEqual(self._get_values(), (0, 2, 2))
+        self.assertEqual(self._get_values(), (0, 2, 2.0, 4))
+        del self.trait.mult
+        self.assertEqual(self._get_values(), (0, 2, 1.0, 2))
         del self.trait.mod
-        self.assertEqual(self._get_values(), (0, 0, 0))
+        self.assertEqual(self._get_values(), (0, 0, 1.0, 0))
 
 
 class TestTraitCounter(_TraitHandlerBase):