← 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/dataset/DataSetServiceTest.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/dataset/DataSetServiceTest.java
 
1
package org.hisp.dhis.dataset;
 
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
 
 
33
import org.hisp.dhis.DhisConvenienceTest;
 
34
import org.hisp.dhis.period.MonthlyPeriodType;
 
35
import org.hisp.dhis.period.PeriodStore;
 
36
import org.hisp.dhis.period.PeriodType;
 
37
import org.hisp.dhis.period.WeeklyPeriodType;
 
38
import org.hisp.dhis.period.YearlyPeriodType;
 
39
import org.hisp.dhis.source.DummySource;
 
40
import org.hisp.dhis.source.Source;
 
41
import org.hisp.dhis.source.SourceStore;
 
42
 
 
43
/**
 
44
 * @author Lars Helge Overland
 
45
 * @version $Id$
 
46
 */
 
47
public class DataSetServiceTest
 
48
    extends DhisConvenienceTest
 
49
{
 
50
    private PeriodStore periodStore;
 
51
 
 
52
    private DataSetService dataSetService;
 
53
 
 
54
    private SourceStore sourceStore;
 
55
 
 
56
    private PeriodType periodType;
 
57
 
 
58
    public void setUpTest()
 
59
        throws Exception
 
60
    {
 
61
        dataSetService = (DataSetService) getBean( DataSetService.ID );
 
62
 
 
63
        periodStore = (PeriodStore) getBean( PeriodStore.ID );
 
64
 
 
65
        sourceStore = (SourceStore) getBean( SourceStore.ID );
 
66
        
 
67
        periodType = PeriodType.getAvailablePeriodTypes().iterator().next();
 
68
    }
 
69
 
 
70
    // -------------------------------------------------------------------------
 
71
    // Supportive methods
 
72
    // -------------------------------------------------------------------------
 
73
 
 
74
    private void assertEquals( char uniqueCharacter, DataSet dataSet )
 
75
    {
 
76
        assertEquals( "DataSet" + uniqueCharacter, dataSet.getName() );
 
77
        assertEquals( "ShortName" + uniqueCharacter, dataSet.getShortName() );
 
78
        assertEquals( periodType, dataSet.getPeriodType() );
 
79
    }
 
80
    
 
81
    // -------------------------------------------------------------------------
 
82
    // DataSet
 
83
    // -------------------------------------------------------------------------
 
84
 
 
85
    public void testAddDataSet()
 
86
    {
 
87
        DataSet dataSetA = createDataSet( 'A', periodType );
 
88
        DataSet dataSetB = createDataSet( 'B', periodType );
 
89
        
 
90
        int idA = dataSetService.addDataSet( dataSetA );
 
91
        int idB = dataSetService.addDataSet( dataSetB );       
 
92
        
 
93
        dataSetA = dataSetService.getDataSet( idA );
 
94
        dataSetB = dataSetService.getDataSet( idB );
 
95
        
 
96
        assertEquals( idA, dataSetA.getId() );
 
97
        assertEquals( 'A', dataSetA );
 
98
        
 
99
        assertEquals( idB, dataSetB.getId() );
 
100
        assertEquals( 'B', dataSetB );
 
101
    }
 
102
    
 
103
    public void testUpdateDataSet()
 
104
    {
 
105
        DataSet dataSet = createDataSet( 'A', periodType );
 
106
        
 
107
        int id = dataSetService.addDataSet( dataSet );
 
108
        
 
109
        dataSet = dataSetService.getDataSet( id );
 
110
        
 
111
        assertEquals( 'A', dataSet );
 
112
        
 
113
        dataSet.setName( "DataSetB" );
 
114
        
 
115
        dataSetService.updateDataSet( dataSet );
 
116
        
 
117
        dataSet = dataSetService.getDataSet( id );
 
118
        
 
119
        assertEquals( dataSet.getName(), "DataSetB" );
 
120
    }
 
121
    
 
122
    public void testDeleteAndGetDataSet()
 
123
    {
 
124
        DataSet dataSetA = createDataSet( 'A', periodType );
 
125
        DataSet dataSetB = createDataSet( 'B', periodType );
 
126
        
 
127
        int idA = dataSetService.addDataSet( dataSetA );
 
128
        int idB = dataSetService.addDataSet( dataSetB );
 
129
        
 
130
        assertNotNull( dataSetService.getDataSet( idA ) );
 
131
        assertNotNull( dataSetService.getDataSet( idB ) );
 
132
        
 
133
        dataSetService.deleteDataSet( dataSetService.getDataSet( idA ) );
 
134
        
 
135
        assertNull( dataSetService.getDataSet( idA ) );
 
136
        assertNotNull( dataSetService.getDataSet( idB ) );
 
137
 
 
138
        dataSetService.deleteDataSet( dataSetService.getDataSet( idB ) );
 
139
        
 
140
        assertNull( dataSetService.getDataSet( idA ) );
 
141
        assertNull( dataSetService.getDataSet( idB ) );        
 
142
    }
 
143
    
 
144
    public void testGetDataSetByName()
 
145
        throws Exception
 
146
    {
 
147
        DataSet dataSetA = createDataSet( 'A', periodType );
 
148
        DataSet dataSetB = createDataSet( 'B', periodType );
 
149
    
 
150
        int idA = dataSetService.addDataSet( dataSetA );
 
151
        int idB = dataSetService.addDataSet( dataSetB );
 
152
    
 
153
        assertEquals( dataSetService.getDataSetByName( "DataSetA" ).getId(), idA );
 
154
        assertEquals( dataSetService.getDataSetByName( "DataSetB" ).getId(), idB );
 
155
        assertNull( dataSetService.getDataSetByName( "DataSetC" ) );
 
156
    }
 
157
 
 
158
    public void testGetDataSetByShortName()
 
159
        throws Exception
 
160
    {
 
161
        DataSet dataSetA = createDataSet( 'A', periodType );
 
162
        DataSet dataSetB = createDataSet( 'B', periodType );
 
163
 
 
164
        int idA = dataSetService.addDataSet( dataSetA );
 
165
        int idB = dataSetService.addDataSet( dataSetB );
 
166
 
 
167
        assertEquals( dataSetService.getDataSetByShortName( "ShortNameA" ).getId(), idA );
 
168
        assertEquals( dataSetService.getDataSetByShortName( "ShortNameB" ).getId(), idB );
 
169
        assertNull( dataSetService.getDataSetByShortName( "ShortNameC" ) );
 
170
    }
 
171
 
 
172
    public void testGetAllDataSets()
 
173
    {
 
174
        DataSet dataSetA = createDataSet( 'A', periodType );
 
175
        DataSet dataSetB = createDataSet( 'B', periodType );
 
176
    
 
177
        dataSetService.addDataSet( dataSetA );
 
178
        dataSetService.addDataSet( dataSetB );
 
179
        
 
180
        Collection<DataSet> dataSets = dataSetService.getAllDataSets();
 
181
        
 
182
        assertEquals( dataSets.size(), 2 );
 
183
        assertTrue( dataSets.contains( dataSetA ) );
 
184
        assertTrue( dataSets.contains( dataSetB ) );            
 
185
    }
 
186
 
 
187
    public void testGetDataSetsBySource()
 
188
    {
 
189
        Source sourceA = new DummySource( "A" );
 
190
        Source sourceB = new DummySource( "B" );
 
191
        Source sourceC = new DummySource( "C" );
 
192
        
 
193
        sourceStore.addSource( sourceA );
 
194
        sourceStore.addSource( sourceB );
 
195
        sourceStore.addSource( sourceC );
 
196
        
 
197
        DataSet dataSetA = createDataSet( 'A', periodType );
 
198
        DataSet dataSetB = createDataSet( 'B', periodType );
 
199
        DataSet dataSetC = createDataSet( 'C', periodType );
 
200
        
 
201
        dataSetA.getSources().add( sourceA );
 
202
        dataSetA.getSources().add( sourceB );
 
203
        
 
204
        dataSetB.getSources().add( sourceB );
 
205
        dataSetB.getSources().add( sourceC );        
 
206
 
 
207
        dataSetC.getSources().add( sourceA );
 
208
        dataSetC.getSources().add( sourceC );
 
209
        
 
210
        dataSetService.addDataSet( dataSetA );
 
211
        dataSetService.addDataSet( dataSetB );
 
212
        dataSetService.addDataSet( dataSetC );
 
213
        
 
214
        Collection<DataSet> dataSets = dataSetService.getDataSetsBySource( sourceA );
 
215
        
 
216
        assertEquals( 2, dataSets.size() );
 
217
        assertTrue( dataSets.contains( dataSetA ) );
 
218
        assertTrue( dataSets.contains( dataSetC ) );
 
219
        
 
220
        dataSets = dataSetService.getDataSetsBySource( sourceB );        
 
221
 
 
222
        assertEquals( 2, dataSets.size() );
 
223
        assertTrue( dataSets.contains( dataSetA ) );
 
224
        assertTrue( dataSets.contains( dataSetB ) );
 
225
 
 
226
        dataSets = dataSetService.getDataSetsBySource( sourceC );
 
227
        
 
228
        assertEquals( 2, dataSets.size() );
 
229
        assertTrue( dataSets.contains( dataSetB ) );
 
230
        assertTrue( dataSets.contains( dataSetC ) );        
 
231
    }
 
232
 
 
233
    public void testGetDataSetsBySources()
 
234
    {
 
235
        Source sourceA = new DummySource( "A" );
 
236
        Source sourceB = new DummySource( "B" );
 
237
        Source sourceC = new DummySource( "C" );
 
238
        Source sourceD = new DummySource( "D" );
 
239
        
 
240
        sourceStore.addSource( sourceA );
 
241
        sourceStore.addSource( sourceB );
 
242
        sourceStore.addSource( sourceC );
 
243
        sourceStore.addSource( sourceD );
 
244
        
 
245
        DataSet dataSetA = createDataSet( 'A', periodType );
 
246
        DataSet dataSetB = createDataSet( 'B', periodType );
 
247
        DataSet dataSetC = createDataSet( 'C', periodType );
 
248
        DataSet dataSetD = createDataSet( 'D', periodType );
 
249
        
 
250
        dataSetA.getSources().add( sourceA );
 
251
        dataSetA.getSources().add( sourceB );
 
252
        
 
253
        dataSetB.getSources().add( sourceB );
 
254
        dataSetB.getSources().add( sourceC );        
 
255
 
 
256
        dataSetC.getSources().add( sourceC );
 
257
        
 
258
        dataSetD.getSources().add( sourceD );
 
259
        
 
260
        dataSetService.addDataSet( dataSetA );
 
261
        dataSetService.addDataSet( dataSetB );
 
262
        dataSetService.addDataSet( dataSetC );
 
263
        dataSetService.addDataSet( dataSetD );
 
264
        
 
265
        Collection<Source> sources = new HashSet<Source>();
 
266
        
 
267
        sources.add( sourceA );
 
268
        sources.add( sourceD );        
 
269
        
 
270
        Collection<DataSet> dataSets = dataSetService.getDataSetsBySources( sources );
 
271
        
 
272
        assertEquals( 2, dataSets.size() );
 
273
        assertTrue( dataSets.contains( dataSetA ) );
 
274
        assertTrue( dataSets.contains( dataSetD ) );
 
275
        
 
276
        sources.clear();
 
277
        
 
278
        sources.add( sourceB );
 
279
        sources.add( sourceC );
 
280
 
 
281
        dataSets = dataSetService.getDataSetsBySources( sources );
 
282
        
 
283
        assertEquals( 3, dataSets.size() );
 
284
        assertTrue( dataSets.contains( dataSetA ) );
 
285
        assertTrue( dataSets.contains( dataSetB ) );
 
286
        assertTrue( dataSets.contains( dataSetC ) );        
 
287
    }
 
288
 
 
289
    public void testGetSourcesAssociatedWithDataSet()
 
290
    {
 
291
        Source sourceA = new DummySource( "A" );
 
292
        Source sourceB = new DummySource( "B" );
 
293
        Source sourceC = new DummySource( "C" );
 
294
        Source sourceD = new DummySource( "D" );
 
295
        Source sourceE = new DummySource( "E" );
 
296
        Source sourceF = new DummySource( "F" );
 
297
        
 
298
        sourceStore.addSource( sourceA );
 
299
        sourceStore.addSource( sourceB );
 
300
        sourceStore.addSource( sourceC );
 
301
        sourceStore.addSource( sourceD );
 
302
        sourceStore.addSource( sourceE );
 
303
        sourceStore.addSource( sourceF );
 
304
        
 
305
        DataSet dataSetA = createDataSet( 'A', periodType );
 
306
        DataSet dataSetB = createDataSet( 'B', periodType );
 
307
        
 
308
        dataSetA.getSources().add( sourceA );
 
309
        dataSetA.getSources().add( sourceB );
 
310
        dataSetA.getSources().add( sourceC );
 
311
 
 
312
        dataSetB.getSources().add( sourceC );
 
313
        dataSetB.getSources().add( sourceD );
 
314
        dataSetB.getSources().add( sourceE );       
 
315
        
 
316
        dataSetService.addDataSet( dataSetA );
 
317
        dataSetService.addDataSet( dataSetB );
 
318
 
 
319
        Collection<Source> sources = new HashSet<Source>();
 
320
        
 
321
        sources.add( sourceA );
 
322
        sources.add( sourceB );
 
323
        sources.add( sourceD );
 
324
        sources.add( sourceE );
 
325
        
 
326
        assertEquals( 2, dataSetService.getSourcesAssociatedWithDataSet( dataSetA, sources ) );
 
327
        assertEquals( 2, dataSetService.getSourcesAssociatedWithDataSet( dataSetB, sources ) );
 
328
    }
 
329
        
 
330
    // -------------------------------------------------------------------------
 
331
    // FrequencyOverrideAssociation
 
332
    // -------------------------------------------------------------------------
 
333
    
 
334
    public void testFrequencyOverrideAssociation()
 
335
        throws Exception
 
336
    {
 
337
        PeriodType periodType1 = periodStore.getPeriodType( YearlyPeriodType.class );
 
338
        PeriodType periodType2 = periodStore.getPeriodType( MonthlyPeriodType.class );
 
339
        PeriodType periodType3 = periodStore.getPeriodType( WeeklyPeriodType.class );
 
340
 
 
341
        DataSet dataSet1 = new DataSet( "name1", periodType1 );
 
342
        DataSet dataSet2 = new DataSet( "name2", periodType2 );
 
343
 
 
344
        dataSetService.addDataSet( dataSet1 );
 
345
        dataSetService.addDataSet( dataSet2 );
 
346
 
 
347
        Source source1 = new DummySource( "Source1" );
 
348
        Source source2 = new DummySource( "Source2" );
 
349
        sourceStore.addSource( source1 );
 
350
        sourceStore.addSource( source2 );
 
351
 
 
352
        FrequencyOverrideAssociation association = new FrequencyOverrideAssociation( dataSet1, source1, periodType3 );
 
353
        
 
354
        dataSetService.addFrequencyOverrideAssociation( association );        
 
355
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet1 ).size(), 1 );
 
