Examples with TiaAgent com.teamscale.tia.client.TiaAgent used on opensource projects

Example 1 with TiaAgent

use of com.teamscale.tia.client.TiaAgent in project teamscale-jacoco-agent by cqse.

the class TestwiseCoverageAgentTest method testAccessViaTiaClientAndReportUploadToTeamscale.

@Test
public void testAccessViaTiaClientAndReportUploadToTeamscale() throws Exception {
    List<ClusteredTestDetails> availableTests = Arrays.asList(new ClusteredTestDetails("test1", "test1", "content", "cluster"), new ClusteredTestDetails("test2", "test2", "content", "cluster"));
    List<PrioritizableTestCluster> impactedClusters = Collections.singletonList(new PrioritizableTestCluster("cluster", Collections.singletonList(new PrioritizableTest("test2"))));
    when(client.getImpactedTests(any(), any(), any(), any(), anyBoolean(), anyBoolean(), anyBoolean())).thenReturn(Response.success(impactedClusters));
    when(reportGenerator.convert(any(File.class))).thenReturn(CoverageToTeamscaleStrategyTest.getDummyTestwiseCoverage("test2"));
    int port = PORT_COUNTER.incrementAndGet();
    AgentOptions options = mockOptions(port);
    when(options.createTempFile(any(), any())).thenReturn(new File(tempDir, "test"));
    new TestwiseCoverageAgent(options, null, reportGenerator);
    TiaAgent agent = new TiaAgent(false, HttpUrl.get("http://localhost:" + port));
    TestRunWithClusteredSuggestions testRun = agent.startTestRun(availableTests);
    assertThat(testRun.getPrioritizedClusters()).hasSize(1);
    assertThat(testRun.getPrioritizedClusters().get(0).tests).hasSize(1);
    PrioritizableTest test = testRun.getPrioritizedClusters().get(0).tests.get(0);
    assertThat(test.uniformPath).isEqualTo("test2");
    RunningTest runningTest = testRun.startTest(test.uniformPath);
    runningTest.endTest(new TestRun.TestResultWithMessage(ETestExecutionResult.PASSED, "message"));
    testRun.endTestRun();
    verify(client).uploadReport(eq(EReportFormat.TESTWISE_COVERAGE), matches("\\Q{\"tests\":[{\"content\":\"content\",\"paths\":[],\"sourcePath\":\"test1\",\"uniformPath\":\"test1\"},{\"content\":\"content\",\"duration\":\\E[^,]*\\Q,\"message\":\"message\",\"paths\":[{\"files\":[{\"coveredLines\":\"1-4\",\"fileName\":\"Main.java\"}],\"path\":\"src/main/java\"}],\"result\":\"PASSED\",\"sourcePath\":\"test2\",\"uniformPath\":\"test2\"}]}\\E"), any(), any(), any(), any());
}

Example 2 with TiaAgent

use of com.teamscale.tia.client.TiaAgent in project teamscale-jacoco-agent by cqse.

the class CustomTestFramework method runTestsWithTia.

/**
 * Talks to the TIA agent and runs all impacted tests. Also uploads a coverage report at the end.
 */
public void runTestsWithTia() throws AgentHttpRequestFailedException {
    TiaAgent agent = new TiaAgent(false, HttpUrl.get("http://localhost:" + agentPort));
    TestRun testRun = agent.startTestRunWithoutTestSelection();
    for (String uniformPath : allTests.keySet()) {
        Runnable runnable = allTests.get(uniformPath);
        RunningTest runningTest = testRun.startTest(uniformPath);
        TestInfo testInfo;
        try {
            runnable.run();
            testInfo = runningTest.endTestAndRetrieveCoverage(new TestRun.TestResultWithMessage(ETestExecutionResult.PASSED, ""));
        } catch (Throwable t) {
            testInfo = runningTest.endTestAndRetrieveCoverage(new TestRun.TestResultWithMessage(ETestExecutionResult.FAILURE, t.getMessage()));
        }
        testInfos.add(testInfo);
    }
}