Skip to content

Commit aa834b9

Browse files
Fixing SunsetAirFlightData reference and test
1 parent 3657671 commit aa834b9

File tree

2 files changed

+18
-5
lines changed

2 files changed

+18
-5
lines changed

solutions/16-merging-streams/src/main/java/flightimporter/FlightImporterJob.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ public static void main(String[] args) throws Exception {
4141
.setValueOnlyDeserializer(new JsonDeserializationSchema(SkyOneAirlinesFlightData.class))
4242
.build();
4343

44-
KafkaSource<SunsetAirFlightData> sunsetSource = KafkaSource.<SkyOneAirlinesFlightData>builder()
44+
KafkaSource<SunsetAirFlightData> sunsetSource = KafkaSource.<SunsetAirFlightData>builder()
4545
.setProperties(consumerConfig)
4646
.setTopics("sunset")
4747
.setStartingOffsets(OffsetsInitializer.latest())
@@ -87,4 +87,4 @@ public static DataStream<FlightData> defineWorkflow(DataStream<SkyOneAirlinesFli
8787

8888
return skyOneFlightStream.union(sunsetFlightStream);
8989
}
90-
}
90+
}

staging/16-merging-streams/src/test/java/models/SunsetAirFlightDataTest.java

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
77
import static org.apache.flink.types.PojoTestUtils.assertSerializedAsPojo;
88
import static org.junit.jupiter.api.Assertions.*;
99

10+
import java.time.ZoneId;
11+
import java.time.ZonedDateTime;
12+
1013
class SunsetAirFlightDataTest {
1114

1215
ObjectMapper mapper = new ObjectMapper().registerModule(new JavaTimeModule());
@@ -93,10 +96,20 @@ public void serializer_shouldSerializeAndDeserializeTheCorrectObject() throws Ex
9396

9497
@Test
9598
public void serializer_shouldHandleUnknownFields() throws Exception {
96-
String json = "{\"emailAddress\":\"[email protected]\",\"flightDepartureTime\":\"2023-10-16T22:25:00.000Z\",\"iataDepartureCode\":\"LAS\",\"flightArrivalTime\":\"2023-10-17T09:38:00.000Z\",\"iataArrivalCode\":\"BOS\",\"flightNumber\":\"SKY1522\",\"confirmation\":\"SKY1OUJUUK\",\"unknownField\":\"ignore\"}";
97-
99+
String json = "{\"customerEmailAddress\":\"[email protected]\",\"departureTime\":\"2023-10-16T22:25:00.000Z\",\"departureAirport\":\"LAS\",\"arrivalTime\":\"2023-10-17T09:38:00.000Z\",\"arrivalAirport\":\"BOS\",\"flightId\":\"SKY1522\",\"referenceNumber\":\"SKY1OUJUUK\",\"unknownField\":\"ignore\"}";
100+
98101
SunsetAirFlightData object = mapper.readValue(json, SunsetAirFlightData.class);
102+
103+
SunsetAirFlightData expected = new SunsetAirFlightData();
104+
expected.setCustomerEmailAddress("[email protected]");
105+
expected.setDepartureTime(ZonedDateTime.parse("2023-10-16T22:25:00.000Z").withZoneSameInstant(ZoneId.of("UTC")));
106+
expected.setDepartureAirport("LAS");
107+
expected.setArrivalTime(ZonedDateTime.parse("2023-10-17T09:38:00.000Z").withZoneSameInstant(ZoneId.of("UTC")));
108+
expected.setArrivalAirport("BOS");
109+
expected.setFlightId("SKY1522");
110+
expected.setReferenceNumber("SKY1OUJUUK");
99111

100112
assertInstanceOf(SunsetAirFlightData.class, object);
113+
assertEquals(expected, object);
101114
}
102-
}
115+
}

0 commit comments

Comments
 (0)