Add null-checks for Map.get()

[deliverable/tracecompass.git] / lttng / org.eclipse.tracecompass.lttng2.control.ui.tests / src / org / eclipse / tracecompass / lttng2 / control / ui / tests / model / component / TraceControlPropertiesTest.java

diff --git a/lttng/org.eclipse.tracecompass.lttng2.control.ui.tests/src/org/eclipse/tracecompass/lttng2/control/ui/tests/model/component/TraceControlPropertiesTest.java b/lttng/org.eclipse.tracecompass.lttng2.control.ui.tests/src/org/eclipse/tracecompass/lttng2/control/ui/tests/model/component/TraceControlPropertiesTest.java
index aaa4d64bae419f25d06303d0a10a5e143d910155..b528dbdd8c37c337786f5e54a5e95b8f6bac2672 100644 (file)
--- a/lttng/org.eclipse.tracecompass.lttng2.control.ui.tests/src/org/eclipse/tracecompass/lttng2/control/ui/tests/model/component/TraceControlPropertiesTest.java
+++ b/lttng/org.eclipse.tracecompass.lttng2.control.ui.tests/src/org/eclipse/tracecompass/lttng2/control/ui/tests/model/component/TraceControlPropertiesTest.java
@@ -352,16 +352,19 @@ public class TraceControlPropertiesTest {
         //-------------------------------------------------------------------------
         event = (TraceEventComponent) domains[1].getChildren()[1].getChildren()[0];
         adapter = event.getAdapter(IPropertySource.class);
+        assertNotNull(adapter);
         eventSource = (TraceEventPropertySource) adapter;
         assertEquals("== TRACE_DEBUG_LINE", eventSource.getPropertyValue(TraceEventPropertySource.TRACE_EVENT_LOGLEVEL_PROPERTY_ID));
 
         event = (TraceEventComponent) domains[1].getChildren()[1].getChildren()[1];
         adapter = event.getAdapter(IPropertySource.class);
+        assertNotNull(adapter);
         eventSource = (TraceEventPropertySource) adapter;
         assertEquals("<= TRACE_INFO", eventSource.getPropertyValue(TraceEventPropertySource.TRACE_EVENT_LOGLEVEL_PROPERTY_ID));
 
         event = (TraceEventComponent) domains[1].getChildren()[1].getChildren()[2];
         adapter = event.getAdapter(IPropertySource.class);
+        assertNotNull(adapter);
         eventSource = (TraceEventPropertySource) adapter;
         assertEquals("TRACE_DEBUG_SYSTEM", eventSource.getPropertyValue(TraceEventPropertySource.TRACE_EVENT_LOGLEVEL_PROPERTY_ID));