Fix failing DatadogBuildListenerTest.

jenkinsci · Jul 20, 2017 · 63d77af · 63d77af
1 parent 724ab8a
commit 63d77af
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/test/java/org/datadog/jenkins/plugins/datadog/DatadogBuildListenerTest.java b/src/test/java/org/datadog/jenkins/plugins/datadog/DatadogBuildListenerTest.java
@@ -2,12 +2,14 @@
 
 import hudson.EnvVars;
 import hudson.model.*;
+import jenkins.model.Jenkins;
 import net.sf.json.JSONArray;
 import net.sf.json.JSONObject;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
+import org.mockito.Mock;
 import org.powermock.api.mockito.PowerMockito;
 import org.powermock.core.classloader.annotations.PrepareForTest;
 import org.powermock.modules.junit4.PowerMockRunner;
@@ -22,21 +24,25 @@
 import static org.mockito.Mockito.*;
 
 @RunWith(PowerMockRunner.class)
-@PrepareForTest({DatadogHttpRequests.class, DatadogUtilities.class})
+@PrepareForTest({DatadogHttpRequests.class, DatadogUtilities.class, Jenkins.class})
 public class DatadogBuildListenerTest {
+    @Mock
+    private Jenkins jenkins;
 
     private DatadogBuildListener datadogBuildListener;
 
     @Before
     public void setUp() throws Exception {
-        datadogBuildListener = spy(new DatadogBuildListener());
+        PowerMockito.mockStatic(Jenkins.class);
+        PowerMockito.when(Jenkins.getInstance()).thenReturn(jenkins);
 
         PowerMockito.mockStatic(DatadogUtilities.class);
         when(DatadogUtilities.isJobTracked(anyString())).thenReturn(true);
         when(DatadogUtilities.assembleTags(any(JSONObject.class), any(HashMap.class))).thenReturn(new JSONArray());
 
         PowerMockito.mockStatic(DatadogHttpRequests.class);
 
+        datadogBuildListener = spy(new DatadogBuildListener());
         doReturn(descriptor()).when(datadogBuildListener).getDescriptor();
     }