1 files changed, 121 insertions, 0 deletions
diff --git a/ACE/tests/Framework_Component_Test.cpp b/ACE/tests/Framework_Component_Test.cpp
new file mode 100644
index 00000000000..58c753c6200
--- /dev/null
+++ b/ACE/tests/Framework_Component_Test.cpp
@@ -0,0 +1,121 @@
+// $Id$
+
+// ============================================================================
+//
+// = LIBRARY
+//    tests
+//
+// = DESCRIPTION
+//    This program tests both the ACE_Framework_Compondent and ACE_Repository.
+//    Since Framework Components are singletons that can live in dlls loaded
+//    via the Service Configurator framework, this test uses that framework
+//    to load services from a dll that has a singleton based on ACE_DLL_Singleton.
+//    When the dll is finally ready to be unloaded, the singleton will be
+//    automatically cleaned up just-in-time.
+//
+// = AUTHOR
+//    Don Hinton <dhinton@ieee.org>
+//
+// ============================================================================
+
+#include "tests/test_config.h"
+#include "ace/Service_Config.h"
+#include "ace/ARGV.h"
+#include "ace/DLL_Manager.h"
+
+ACE_RCSID(tests, Framework_Component_Test, "$Id$")
+
+// Define a few macros--because they're so much fun, and keep the
+// code below a little cleaner...
+#if (ACE_USES_CLASSIC_SVC_CONF == 1)
+
+# define ADD_SERVICE(X) \
+    ACE_TEXT ("dynamic Server_") ACE_TEXT(#X) \
+    ACE_TEXT (" Service_Object * ") \
+    ACE_TEXT ("Framework_Component_DLL:_make_Server_") ACE_TEXT(#X)  \
+    ACE_TEXT ("() ''")
+
+# define REMOVE_SERVICE(X) \
+    ACE_TEXT ("remove Server_") ACE_TEXT(#X)
+
+#else /* ACE_USES_CLASSIC_SVC_CONF */
+
+# define ADD_SERVICE(X) \
+    ACE_TEXT ("<?xml version='1.0'?> <dynamic id='Server_") ACE_TEXT(#X) ACE_TEXT("' ") \
+    ACE_TEXT ("type='service_object'> <initializer init='_make_Server_") ACE_TEXT(#X) ACE_TEXT("' ") \
+    ACE_TEXT ("path='Framework_Component_DLL' params=''/> </dynamic>")
+
+# define REMOVE_SERVICE(X) \
+   ACE_TEXT ( "<?xml version='1.0'?> <remove id='Server_") ACE_TEXT(#X) \
+   ACE_TEXT("'> </remove>")
+
+#endif /* ACE_USES_CLASSIC_SVC_CONF */
+
+int
+run_test (u_long unload_mask = 0)
+{
+  ACE_DEBUG ((LM_DEBUG, ACE_TEXT ("Running test with mask = %s|%s\n"),
+              ACE_BIT_ENABLED(unload_mask, ACE_DLL_UNLOAD_POLICY_PER_DLL) == 0
+              ? ACE_TEXT ("PER_PROCESS") : ACE_TEXT ("PER_DLL"),
+              ACE_BIT_ENABLED(unload_mask, ACE_DLL_UNLOAD_POLICY_LAZY) == 0
+              ? ACE_TEXT ("EAGER") : ACE_TEXT ("LAZY")));
+
+  ACE_DLL_Manager::instance ()->unload_policy (unload_mask);
+
+  // Now, let the ACE Service Configurator framework load our service from a
+  // dll, which contains a singleton.
+  ACE_Service_Config::open (ACE_TEXT ("Framework_Component_Test"),
+                            ACE_DEFAULT_LOGGER_KEY,
+                            1, 1, 1);
+
+  // Now add server 1.
+  ACE_Service_Config::process_directive (ADD_SERVICE(1));
+
+  // And unload the first one, could unload the dll.
+  ACE_Service_Config::process_directive (REMOVE_SERVICE(1));
+
+  // Now readd server 1.
+  ACE_Service_Config::process_directive (ADD_SERVICE(1));
+
+  // And load another service from the same library.
+  ACE_Service_Config::process_directive (ADD_SERVICE(2));
+
+  // Unload the first one again, should *not* unload the dll this time.
+  ACE_Service_Config::process_directive (REMOVE_SERVICE(1));
+
+  // And unload the second service.  Since the ACE_DLL_Handle will no longer
+  // have any references, the ACE_DLL_Manager will apply it's current unloading
+  // strategy and either call ACE_OS::dlclose() immediately, schedule a timeout
+  // the the reactor to call dlclose() some time in the future, or keep the
+  // dll loaded until program termination.
+  ACE_Service_Config::process_directive (REMOVE_SERVICE(2));
+
+  // Force unloading so we'll be ready for the next test.
+  ACE_DLL_Manager::instance ()->unload_policy (ACE_DLL_UNLOAD_POLICY_DEFAULT);
+
+  ACE_Service_Config::close ();
+  return 0;
+}
+
+int
+run_main (int, ACE_TCHAR *[])
+{
+  ACE_START_TEST (ACE_TEXT("Framework_Component_Test"));
+
+  int retval = 0;
+
+  // Use defaults, i.e., per process, eager unloading.
+  retval += run_test (ACE_DLL_UNLOAD_POLICY_DEFAULT);
+
+  // Per process, lazy unloading
+  retval += run_test (ACE_DLL_UNLOAD_POLICY_LAZY);
+
+  // Per dll, eager unloading
+  retval += run_test (ACE_DLL_UNLOAD_POLICY_PER_DLL);
+
+  // Per dll, lazy unloading.
+  retval += run_test (ACE_DLL_UNLOAD_POLICY_PER_DLL | ACE_DLL_UNLOAD_POLICY_LAZY);
+
+  ACE_END_TEST;
+  return retval == 0 ? 0 : -1;
+}