1
1
package org.hisp.dhis.organisationunit;
4
* Copyright (c) 2004-2010, University of Oslo
4
* Copyright (c) 2004-2012, University of Oslo
5
5
* All rights reserved.
7
7
* Redistribution and use in source and binary forms, with or without
44
44
public class OrganisationUnitStoreTest
45
45
extends DhisSpringTest
47
private OrganisationUnitStore organisationUnitStore;
47
private OrganisationUnitLevelStore organisationUnitLevelStore;
50
50
public void setUpTest()
52
organisationUnitStore = (OrganisationUnitStore) getBean( OrganisationUnitStore.ID );
52
organisationUnitLevelStore = (OrganisationUnitLevelStore) getBean( OrganisationUnitLevelStore.ID );
55
55
// -------------------------------------------------------------------------
56
56
// OrganisationUnitLevel
57
57
// -------------------------------------------------------------------------
62
62
OrganisationUnitLevel levelA = new OrganisationUnitLevel( 1, "National" );
63
63
OrganisationUnitLevel levelB = new OrganisationUnitLevel( 2, "District" );
65
int idA = organisationUnitStore.addOrganisationUnitLevel( levelA );
66
int idB = organisationUnitStore.addOrganisationUnitLevel( levelB );
68
assertEquals( levelA, organisationUnitStore.getOrganisationUnitLevel( idA ) );
69
assertEquals( levelB, organisationUnitStore.getOrganisationUnitLevel( idB ) );
65
int idA = organisationUnitLevelStore.save( levelA );
66
int idB = organisationUnitLevelStore.save( levelB );
68
assertEquals( levelA, organisationUnitLevelStore.get( idA ) );
69
assertEquals( levelB, organisationUnitLevelStore.get( idB ) );
75
75
OrganisationUnitLevel levelA = new OrganisationUnitLevel( 1, "National" );
76
76
OrganisationUnitLevel levelB = new OrganisationUnitLevel( 2, "District" );
78
organisationUnitStore.addOrganisationUnitLevel( levelA );
79
organisationUnitStore.addOrganisationUnitLevel( levelB );
81
Collection<OrganisationUnitLevel> actual = organisationUnitStore.getOrganisationUnitLevels();
78
organisationUnitLevelStore.save( levelA );
79
organisationUnitLevelStore.save( levelB );
81
Collection<OrganisationUnitLevel> actual = organisationUnitLevelStore.getAll();
83
83
assertNotNull( actual );
84
84
assertEquals( 2, actual.size() );
85
85
assertTrue( actual.contains( levelA ) );
92
92
OrganisationUnitLevel levelA = new OrganisationUnitLevel( 1, "National" );
93
93
OrganisationUnitLevel levelB = new OrganisationUnitLevel( 2, "District" );
95
int idA = organisationUnitStore.addOrganisationUnitLevel( levelA );
96
int idB = organisationUnitStore.addOrganisationUnitLevel( levelB );
98
assertNotNull( organisationUnitStore.getOrganisationUnitLevel( idA ) );
99
assertNotNull( organisationUnitStore.getOrganisationUnitLevel( idB ) );
101
organisationUnitStore.deleteOrganisationUnitLevel( levelA );
103
assertNull( organisationUnitStore.getOrganisationUnitLevel( idA ) );
104
assertNotNull( organisationUnitStore.getOrganisationUnitLevel( idB ) );
106
organisationUnitStore.deleteOrganisationUnitLevel( levelB );
108
assertNull( organisationUnitStore.getOrganisationUnitLevel( idA ) );
109
assertNull( organisationUnitStore.getOrganisationUnitLevel( idB ) );
95
int idA = organisationUnitLevelStore.save( levelA );
96
int idB = organisationUnitLevelStore.save( levelB );
98
assertNotNull( organisationUnitLevelStore.get( idA ) );
99
assertNotNull( organisationUnitLevelStore.get( idB ) );
101
organisationUnitLevelStore.delete( levelA );
103
assertNull( organisationUnitLevelStore.get( idA ) );
104
assertNotNull( organisationUnitLevelStore.get( idB ) );
106
organisationUnitLevelStore.delete( levelB );
108
assertNull( organisationUnitLevelStore.get( idA ) );
109
assertNull( organisationUnitLevelStore.get( idB ) );
b'\\ No newline at end of file'