From 52b93a08778f786d6cbbad5a5047a660974e689f Mon Sep 17 00:00:00 2001
From: Jordan Brown <Cyberboss@users.noreply.github.com>
Date: Mon, 27 Apr 2020 23:31:40 -0400
Subject: [PATCH] Improve the BasicOperation DMAPI test

---
 tests/DMAPI/BasicOperation/Test.dm | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/DMAPI/BasicOperation/Test.dm b/tests/DMAPI/BasicOperation/Test.dm
index 8b7ecd6253..fc36d40382 100644
--- a/tests/DMAPI/BasicOperation/Test.dm
+++ b/tests/DMAPI/BasicOperation/Test.dm
@@ -1,11 +1,21 @@
 /world/New()
 	TgsNew()
+
+	// Validate TGS_DMAPI_VERSION against DMAPI version used
+	var/datum/tgs_version/active_version = world.TgsApiVersion()
+	var/datum/tgs_version/dmapi_version = new /datum/tgs_version(TGS_DMAPI_VERSION)
+
+	if(!active_version.Equals(dmapi_version))
+		del(src)
+
 	StartAsync()
 
 /proc/StartAsync()
 	set waitfor = FALSE
 	sleep(100)
 	TgsInitializationComplete()
+	sleep(100)
+	world.TgsEndProcess()
 
 /world/Topic(T, Addr, Master, Keys)
 	TGS_TOPIC