Added handling for infection in other zones

media/lua/client/TOC_Test.lua

@@ -1,26 +1,23 @@
--- TODO TestFramework stuff here
-
 if not getActivatedMods():contains("TEST_FRAMEWORK") or not isDebugEnabled() then return end
 local TestFramework = require("TestFramework/TestFramework")
 local TestUtils = require("TestFramework/TestUtils")
 
 local PlayerHandler = require("TOC_PlayerHandler")
 
-TestFramework.registerTestModule("Functionality", "Setup", function()
 
-    local Tests = {}
-    function Tests.InitializePlayer()
-        local pl = getPlayer()
-        PlayerHandler.InitializePlayer(nil, pl)
-        return true
-    end
-
-    return Tests
-end)
 
 TestFramework.registerTestModule("Functionality", "Amputation", function()
     local Tests = {}
 
+    function Tests.InitializePlayer()
+        return true
+
+        -- TODO This breaks the Test Framework mod for some reason.
+        -- local pl = getPlayer()
+        -- PlayerHandler.InitializePlayer(nil, pl)
+        -- return true
+    end
+
     function Tests.CutLeftHand()
         PlayerHandler.ForceCutLimb("Hand_L")
         return PlayerHandler.modDataHandler:getIsCut("Hand_L")