Enable to set connection context without changing deleter.

src/factory/WFConnection.h

@@ -38,6 +38,11 @@ class WFConnection : public CommConnection
 		this->deleter = std::move(deleter);
 	}
 
+	void set_context(void *context)
+	{
+		this->context = context;
+	}
+
 	void *test_set_context(void *test_context, void *new_context,
 						   std::function<void (void *)> deleter)
 	{
@@ -50,6 +55,14 @@ class WFConnection : public CommConnection
 		return test_context;
 	}
 
+	void *test_set_context(void *test_context, void *new_context)
+	{
+		if (this->context.compare_exchange_strong(test_context, new_context))
+			return new_context;
+
+		return test_context;
+	}
+
 private:
 	std::atomic<void *> context;
 	std::function<void (void *)> deleter;