356
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source1 ).size(), 1 );
 
357
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet2 ).size(), 0 );
 
358
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source2 ).size(), 0 );
 
359
        
 
360
        dataSetService.addFrequencyOverrideAssociation( new FrequencyOverrideAssociation( dataSet1, source2, periodType3 ) );
 
361
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet1 ).size(), 2 );
 
362
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source1 ).size(), 1 );
 
363
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet2 ).size(), 0 );
 
364
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source2 ).size(), 1 );
 
365
        
 
366
        dataSetService.addFrequencyOverrideAssociation( new FrequencyOverrideAssociation( dataSet2, source1, periodType3 ) );
 
367
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet1 ).size(), 2 );
 
368
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source1 ).size(), 2 );
 
369
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet2 ).size(), 1 );
 
370
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source2 ).size(), 1 );
 
371
        
 
372
        dataSetService.removeFrequencyOverrideAssociation( association );
 
373
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet1 ).size(), 1 );
 
374
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source1 ).size(), 1 );
 
375
        assertEquals( dataSetService.getFrequencyOverrideAssociationsByDataSet( dataSet2 ).size(), 1 );
 
376
        assertEquals( dataSetService.getFrequencyOverrideAssociationsBySource( source2 ).size(), 1 );
 
377
    }
 
378
}