/
CoverageDataReceiver.java
93 lines (82 loc) · 2.78 KB
/
CoverageDataReceiver.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
package org.jboss.arquillian.extension.jacoco.client;
import org.jacoco.core.data.ExecutionData;
import org.jacoco.core.data.ExecutionDataReader;
import org.jacoco.core.data.ExecutionDataStore;
import org.jacoco.core.data.IExecutionDataVisitor;
import org.jacoco.core.data.ISessionInfoVisitor;
import org.jacoco.core.data.SessionInfoStore;
import org.jboss.arquillian.core.api.annotation.Observes;
import org.jboss.arquillian.extension.jacoco.CoverageDataCommand;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.UUID;
public class CoverageDataReceiver
{
public void storeCoverageData(@Observes CoverageDataCommand coverageDataCommandEvent)
{
try
{
ExecutionDataStore dataStore = new ExecutionDataStore();
SessionInfoStore sessionStore = new SessionInfoStore();
read(new ByteArrayInputStream(coverageDataCommandEvent.getCoverageDate()), dataStore, sessionStore);
copyToAgentExecutionStore(dataStore);
coverageDataCommandEvent.setResult("SUCCESS");
}
catch (Exception e)
{
coverageDataCommandEvent.setResult("FAILURE");
e.printStackTrace();
}
}
private void copyToAgentExecutionStore(ExecutionDataStore dataStore) throws Exception
{
Field f = UUID.class.getDeclaredField("$jacocoAccess");
Object executor = f.get(null);
Method m = executor.getClass().getDeclaredMethod("getProbes", Object[].class);
m.setAccessible(true);
for (ExecutionData data : dataStore.getContents())
{
Object[] probeData = new Object[] { data.getId(), data.getName(), data.getProbes().length };
m.invoke(executor, new Object[] { probeData });
boolean[] resultData = (boolean[]) probeData[0];
for (int i = 0; i < data.getProbes().length; i++)
{
if (!resultData[i])
{
resultData[i] = data.getProbes()[i];
}
}
}
}
private void read(InputStream stream, IExecutionDataVisitor executionDataVisitor, ISessionInfoVisitor sessionVisitor)
{
ExecutionDataReader reader = new ExecutionDataReader(stream);
reader.setExecutionDataVisitor(executionDataVisitor);
reader.setSessionInfoVisitor(sessionVisitor);
try
{
reader.read();
}
catch (IOException e)
{
throw new RuntimeException("Could not read data to ExecutionDataStore from InputStream", e);
}
finally
{
if (stream != null)
{
try
{
stream.close();
}
catch (Exception e2)
{
e2.printStackTrace();
}
}
}
}
}