fixed

Author: pashidlos

src/test/java/io/visual_regression_tracker/sdk_java/VisualRegressionTrackerTest.java

@@ -12,8 +12,8 @@
 import org.hamcrest.CoreMatchers;
 import org.hamcrest.MatcherAssert;
 import org.mockito.Mockito;
-import org.testng.annotations.AfterTest;
-import org.testng.annotations.BeforeTest;
+import org.testng.annotations.AfterMethod;
+import org.testng.annotations.BeforeMethod;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
@@ -24,14 +24,14 @@ public class VisualRegressionTrackerTest {
     MockWebServer server;
     VisualRegressionTracker vrt;
     VisualRegressionTrackerConfig config = new VisualRegressionTrackerConfig(
-            "http://localhost:4200",
+            "http://localhost",
             "733c148e-ef70-4e6d-9ae5-ab22263697cc",
             "XHGDZDFD3GMJDNM87JKEMP0JS1G5",
             "develop"
     );
 
     @SneakyThrows
-    @BeforeTest
+    @BeforeMethod
     void setup() {
         server = new MockWebServer();
         server.start();
@@ -42,7 +42,7 @@ void setup() {
     }
 
     @SneakyThrows
-    @AfterTest
+    @AfterMethod
     void tearDown() {
         server.shutdown();
     }
@@ -59,7 +59,9 @@ void shouldStartBuild() throws IOException, InterruptedException {
                 .id(buildId)
                 .projectId(projectId)
                 .build();
-        server.enqueue(new MockResponse().setBody(gson.toJson(buildResponse)));
+        server.enqueue(new MockResponse()
+                .setResponseCode(200)
+                .setBody(gson.toJson(buildResponse)));
 
         vrt.startBuild();
 
@@ -130,6 +132,7 @@ void shouldSubmitTestRun() throws IOException, InterruptedException {
                 .build();
         server.enqueue(new MockResponse().setBody(gson.toJson(testRunResponse)));
         vrt.buildId = buildId;
+        vrt.projectId = projectId;
 
         TestRunResponse result = vrt.submitTestRun(name, imageBase64, testRunOptions);