← Back to branch summary

~mortenoh/+junk/dhis2-detailed-import-export

« back to all changes in this revision

Viewing changes to dhis-2/dhis-web/dhis-web-commons/src/test/java/org/hisp/dhis/ouwt/manager/OrganisationUnitSelectionManagerTest.java

  • Committer: larshelge at gmail
  • Date: 2009-03-03 16:46:36 UTC
  • Revision ID: larshelge@gmail.com-20090303164636-2sjlrquo7ib1gf7r
Initial check-in

Show diffs side-by-side

added added

removed removed

Lines of Context:
dhis-2/dhis-web/dhis-web-commons/src/test/java/org/hisp/dhis/ouwt/manager/OrganisationUnitSelectionManagerTest.java
 
1
package org.hisp.dhis.ouwt.manager;
 
2
 
 
3
/*
 
4
 * Copyright (c) 2004-2007, University of Oslo
 
5
 * All rights reserved.
 
6
 *
 
7
 * Redistribution and use in source and binary forms, with or without
 
8
 * modification, are permitted provided that the following conditions are met:
 
9
 * * Redistributions of source code must retain the above copyright notice, this
 
10
 *   list of conditions and the following disclaimer.
 
11
 * * Redistributions in binary form must reproduce the above copyright notice,
 
12
 *   this list of conditions and the following disclaimer in the documentation
 
13
 *   and/or other materials provided with the distribution.
 
14
 * * Neither the name of the HISP project nor the names of its contributors may
 
15
 *   be used to endorse or promote products derived from this software without
 
16
 *   specific prior written permission.
 
17
 *
 
18
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 
19
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 
20
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 
21
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 
22
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 
23
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 
24
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 
25
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 
26
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 
27
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
28
 */
 
29
 
 
30
import java.util.Collection;
 
31
import java.util.Date;
 
32
import java.util.HashSet;
 
33
import java.util.Set;
 
34
 
 
35
import org.hisp.dhis.DhisSpringTest;
 
36
import org.hisp.dhis.organisationunit.OrganisationUnit;
 
37
import org.hisp.dhis.organisationunit.OrganisationUnitService;
 
38
import org.hisp.dhis.transaction.TransactionManager;
 
39
 
 
40
/**
 
41
 * @author Torgeir Lorange Ostby
 
42
 * @version $Id: OrganisationUnitSelectionManagerTest.java 6335 2008-11-20 11:11:26Z larshelg $
 
43
 */
 
44
public class OrganisationUnitSelectionManagerTest
 
45
    extends DhisSpringTest
 
