source: src/Fragmentation/Automation/unittests/WorkerPoolUnitTest.cpp@ 6ee809

Action_Thermostats Add_AtomRandomPerturbation Add_FitFragmentPartialChargesAction Add_RotateAroundBondAction Add_SelectAtomByNameAction Added_ParseSaveFragmentResults AddingActions_SaveParseParticleParameters Adding_Graph_to_ChangeBondActions Adding_MD_integration_tests Adding_ParticleName_to_Atom Adding_StructOpt_integration_tests AtomFragments Automaking_mpqc_open AutomationFragmentation_failures Candidate_v1.5.4 Candidate_v1.6.0 Candidate_v1.6.1 Candidate_v1.7.0 ChangeBugEmailaddress ChangingTestPorts ChemicalSpaceEvaluator CombiningParticlePotentialParsing Combining_Subpackages Debian_Package_split Debian_package_split_molecuildergui_only Disabling_MemDebug Docu_Python_wait EmpiricalPotential_contain_HomologyGraph EmpiricalPotential_contain_HomologyGraph_documentation Enable_parallel_make_install Enhance_userguide Enhanced_StructuralOptimization Enhanced_StructuralOptimization_continued Example_ManyWaysToTranslateAtom Exclude_Hydrogens_annealWithBondGraph FitPartialCharges_GlobalError Fix_BoundInBox_CenterInBox_MoleculeActions Fix_ChargeSampling_PBC Fix_ChronosMutex Fix_FitPartialCharges Fix_FitPotential_needs_atomicnumbers Fix_ForceAnnealing Fix_IndependentFragmentGrids Fix_ParseParticles Fix_ParseParticles_split_forward_backward_Actions Fix_PopActions Fix_QtFragmentList_sorted_selection Fix_Restrictedkeyset_FragmentMolecule Fix_StatusMsg Fix_StepWorldTime_single_argument Fix_Verbose_Codepatterns Fix_fitting_potentials Fixes ForceAnnealing_goodresults ForceAnnealing_oldresults ForceAnnealing_tocheck ForceAnnealing_with_BondGraph ForceAnnealing_with_BondGraph_continued ForceAnnealing_with_BondGraph_continued_betteresults ForceAnnealing_with_BondGraph_contraction-expansion FragmentAction_writes_AtomFragments FragmentMolecule_checks_bonddegrees GeometryObjects Gui_Fixes Gui_displays_atomic_force_velocity ImplicitCharges IndependentFragmentGrids IndependentFragmentGrids_IndividualZeroInstances IndependentFragmentGrids_IntegrationTest IndependentFragmentGrids_Sole_NN_Calculation JobMarket_RobustOnKillsSegFaults JobMarket_StableWorkerPool JobMarket_unresolvable_hostname_fix MoreRobust_FragmentAutomation ODR_violation_mpqc_open PartialCharges_OrthogonalSummation PdbParser_setsAtomName PythonUI_with_named_parameters QtGui_reactivate_TimeChanged_changes Recreated_GuiChecks Rewrite_FitPartialCharges RotateToPrincipalAxisSystem_UndoRedo SaturateAtoms_findBestMatching SaturateAtoms_singleDegree StoppableMakroAction Subpackage_CodePatterns Subpackage_JobMarket Subpackage_LinearAlgebra Subpackage_levmar Subpackage_mpqc_open Subpackage_vmg Switchable_LogView ThirdParty_MPQC_rebuilt_buildsystem TrajectoryDependenant_MaxOrder TremoloParser_IncreasedPrecision TremoloParser_MultipleTimesteps TremoloParser_setsAtomName Ubuntu_1604_changes stable
Last change on this file since 6ee809 was 6b3a37, checked in by Frederik Heber <heber@…>, 13 years ago

