← Back to branch summary

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

« back to all changes in this revision

Viewing changes to dhis-2/dhis-services/dhis-service-core/src/test/java/org/hisp/dhis/dataelement/DataElementCategoryOptionComboStoreTest.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-services/dhis-service-core/src/test/java/org/hisp/dhis/dataelement/DataElementCategoryOptionComboStoreTest.java
 
1
package org.hisp.dhis.dataelement;
 
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.HashSet;
 
32
import java.util.Set;
 
33
 
 
34
import org.hisp.dhis.DhisConvenienceTest;
 
35
 
 
36
/**
 
37
 * @author Lars Helge Overland
 
38
 * @version $Id$
 
39
 */
 
40
public class DataElementCategoryOptionComboStoreTest
 
41
    extends DhisConvenienceTest
 
42
{
 
43
    private DataElementCategoryOptionComboStore categoryOptionComboStore;
 
44
    
 
45
    private DataElementCategory categoryA;
 
46
    private DataElementCategory categoryB;
 
47
        
 
48
    private DataElementCategoryCombo categoryComboA;
 
49
    private DataElementCategoryCombo categoryComboB;
 
50
    
 
51
    private DataElementCategoryOption categoryOptionA;
 
52
    private DataElementCategoryOption categoryOptionB;
 
53
    private DataElementCategoryOption categoryOptionC;
 
54
    private DataElementCategoryOption categoryOptionD;
 
55
    
 
56
    private DataElementCategoryOptionCombo categoryOptionComboA;
 
57
    private DataElementCategoryOptionCombo categoryOptionComboB;
 
58
    private DataElementCategoryOptionCombo categoryOptionComboC;
 
59
    
 
60
    // -------------------------------------------------------------------------
 
61
    // Fixture
 
62
    // -------------------------------------------------------------------------
 
63
 
 
64
    public void setUpTest()
 
65
        throws Exception
 
66
    {
 
67
        categoryService = (DataElementCategoryService) getBean( DataElementCategoryService.ID );
 
68
        
 
69
        categoryComboService = (DataElementCategoryComboService) getBean( DataElementCategoryComboService.ID );
 
70
        
 
71
        categoryOptionService = (DataElementCategoryOptionService) getBean( DataElementCategoryOptionService.ID );
 
72
        
 
73
        categoryOptionComboStore = (DataElementCategoryOptionComboStore) getBean( DataElementCategoryOptionComboStore.ID );
 
74
        
 
75
        categoryOptionA = new DataElementCategoryOption( "Male" );
 
76
        categoryOptionB = new DataElementCategoryOption( "Female" );
 
77
        categoryOptionC = new DataElementCategoryOption( "0-20" );
 
78
        categoryOptionD = new DataElementCategoryOption( "20-100" );
 
79
 
 
80
        categoryOptionService.addDataElementCategoryOption( categoryOptionA );
 
81
        categoryOptionService.addDataElementCategoryOption( categoryOptionB );
 
82
        categoryOptionService.addDataElementCategoryOption( categoryOptionC );
 
83
        categoryOptionService.addDataElementCategoryOption( categoryOptionD );
 
84
                
 
85
        categoryA = new DataElementCategory( "Gender" );
 
86
        categoryB = new DataElementCategory( "Agegroup" );
 
87
        
 
88
        categoryA.getCategoryOptions().add( categoryOptionA );
 
89
        categoryA.getCategoryOptions().add( categoryOptionB );        
 
90
        categoryB.getCategoryOptions().add( categoryOptionC );
 
91
        categoryB.getCategoryOptions().add( categoryOptionD );
 
92
        
 
93
        categoryService.addDataElementCategory( categoryA );
 
94
        categoryService.addDataElementCategory( categoryB );
 
95
        
 
96
        categoryComboA = new DataElementCategoryCombo( "GenderAgegroup" );
 
97
        categoryComboB = new DataElementCategoryCombo( "Gender" );
 
98
        
 
99
        categoryComboA.getCategories().add( categoryA );
 
100
        categoryComboA.getCategories().add( categoryB );
 
101
        categoryComboB.getCategories().add( categoryA );
 
102
        
 
103
        categoryComboService.addDataElementCategoryCombo( categoryComboA );
 
104
        categoryComboService.addDataElementCategoryCombo( categoryComboB ); 
 
105
    }
 
106
 
 
107
    // -------------------------------------------------------------------------
 
108
    // Tests
 
109
    // -------------------------------------------------------------------------
 
110
 
 
111
    public void testAddGetDataElementCategoryOptionCombo()
 
112
    {
 
113
        categoryOptionComboA = new DataElementCategoryOptionCombo();
 
114
        
 
115
        Set<DataElementCategoryOption> categoryOptions = new HashSet<DataElementCategoryOption>();
 
116
        
 
117
        categoryOptions.add( categoryOptionA );
 
118
        categoryOptions.add( categoryOptionB );        
 
119
        
 
120
        categoryOptionComboA.setCategoryCombo( categoryComboA );
 
121
        categoryOptionComboA.setCategoryOptions( categoryOptions );        
 
122
        
 
123
        int id = categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboA );
 
124
        
 
125
        categoryOptionComboA = categoryOptionComboStore.getDataElementCategoryOptionCombo( id );
 
126
        
 
127
        assertNotNull( categoryOptionComboA );
 
128
        assertEquals( categoryComboA, categoryOptionComboA.getCategoryCombo() );
 
129
        assertEquals( categoryOptions, categoryOptionComboA.getCategoryOptions() );
 
130
    }
 
