summaryrefslogtreecommitdiff
path: root/tests/Framework_Component_Test.cpp
blob: 2b7092a22566eb9aacac5c41f38918d8ce6ddc81 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
// $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 "ace/Service_Config.h"
#include "ace/ARGV.h"
#include "tests/test_config.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_" #X " Service_Object * " \
    "Framework_Component_DLL:_make_Server_" #X "() ''")

# define REMOVE_SERVICE(X) ACE_TEXT ( \
    "remove Server_" #X)

#else /* ACE_USES_CLASSIC_SVC_CONF */

# define ADD_SERVICE(X) ACE_TEXT ( \
    "<?xml version='1.0'?> <dynamic id='Server_" #X "' " \
    "type='service_object'> <initializer init='_make_Server_" #X "' " \
    "path='Framework_Component_DLL' params=''/> </dynamic>")

# define REMOVE_SERVICE(X) ACE_TEXT ( \
    "<?xml version='1.0'?> <remove id='Server_" #X "'> </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_Manager_Ex::PER_DLL) == 0
              ? ACE_TEXT ("PER_PROCESS") : ACE_TEXT ("PER_DLL"),
              ACE_BIT_ENABLED(unload_mask, ACE_DLL_Manager_Ex::LAZY) == 0
              ? ACE_TEXT ("EAGER") : ACE_TEXT ("LAZY")));

  ACE_DLL_Manager::instance ()->unload_strategy (unload_mask);

  // Now, let the ACE Service Configurator framework load our service from a
  // dll, which contains a singleton.
  ACE_ARGV args;
  args.add (ACE_TEXT ("Framework_Component_Test"));
  args.add (ACE_TEXT ("-n"));
  args.add (ACE_TEXT ("-d"));

  // Initialize Service Config.
  ACE_Service_Config::open (args.argc (), args.argv ());

  // 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_strategy (ACE_DLL_Manager_Ex::DEFAULT);

  ACE_Service_Config::close ();
  return 0;
}

int
ACE_TMAIN (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 ();

  // Per process, lazy unloading
  retval += run_test (ACE_DLL_Manager_Ex::LAZY);

  // Per dll, eager unloading
  retval += run_test (ACE_DLL_Manager_Ex::PER_DLL);

  // Per dll, lazy unloading.
  retval += run_test (ACE_DLL_Manager_Ex::PER_DLL | ACE_DLL_Manager_Ex::LAZY);

  ACE_END_TEST;
  return retval == 0 ? 0 : -1;
}