~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/expression/ExpressionStoreTest.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:
 
1
package org.hisp.dhis.expression;
 
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
import org.hisp.dhis.dataelement.DataElement;
 
36
import org.hisp.dhis.dataelement.DataElementService;
 
37
 
 
38
/**
 
39
 * @author Lars Helge Overland
 
40
 * @version $Id$
 
41
 */
 
42
public class ExpressionStoreTest
 
43
    extends DhisConvenienceTest
 
44
{
 
45
    private ExpressionStore expressionStore;
 
46
    
 
47
    private int dataElementIdA;    
 
48
    private int dataElementIdB;    
 
49
    private int dataElementIdC;    
 
50
    private int dataElementIdD;
 
51
    
 
52
    private String expressionA;
 
53
    private String expressionB;
 
54
    
 
55
    private String descriptionA;
 
56
    private String descriptionB;
 
57
    
 
58
    private Set<DataElement> dataElements = new HashSet<DataElement>();
 
59
    
 
60
    // -------------------------------------------------------------------------
 
61
    // Fixture
 
62
    // -------------------------------------------------------------------------
 
63
    
 
64
    public void setUpTest()
 
65
        throws Exception
 
66
    {
 
67
        expressionStore = (ExpressionStore) getBean( ExpressionStore.ID );
 
68
        
 
69
        dataElementService = (DataElementService) getBean( DataElementService.ID );
 
70
        
 
71
        DataElement dataElementA = createDataElement( 'A' );
 
72
        DataElement dataElementB = createDataElement( 'B' );
 
73
        DataElement dataElementC = createDataElement( 'C' );
 
74
        DataElement dataElementD = createDataElement( 'D' );        
 
75
        
 
76
        dataElementIdA = dataElementService.addDataElement( dataElementA );
 
77
        dataElementIdB = dataElementService.addDataElement( dataElementB );
 
78
        dataElementIdC = dataElementService.addDataElement( dataElementC );
 
79
        dataElementIdD = dataElementService.addDataElement( dataElementD );
 
80
        
 
81
        expressionA = "[" + dataElementIdA + "] + [" + dataElementIdB + "]";
 
82
        expressionB = "[" + dataElementIdC + "] - [" + dataElementIdD + "]";
 
83
        
 
84
        descriptionA = "Expression A";
 
85
        descriptionB = "Expression B";
 
86
                
 
87
        dataElements.add( dataElementA );
 
88
        dataElements.add( dataElementB );
 
89
        dataElements.add( dataElementC );
 
90
        dataElements.add( dataElementD );
 
91
    }
 
92
 
 
93
    // -------------------------------------------------------------------------
 
94
    // Tests
 
95
    // -------------------------------------------------------------------------
 
96
    
 
97
    public void testAddGetExpression()
 
98
    {
 
99
        Expression expr = new Expression( expressionA, descriptionA, dataElements );
 
100
        
 
101
        int id = expressionStore.addExpression( expr );
 
102
        
 
103
        expr = expressionStore.getExpression( id );
 
104
        
 
105
        assertEquals( expr.getExpression(), expressionA );
 
106
        assertEquals( expr.getDescription(), descriptionA );
 
107
        assertEquals( expr.getDataElementsInExpression(), dataElements );
 
108
    }
 
109
    
 
110
    public void testUpdateExpression()
 
111
    {
 
112
        Expression expr = new Expression( expressionA, descriptionA, dataElements );
 
113
        
 
114
        int id = expressionStore.addExpression( expr );
 
115
        
 
116
        expr = expressionStore.getExpression( id );
 
117
        
 
118
        assertEquals( expr.getExpression(), expressionA );
 
119
        assertEquals( expr.getDescription(), descriptionA );
 
120
        
 
121
        expr.setExpression( expressionB );
 
122
        expr.setDescription( descriptionB );
 
123
 
 
124
        expressionStore.updateExpression( expr );
 
125
 
 
126
        expr = expressionStore.getExpression( id );
 
127
        
 
128
        assertEquals( expr.getExpression(), expressionB );
 
129
        assertEquals( expr.getDescription(), descriptionB );
 
130
    }
 
131
    
 
132
    public void testDeleteExpression()
 
133
    {
 
134
        Expression exprA = new Expression( expressionA, descriptionA, dataElements );
 
135
        Expression exprB = new Expression( expressionB, descriptionB, dataElements );
 
136
        
 
137
        int idA = expressionStore.addExpression( exprA );
 
138
        int idB = expressionStore.addExpression( exprB );
 
139
        
 
140
        assertNotNull( expressionStore.getExpression( idA ) );
 
141
        assertNotNull( expressionStore.getExpression( idB ) );
 
142
        
 
143
        expressionStore.deleteExpression( exprA );
 
144
 
 
145
        assertNull( expressionStore.getExpression( idA ) );
 
146
        assertNotNull( expressionStore.getExpression( idB ) );
 
147
 
 
148
        expressionStore.deleteExpression( exprB );
 
149
 
 
150
        assertNull( expressionStore.getExpression( idA ) );
 
151
        assertNull( expressionStore.getExpression( idB ) );
 
152
    }
 
153
    
 
154
    public void testGetAllExpressions()
 
155
    {
 
156
        Expression exprA = new Expression( expressionA, descriptionA, dataElements );
 
157
        Expression exprB = new Expression( expressionB, descriptionB, dataElements );
 
158
        
 
159
        expressionStore.addExpression( exprA );
 
160
        expressionStore.addExpression( exprB );
 
161
        
 
162
        Collection<Expression> expressions = expressionStore.getAllExpressions();
 
163
                
 
164
        assertTrue( expressions.size() == 2 );
 
165
        assertTrue( expressions.contains( exprA ) );
 
166
        assertTrue( expressions.contains( exprB ) );
 
167
    }
 
168
}