Added some more JUnit tests

[deliverable/tracecompass.git] / org.eclipse.linuxtools.lttng.ui / src / org / eclipse / linuxtools / lttng / ui / views / control / dialogs / EnableEventsDialog.java

diff --git a/org.eclipse.linuxtools.lttng.ui/src/org/eclipse/linuxtools/lttng/ui/views/control/dialogs/EnableEventsDialog.java b/org.eclipse.linuxtools.lttng.ui/src/org/eclipse/linuxtools/lttng/ui/views/control/dialogs/EnableEventsDialog.java
index dbeb8586e0e1442cb527b6f34186c4c5406102cc..99bd3a20037af71083cfcd8cd09470136ba183f8 100644 (file)
--- a/org.eclipse.linuxtools.lttng.ui/src/org/eclipse/linuxtools/lttng/ui/views/control/dialogs/EnableEventsDialog.java
+++ b/org.eclipse.linuxtools.lttng.ui/src/org/eclipse/linuxtools/lttng/ui/views/control/dialogs/EnableEventsDialog.java
@@ -96,28 +96,11 @@ public class EnableEventsDialog extends Dialog implements IEnableEventsDialog  {
      * @param shell - a shell for the display of the dialog
      * @param providerGroup - the trace provider group
      */
-    public EnableEventsDialog(Shell shell, TraceProviderGroup providerGroup) {
-        this(shell, providerGroup, null);
-    }
-    
-    /**
-     * Constructor
-     * @param shell - a shell for the display of the dialog
-     * @param providerGroup - a trace provider group
-     * @param domain - a domain of the events (null if not known) 
-     */
-    public EnableEventsDialog(Shell shell, TraceProviderGroup providerGroup, TraceDomainComponent domain) {
+    public EnableEventsDialog(Shell shell) {
         super(shell);
-        fProviderGroup = providerGroup;
         setShellStyle(SWT.RESIZE);
-        fDomain = domain;
-        if (fDomain != null) {
-            fIsKernel = fDomain.isKernel();
-        } else {
-            fIsKernel = true;
-        }
     }
-
+    
     // ------------------------------------------------------------------------
     // Accessors
     // ------------------------------------------------------------------------
@@ -331,6 +314,29 @@ public class EnableEventsDialog extends Dialog implements IEnableEventsDialog  {
     public boolean isKernel() {
         return fIsKernel;
     }
+    
+    /*
+     * (non-Javadoc)
+     * @see org.eclipse.linuxtools.lttng.ui.views.control.dialogs.IEnableEventsDialog#setTraceProviderGroup(org.eclipse.linuxtools.lttng.ui.views.control.model.impl.TraceProviderGroup)
+     */
+    @Override
+    public void setTraceProviderGroup(TraceProviderGroup providerGroup) {
+        fProviderGroup = providerGroup;
+    }
+
+    /*
+     * (non-Javadoc)
+     * @see org.eclipse.linuxtools.lttng.ui.views.control.dialogs.IEnableEventsDialog#setTraceDomainComponent(org.eclipse.linuxtools.lttng.ui.views.control.model.impl.TraceDomainComponent)
+     */
+    @Override
+    public void setTraceDomainComponent(TraceDomainComponent domain) {
+        fDomain = domain;
+        if (fDomain != null) {
+            fIsKernel = fDomain.isKernel();
+        } else {
+            fIsKernel = true;
+        }
+    }
 
     // ------------------------------------------------------------------------
     // Operations
@@ -391,8 +397,11 @@ public class EnableEventsDialog extends Dialog implements IEnableEventsDialog  {
         // ------------------------------------------------------------------------
         // Kernel or UST event data group 
         // ------------------------------------------------------------------------
+        fUstComposite = null;
+        fKernelComposite = null;
         if (fIsKernel) {
             createKernelComposite();
+            fUstComposite = null;
         } else {
             createUstComposite();
         }