46
{
 
47
    private OrganisationUnitService organisationUnitService;
 
48
 
 
49
    private TransactionManager transactionManager;
 
50
 
 
51
    private OrganisationUnitSelectionManager selectionManager;
 
52
 
 
53
    public void setUpTest()
 
54
        throws Exception
 
55
    {
 
56
        organisationUnitService = (OrganisationUnitService) getBean( OrganisationUnitService.ID );
 
57
 
 
58
        transactionManager = (TransactionManager) getBean( TransactionManager.ID );
 
59
 
 
60
        // ---------------------------------------------------------------------
 
61
        // Have to replace the session part with a dummy
 
62
        // ---------------------------------------------------------------------
 
63
 
 
64
        SessionReplacerOUSManager selectionManager = new SessionReplacerOUSManager();
 
65
        selectionManager.setOrganisationUnitService( organisationUnitService );
 
66
        this.selectionManager = selectionManager;
 
67
 
 
68
        // ---------------------------------------------------------------------
 
69
        // Add a couple of org units
 
70
        // ---------------------------------------------------------------------
 
71
 
 
72
        transactionManager.enter();
 
73
        OrganisationUnit parentUnit = new OrganisationUnit( "OrganisationUnit1", "OrgUnit1", "OU1", new Date(),
 
74
            new Date(), true, "Comment1" );
 
75
        organisationUnitService.addOrganisationUnit( parentUnit );
 
76
        OrganisationUnit childUnit = new OrganisationUnit( "OrganisationUnit2", parentUnit, "OrgUnit2", "OU2",
 
77
            new Date(), new Date(), true, "Comment2" );
 
78
        parentUnit.getChildren().add( childUnit );
 
79
        organisationUnitService.addOrganisationUnit( childUnit );
 
80
        transactionManager.leave();
 
81
    }
 
82
 
 
83
    public void testRoot()
 
84
        throws Exception
 
85
    {
 
86
        transactionManager.enter();
 
87
        OrganisationUnit rootUnit = getSingleRootOrganisationUnit();
 
88
        OrganisationUnit child = rootUnit.getChildren().iterator().next();
 
89
        transactionManager.leave();
 
90
 
 
91
        assertNotNull( rootUnit );
 
92
 
 
93
        transactionManager.enter();
 
94
        setSingleRootOrganisationUnitParent( child );
 
95
        assertEquals( child.getId(), getSingleRootOrganisationUnit().getId() );
 
96
        transactionManager.leave();
 
97
 
 
98
        transactionManager.enter();
 
99
        selectionManager.resetRootOrganisationUnits();
 
100
        assertEquals( rootUnit.getId(), getSingleRootOrganisationUnit().getId() );
 
101
        transactionManager.leave();
 
102
    }
 
103
 
 
104
    public void testSelection()
 
105
        throws Exception
 
106
    {
 
107
        transactionManager.enter();
 
108
        assertTrue( selectionManager.getSelectedOrganisationUnits().isEmpty() );
 
109
        transactionManager.leave();
 
110
 
 
111
        transactionManager.enter();
 
112
        OrganisationUnit rootUnit = getSingleRootOrganisationUnit();
 
113
        OrganisationUnit child = rootUnit.getChildren().iterator().next();
 
114
        transactionManager.leave();
 
115
 
 
116
        transactionManager.enter();
 
117
        setSingleSelectedOrganisationUnit( child );
 
118
        assertEquals( child.getId(), getSingleSelectedOrganisationUnit().getId() );
 
119
        transactionManager.leave();
 
120
 
 
121
        transactionManager.enter();
 
122
        selectionManager.clearSelectedOrganisationUnits();
 
123
        assertTrue( selectionManager.getSelectedOrganisationUnits().isEmpty() );
 
124
        transactionManager.leave();
 
125
    }
 
126
 
 
127
    public void testParentChildPaths()
 
128
        throws Exception
 
129
    {
 
130
        transactionManager.enter();
 
131
        OrganisationUnit rootUnit = getSingleRootOrganisationUnit();
 
132
        OrganisationUnit child = rootUnit.getChildren().iterator().next();
 
133
        transactionManager.leave();
 
134
 
 
135
        transactionManager.enter();
 
136
        setSingleRootOrganisationUnitParent( rootUnit );
 
137
        setSingleSelectedOrganisationUnit( child );
 
138
        assertEquals( rootUnit.getId(), getSingleRootOrganisationUnit().getId() );
 
139
        assertEquals( child.getId(), getSingleSelectedOrganisationUnit().getId() );
 
140
        transactionManager.leave();
 
141
 
 
142
        transactionManager.enter();
 
143
        setSingleRootOrganisationUnitParent( child );
 
144
        assertEquals( child.getId(), getSingleRootOrganisationUnit().getId() );
 
145
        assertNull( getSingleSelectedOrganisationUnit() );
 
146
        transactionManager.leave();
 
147
 
 
148
        transactionManager.enter();
 
149
        setSingleSelectedOrganisationUnit( rootUnit );
 
150
        //assertTrue( selectionManager.getSelectedOrganisationUnits().size() == 0 );
 
151
        transactionManager.leave();
 
152
 
 
153
        transactionManager.enter();
 
154
        setSingleRootOrganisationUnitParent( rootUnit );
 
155
        setSingleSelectedOrganisationUnit( rootUnit );
 
156
        assertEquals( rootUnit.getId(), getSingleRootOrganisationUnit().getId() );
 
157
        assertEquals( rootUnit.getId(), getSingleSelectedOrganisationUnit().getId() );
 
158
        transactionManager.leave();
 
159
 
 
160
        transactionManager.enter();
 
161
        setSingleRootOrganisationUnitParent( child );
 
162
        assertEquals( child.getId(), getSingleRootOrganisationUnit().getId() );
 
163
        assertNull( getSingleSelectedOrganisationUnit() );
 
164
        transactionManager.leave();
 
165
    }
 
166
 
 
167
    private void setSingleRootOrganisationUnitParent( OrganisationUnit unit )
 
168
        throws Exception
 
169
    {
 
170
        OrganisationUnit parent = unit.getParent();
 
171
 
 
172
        if ( parent == null )
 
173
        {
 
174
            selectionManager.resetRootOrganisationUnits();
 
175
        }
 
176
        else
 
177
        {
 
178
            selectionManager.setRootOrganisationUnitsParent( parent );
 
179
        }
 
180
    }
 
181
 
 
182
    private void setSingleSelectedOrganisationUnit( OrganisationUnit unit )
 
183
        throws Exception
 
184
    {
 
185
        Set<OrganisationUnit> selectedUnits = new HashSet<OrganisationUnit>( 1 );
 
186
        selectedUnits.add( unit );
 
187
        selectionManager.setSelectedOrganisationUnits( selectedUnits );
 
188
    }
 
189
 
 
190
    private OrganisationUnit getSingleRootOrganisationUnit()
 
191
        throws Exception
 
192
    {
 
193
        Collection<OrganisationUnit> rootUnits = getRootOrganisationUnits();
 
194
 
 
195
        return rootUnits.iterator().next();
 
196
    }
 
197
 
 
198
    private OrganisationUnit getSingleSelectedOrganisationUnit()
 
199
        throws Exception
 
200
    {
 
201
        Collection<OrganisationUnit> selectedUnits = selectionManager.getSelectedOrganisationUnits();
 
202
 
 
203
        if ( selectedUnits.isEmpty() )
 
204
        {
 
205
            return null;
 
206
        }
 
207
        return selectedUnits.iterator().next();
 
208
    }
 
209
 
 
210
    private Collection<OrganisationUnit> getRootOrganisationUnits()
 
211
        throws Exception
 
212
    {
 
213
        return selectionManager.getRootOrganisationUnits();
 
214
    }
 
215
}