|
14 | 14 | import static hudson.tasks.test.TestResultTrendSeriesBuilder.*; |
15 | 15 |
|
16 | 16 | public class TestResultTrendChart { |
| 17 | + enum PassedColor {GREEN, BLUE} |
17 | 18 |
|
18 | 19 | public LinesChartModel create(final List<TrendTestResultSummary> results) { |
| 20 | + return create(results, PassedColor.BLUE); |
| 21 | + } |
| 22 | + |
| 23 | + public LinesChartModel create(final List<TrendTestResultSummary> results, final PassedColor passedColor) { |
19 | 24 | LinesDataSet dataset = new LinesDataSet(); |
20 | 25 | results.forEach(result -> dataset.add(result.getDisplayName(), result.toMap(), result.getBuildNumber())); |
21 | 26 |
|
22 | | - return getLinesChartModel(dataset); |
| 27 | + return getLinesChartModel(dataset, passedColor); |
23 | 28 | } |
24 | 29 |
|
25 | | - public LinesChartModel create(@NonNull final Iterable results, |
26 | | - final ChartModelConfiguration configuration) { |
| 30 | + public LinesChartModel create(@NonNull final Iterable results, final ChartModelConfiguration configuration) { |
| 31 | + return create(results, configuration, PassedColor.GREEN); |
| 32 | + } |
| 33 | + |
| 34 | + public LinesChartModel create(@NonNull final Iterable results, final ChartModelConfiguration configuration, |
| 35 | + final PassedColor passedColor) { |
27 | 36 | TestResultTrendSeriesBuilder builder = new TestResultTrendSeriesBuilder(); |
28 | 37 | LinesDataSet dataSet = builder.createDataSet(configuration, results); |
29 | 38 |
|
30 | | - return getLinesChartModel(dataSet); |
| 39 | + return getLinesChartModel(dataSet, passedColor); |
31 | 40 | } |
32 | 41 |
|
33 | 42 | public LinesChartModel createFromTestObject(final Iterable results, |
34 | | - final ChartModelConfiguration configuration) { |
| 43 | + final ChartModelConfiguration configuration) { |
| 44 | + return createFromTestObject(results, configuration, PassedColor.GREEN); |
| 45 | + } |
| 46 | + |
| 47 | + public LinesChartModel createFromTestObject(final Iterable results, final ChartModelConfiguration configuration, |
| 48 | + final PassedColor passedColor) { |
35 | 49 | TestObjectTrendSeriesBuilder builder = new TestObjectTrendSeriesBuilder(); |
36 | 50 | LinesDataSet dataSet = builder.createDataSet(configuration, results); |
37 | 51 |
|
38 | | - return getLinesChartModel(dataSet); |
| 52 | + return getLinesChartModel(dataSet, passedColor); |
39 | 53 | } |
40 | 54 |
|
41 | | - private LinesChartModel getLinesChartModel(final LinesDataSet dataSet) { |
| 55 | + private LinesChartModel getLinesChartModel(final LinesDataSet dataSet, final PassedColor passedColor) { |
42 | 56 | LinesChartModel model = new LinesChartModel(dataSet); |
43 | | - LineSeries failed = new LineSeries("Failed", Palette.RED.getNormal(), |
| 57 | + |
| 58 | + LineSeries passed = new LineSeries("Passed", |
| 59 | + passedColor == PassedColor.BLUE ? Palette.BLUE.getNormal() : Palette.GREEN.getNormal(), |
44 | 60 | LineSeries.StackedMode.STACKED, LineSeries.FilledMode.FILLED); |
45 | | - failed.addAll(dataSet.getSeries(FAILED_KEY)); |
46 | | - model.addSeries(failed); |
| 61 | + passed.addAll(dataSet.getSeries(PASSED_KEY)); |
| 62 | + model.addSeries(passed); |
47 | 63 |
|
48 | 64 | LineSeries skipped = new LineSeries("Skipped", Palette.GRAY.getNormal(), |
49 | 65 | LineSeries.StackedMode.STACKED, LineSeries.FilledMode.FILLED); |
50 | 66 | skipped.addAll(dataSet.getSeries(SKIPPED_KEY)); |
51 | 67 | model.addSeries(skipped); |
52 | 68 |
|
53 | | - LineSeries passed = new LineSeries("Passed", Palette.BLUE.getNormal(), |
| 69 | + LineSeries failed = new LineSeries("Failed", Palette.RED.getNormal(), |
54 | 70 | LineSeries.StackedMode.STACKED, LineSeries.FilledMode.FILLED); |
55 | | - passed.addAll(dataSet.getSeries(PASSED_KEY)); |
56 | | - model.addSeries(passed); |
| 71 | + failed.addAll(dataSet.getSeries(FAILED_KEY)); |
| 72 | + model.addSeries(failed); |
57 | 73 |
|
58 | 74 | return model; |
59 | 75 | } |
|
0 commit comments