66import com .google .common .base .Splitter ;
77import de .muenchen .dave .domain .elasticsearch .MessquerschnittRandomFactory ;
88import de .muenchen .dave .domain .elasticsearch .MessstelleRandomFactory ;
9+ import de .muenchen .dave .domain .elasticsearch .detektor .Messfaehigkeit ;
910import de .muenchen .dave .domain .elasticsearch .detektor .Messquerschnitt ;
1011import de .muenchen .dave .domain .elasticsearch .detektor .Messstelle ;
12+ import de .muenchen .dave .domain .enums .Fahrzeugklasse ;
1113import de .muenchen .dave .domain .enums .MessstelleStatus ;
1214import de .muenchen .dave .domain .mapper .detektor .MessstelleReceiverMapper ;
1315import de .muenchen .dave .domain .mapper .detektor .MessstelleReceiverMapperImpl ;
1618import java .util .ArrayList ;
1719import java .util .HashSet ;
1820import java .util .Set ;
21+ import java .util .stream .Collectors ;
1922import lombok .extern .slf4j .Slf4j ;
2023import org .apache .commons .collections4 .CollectionUtils ;
2124import org .apache .commons .lang3 .reflect .FieldUtils ;
@@ -70,6 +73,18 @@ void testCreateMessstelle() {
7073 expected .setMessquerschnitte (mapper .createMessquerschnitte (dto .getMessquerschnitte ()));
7174 expected .setMessfaehigkeiten (mapper .createMessfaehigkeit (dto .getMessfaehigkeiten ()));
7275
76+ final Set <Fahrzeugklasse > distinctFahrzeugklassenOfMessfaehigkeiten = expected .getMessfaehigkeiten ().stream ().map (Messfaehigkeit ::getFahrzeugklasse )
77+ .collect (Collectors .toSet ());
78+ if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .ACHT_PLUS_EINS )) {
79+ expected .setFahrzeugklasse (Fahrzeugklasse .ACHT_PLUS_EINS );
80+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .ZWEI_PLUS_EINS )) {
81+ expected .setFahrzeugklasse (Fahrzeugklasse .ZWEI_PLUS_EINS );
82+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .SUMME_KFZ )) {
83+ expected .setFahrzeugklasse (Fahrzeugklasse .SUMME_KFZ );
84+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .RAD )) {
85+ expected .setFahrzeugklasse (Fahrzeugklasse .RAD );
86+ }
87+
7388 final Messstelle messstelle = this .mapper .createMessstelle (dto , stadtbezirkMapper );
7489 Assertions .assertThat (messstelle )
7590 .isNotNull ()
@@ -124,7 +139,6 @@ void testUpdateMessstelle() {
124139 expected .setRealisierungsdatum (updatedData .getRealisierungsdatum ());
125140 expected .setAbbaudatum (updatedData .getAbbaudatum ());
126141 expected .setHersteller (updatedData .getHersteller ());
127- expected .setFahrzeugklasse (new FahrzeugklassenMapperImpl ().map (updatedData .getFahrzeugklasse ()));
128142 expected .setDetektierteVerkehrsarten (updatedData .getDetektierteVerkehrsarten ());
129143 expected .setSuchwoerter (new ArrayList <>());
130144 expected .getSuchwoerter ().addAll (bean .getCustomSuchwoerter ());
@@ -150,6 +164,18 @@ void testUpdateMessstelle() {
150164 expected .setMessquerschnitte (bean .getMessquerschnitte ());
151165 expected .setDatumLetztePlausibleMessung (bean .getDatumLetztePlausibleMessung ());
152166
167+ final Set <Fahrzeugklasse > distinctFahrzeugklassenOfMessfaehigkeiten = expected .getMessfaehigkeiten ().stream ().map (Messfaehigkeit ::getFahrzeugklasse )
168+ .collect (Collectors .toSet ());
169+ if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .ACHT_PLUS_EINS )) {
170+ expected .setFahrzeugklasse (Fahrzeugklasse .ACHT_PLUS_EINS );
171+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .ZWEI_PLUS_EINS )) {
172+ expected .setFahrzeugklasse (Fahrzeugklasse .ZWEI_PLUS_EINS );
173+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .SUMME_KFZ )) {
174+ expected .setFahrzeugklasse (Fahrzeugklasse .SUMME_KFZ );
175+ } else if (distinctFahrzeugklassenOfMessfaehigkeiten .contains (Fahrzeugklasse .RAD )) {
176+ expected .setFahrzeugklasse (Fahrzeugklasse .RAD );
177+ }
178+
153179 final StadtbezirkMapper stadtbezirkMapper = Mockito .mock (StadtbezirkMapper .class );
154180 when (stadtbezirkMapper .bezeichnungOf (any ())).thenReturn ("Schwabing-West" );
155181 final Messstelle actual = this .mapper .updateMessstelle (bean , updatedData , stadtbezirkMapper );
0 commit comments