Added begin(),end() for idle_queue of WorkerPool.

  • this allows shutting down Workers without marking and then unmarking them as busy by FragmentScheduler.
  • also the returned iterators are const and hence allow to see the which workers are idle but not to modify them this way.
  • also added WorkerPool::hasBusyWorkers() to allow for waiting server shutdown until all workers have returned from work.
  • added some tests for this in WorkerPoolUnitTest.
  • Property mode set to 100644
File size: 6.0 KB
Line 
1/*
2 * Project: MoleCuilder
3 * Description: creates and alters molecular systems
4 * Copyright (C) 2010 University of Bonn. All rights reserved.
5 * Please see the LICENSE file or "Copyright notice" in builder.cpp for details.
6 */
7
8/*
9 * WorkerPoolUnitTest.cpp
10 *
11 * Created on: Feb 28, 2012
12 * Author: heber
13 */
14
15// include config.h
16#ifdef HAVE_CONFIG_H
17#include <config.h>
18#endif
19
20#include <cppunit/CompilerOutputter.h>
21#include <cppunit/extensions/TestFactoryRegistry.h>
22#include <cppunit/ui/text/TestRunner.h>
23
24#include "WorkerPoolUnitTest.hpp"
25
26#include <vector>
27
28#include "CodePatterns/Assert.hpp"
29#include "CodePatterns/Observer/Channels.hpp"
30#include "Pool/WorkerPool.hpp"
31#include "WorkerAddress.hpp"
32
33
34#ifdef HAVE_TESTRUNNER
35#include "UnitTestMain.hpp"
36#endif /*HAVE_TESTRUNNER*/
37
38/********************************************** Test classes **************************************/
39
40#include "stubs/FragmentJobStub.hpp"
41#include "unittests/stubs/ObserverStub.hpp"
42
43// Registers the fixture into the 'registry'
44CPPUNIT_TEST_SUITE_REGISTRATION( WorkerPoolTest );
45
46
47void WorkerPoolTest::setUp()
48{
49 // Throw assertions
50 ASSERT_DO(Assert::Throw);
51
52 pool = new WorkerPool();
53 addobserver = new NotificationObserver(pool->getChannel(WorkerPool::WorkerAdded));
54 removeobserver = new NotificationObserver(pool->getChannel(WorkerPool::WorkerRemoved));
55 idleobserver = new NotificationObserver(pool->getChannel(WorkerPool::WorkerIdle));
56
57 // and sign on
58 pool->signOn(addobserver, WorkerPool::WorkerAdded);
59 pool->signOn(removeobserver, WorkerPool::WorkerRemoved);
60 pool->signOn(idleobserver, WorkerPool::WorkerIdle);
61}
62
63void WorkerPoolTest::tearDown()
64{
65 pool->signOff(addobserver, WorkerPool::WorkerAdded);
66 pool->signOff(removeobserver, WorkerPool::WorkerRemoved);
67 pool->signOff(idleobserver, WorkerPool::WorkerIdle);
68
69 delete removeobserver;
70 delete addobserver;
71 delete idleobserver;
72 delete pool;
73}
74
75/** UnitTest for working add/removeWorker
76 */
77void WorkerPoolTest::WorkerTest()
78{
79 WorkerAddress address("service", "port");
80 WorkerAddress otheraddress("otherservice", "otherport");
81
82 // add worker
83 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->idle_queue.size() );
84 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
85 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->pool.size() );
86 CPPUNIT_ASSERT ( pool->begin_idle() == pool->end_idle() );
87 CPPUNIT_ASSERT( pool->addWorker(address) );
88 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->idle_queue.size() );
89 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
90 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->pool.size() );
91 CPPUNIT_ASSERT ( pool->begin_idle() != pool->end_idle() );
92 CPPUNIT_ASSERT_EQUAL( address, pool->begin_idle()->second );
93 CPPUNIT_ASSERT( addobserver->wasNotified );
94 CPPUNIT_ASSERT( !removeobserver->wasNotified );
95 CPPUNIT_ASSERT( idleobserver->wasNotified );
96 addobserver->wasNotified = false;
97 idleobserver->wasNotified = false;
98
99 // remove non-present worker
100 std::cout << "The following error message is intended and does not indicate a failure." << std::endl;
101 CPPUNIT_ASSERT( !pool->removeWorker(otheraddress) );
102 CPPUNIT_ASSERT( !addobserver->wasNotified );
103 CPPUNIT_ASSERT( !removeobserver->wasNotified );
104 CPPUNIT_ASSERT( !idleobserver->wasNotified );
105
106 // check for presence
107 CPPUNIT_ASSERT( pool->presentInPool(address) );
108 CPPUNIT_ASSERT( !pool->presentInPool(otheraddress) );
109
110 // remove worker
111 CPPUNIT_ASSERT( pool->removeWorker(address) );
112 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->idle_queue.size() );
113 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
114 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->pool.size() );
115 CPPUNIT_ASSERT( !addobserver->wasNotified );
116 CPPUNIT_ASSERT( removeobserver->wasNotified );
117 CPPUNIT_ASSERT( !idleobserver->wasNotified );
118 removeobserver->wasNotified = false;
119}
120
121/** UnitTest for working (un)markWorkerBusy
122 */
123void WorkerPoolTest::markBusyTest()
124{
125 WorkerAddress address("service", "port");
126 CPPUNIT_ASSERT( pool->addWorker(address) );
127
128 // check that not busy
129 CPPUNIT_ASSERT( !pool->isWorkerBusy(address) );
130 CPPUNIT_ASSERT( !pool->hasBusyWorkers() );
131 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->idle_queue.size() );
132 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
133 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->pool.size() );
134 CPPUNIT_ASSERT( addobserver->wasNotified );
135 CPPUNIT_ASSERT( !removeobserver->wasNotified );
136 CPPUNIT_ASSERT( idleobserver->wasNotified );
137 addobserver->wasNotified = false;
138 idleobserver->wasNotified = false;
139
140 // unmark and check
141 pool->unmarkWorkerBusy(address);
142 CPPUNIT_ASSERT( !pool->isWorkerBusy(address) );
143 CPPUNIT_ASSERT( !pool->hasBusyWorkers() );
144 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->idle_queue.size() );
145 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
146
147 // mark as busy and check
148 const WorkerAddress otheraddress = pool->getNextIdleWorker();
149 CPPUNIT_ASSERT( pool->isWorkerBusy(address) );
150 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->idle_queue.size() );
151 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->busy_queue.size() );
152
153 // get one more
154 CPPUNIT_ASSERT( !pool->presentIdleWorkers() );
155#ifndef NDEBUG
156 std::cout << "The following assertion is intended and does not indicate a failure." << std::endl;
157 CPPUNIT_ASSERT_THROW( pool->getNextIdleWorker(), Assert::AssertionFailure );
158#else
159 const WorkerAddress emptyAddress = pool->getNextIdleWorker();
160#endif
161 CPPUNIT_ASSERT( pool->isWorkerBusy(address) );
162 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->idle_queue.size() );
163 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->busy_queue.size() );
164 CPPUNIT_ASSERT( !addobserver->wasNotified );
165 CPPUNIT_ASSERT( !removeobserver->wasNotified );
166 CPPUNIT_ASSERT( !idleobserver->wasNotified );
167
168 // unmark and check
169 pool->unmarkWorkerBusy(address);
170 CPPUNIT_ASSERT( !pool->isWorkerBusy(address) );
171 CPPUNIT_ASSERT_EQUAL( (size_t)1, pool->idle_queue.size() );
172 CPPUNIT_ASSERT_EQUAL( (size_t)0, pool->busy_queue.size() );
173 CPPUNIT_ASSERT( !addobserver->wasNotified );
174 CPPUNIT_ASSERT( !removeobserver->wasNotified );
175 CPPUNIT_ASSERT( idleobserver->wasNotified );
176}
Note: See TracBrowser for help on using the repository browser.