Project

General

Profile

« Previous | Next » 

Revision dd189293

Added by Andreas Kohlbecker about 7 years ago

ref #6169 support for working sets in the service mock and DTO

View differences:

src/main/java/eu/etaxonomy/cdm/mock/RegistrationService.java
13 13
import java.util.HashSet;
14 14
import java.util.List;
15 15
import java.util.Map;
16
import java.util.Set;
16 17
import java.util.UUID;
17 18

  
18 19
import org.springframework.beans.factory.annotation.Autowired;
......
43 44
    private Map<UUID, Registration> registrationsByUUID = new HashMap<>();
44 45
    private Map<String, Registration> registrationsByRegID = new HashMap<>();
45 46
    private Map<String, RegistrationDTO> registrationDTOsByRegID = new HashMap<>();
47
    private Map<Integer, Set<RegistrationDTO>> registrationDTOsByCitationID = new HashMap<>();
46 48

  
47 49
    private Collection<CdmBase> cdmEntities = new HashSet<>();
48 50

  
......
87 89
        registrationsByUUID.put(reg.getUuid(), reg);
88 90
        registrationsByRegID.put(reg.getSpecificIdentifier(), reg);
89 91
        registrationDTOsByRegID.put(reg.getSpecificIdentifier(), dto);
92
        if(! registrationDTOsByCitationID.containsKey(dto.getCitationID())){
93
            registrationDTOsByCitationID.put(dto.getCitationID(), new HashSet<RegistrationDTO>());
94
        }
95
        registrationDTOsByCitationID.get(dto.getCitationID()).add(dto);
90 96
    }
91 97

  
92 98
    private void mergeBack(){
......
111 117
        return registrationDTOsByRegID.values();
112 118
    }
113 119

  
120
    public Map<Integer, Set<RegistrationDTO>> listDTOsByWorkingSet() {
121
        init();
122
        return registrationDTOsByCitationID;
123
    }
124

  
114 125
    /**
115 126
     * @param registrationID
116 127
     * @return

Also available in: Unified diff