Added reference counting leak check to integration test

FredyH · Jan 24, 2022 · a2b5f29 · a2b5f29
1 parent 18046a0
commit a2b5f29
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/IntegrationTest/lua/mysqloo/testframework.lua b/IntegrationTest/lua/mysqloo/testframework.lua
@@ -68,6 +68,17 @@ function TestFramework:OnCompleted()
 		else
 			MsgC(Color(255, 255, 255), "All allocated objects were freed\n")
 		end
+
+
+		diffBefore = TestFramework.ReferenceCreatedCount - TestFramework.ReferenceFreedCount
+		diffAfter = mysqloo.referenceCreatedCount() - mysqloo.referenceFreedCount()
+		if (diffAfter > diffBefore) then
+			MsgC(Color(255, 255, 255), "Found potential memory leak with ", diffAfter - diffBefore, " new references created that were not freed\n")
+		else
+			MsgC(Color(255, 255, 255), "All created references were freed\n")
+		end
+
+
 		MsgC(Color(255, 255, 255), "Lua Heap Before: ", TestFramework.LuaMemory, " After: ", collectgarbage("count"), "\n")
 	end)
 end
@@ -79,6 +90,8 @@ function TestFramework:Start()
 	TestFramework.CurrentIndex = 0
 	TestFramework.SuccessCount = 0
 	TestFramework.FailureCount = 0
+	TestFramework.ReferenceCreatedCount = mysqloo.referenceCreatedCount()
+	TestFramework.ReferenceFreedCount = mysqloo.referenceFreedCount()
 	TestFramework.AllocationCount = mysqloo.allocationCount()
 	TestFramework.DeallocationCount = mysqloo.deallocationCount()
 	TestFramework.LuaMemory = collectgarbage("count")