131
    
 
132
    public void testUpdateGetDataElementCategoryOptionCombo()
 
133
    {
 
134
        categoryOptionComboA = new DataElementCategoryOptionCombo();
 
135
        
 
136
        Set<DataElementCategoryOption> categoryOptions = new HashSet<DataElementCategoryOption>();
 
137
        
 
138
        categoryOptions.add( categoryOptionA );
 
139
        categoryOptions.add( categoryOptionB );        
 
140
        
 
141
        categoryOptionComboA.setCategoryCombo( categoryComboA );
 
142
        categoryOptionComboA.setCategoryOptions( categoryOptions );        
 
143
        
 
144
        int id = categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboA );
 
145
        
 
146
        categoryOptionComboA = categoryOptionComboStore.getDataElementCategoryOptionCombo( id );
 
147
        
 
148
        assertNotNull( categoryOptionComboA );
 
149
        assertEquals( categoryComboA, categoryOptionComboA.getCategoryCombo() );
 
150
        assertEquals( categoryOptions, categoryOptionComboA.getCategoryOptions() );
 
151
        
 
152
        categoryOptionComboA.setCategoryCombo( categoryComboB );
 
153
        
 
154
        categoryOptionComboStore.updateDataElementCategoryOptionCombo( categoryOptionComboA );
 
155
        
 
156
        categoryOptionComboA = categoryOptionComboStore.getDataElementCategoryOptionCombo( id );
 
157
        
 
158
        assertNotNull( categoryOptionComboA );
 
159
        assertEquals( categoryComboB, categoryOptionComboA.getCategoryCombo() );
 
160
        assertEquals( categoryOptions, categoryOptionComboA.getCategoryOptions() );
 
161
    }
 
162
    
 
163
    public void testDeleteDataElementCategoryOptionCombo()
 
164
    {
 
165
        categoryOptionComboA = new DataElementCategoryOptionCombo();
 
166
        categoryOptionComboB = new DataElementCategoryOptionCombo();
 
167
        categoryOptionComboC = new DataElementCategoryOptionCombo();
 
168
        
 
169
        int idA = categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboA );
 
170
        int idB = categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboB );
 
171
        int idC = categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboC );
 
172
        
 
173
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idA ) );
 
174
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idB ) );
 
175
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idC ) );
 
176
        
 
177
        categoryOptionComboStore.deleteDataElementCategoryOptionCombo( categoryOptionComboStore.getDataElementCategoryOptionCombo( idA ) );
 
178
 
 
179
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idA ) );
 
180
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idB ) );
 
181
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idC ) );
 
182
 
 
183
        categoryOptionComboStore.deleteDataElementCategoryOptionCombo( categoryOptionComboStore.getDataElementCategoryOptionCombo( idB ) );
 
184
 
 
185
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idA ) );
 
186
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idB ) );
 
187
        assertNotNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idC ) );
 
188
 
 
189
        categoryOptionComboStore.deleteDataElementCategoryOptionCombo( categoryOptionComboStore.getDataElementCategoryOptionCombo( idC ) );
 
190
 
 
191
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idA ) );
 
192
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idB ) );
 
193
        assertNull( categoryOptionComboStore.getDataElementCategoryOptionCombo( idC ) );
 
194
    }
 
195
    
 
196
    public void testGetAllDataElementCategoryOptionCombos()
 
197
    {
 
198
        categoryOptionComboA = new DataElementCategoryOptionCombo();
 
199
        categoryOptionComboB = new DataElementCategoryOptionCombo();
 
200
        categoryOptionComboC = new DataElementCategoryOptionCombo();
 
201
        
 
202
        categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboA );
 
203
        categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboB );
 
204
        categoryOptionComboStore.addDataElementCategoryOptionCombo( categoryOptionComboC );
 
205
        
 
206
        Collection<DataElementCategoryOptionCombo> categoryOptionCombos = 
 
207
            categoryOptionComboStore.getAllDataElementCategoryOptionCombos();
 
208
        
 
209
        assertNotNull( categoryOptionCombos );
 
210
        assertEquals( 4, categoryOptionCombos.size() ); // Including default category option combo
 
211
    }
 
212
}