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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.node.verification.hardware.benchmarks;
import com.yahoo.vespa.hosted.node.verification.commons.parser.ParseResult;
import com.yahoo.vespa.hosted.node.verification.mock.MockCommandExecutor;
import org.junit.Before;
import org.junit.Test;
import java.util.List;
import java.util.Optional;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
/**
* @author sgrostad
* @author olaaun
*/
public class DiskBenchmarkTest {
private DiskBenchmark diskBenchmark;
private BenchmarkResults benchmarkResults;
private MockCommandExecutor commandExecutor;
private static final String VALID_OUTPUT_FILE = "src/test/java/com/yahoo/vespa/hosted/node/verification/hardware/resources/diskBenchmarkValidOutput";
private static final String INVALID_OUTPUT_FILE = "src/test/java/com/yahoo/vespa/hosted/node/verification/hardware/resources/diskBenchmarkInvalidOutput";
private static final double DELTA = 0.1;
@Before
public void setup() {
commandExecutor = new MockCommandExecutor();
benchmarkResults = new BenchmarkResults();
diskBenchmark = new DiskBenchmark(benchmarkResults, commandExecutor);
}
@Test
public void doBenchmark_should_store_diskSpeed_when_valid_output() {
String mockCommand = "cat " + VALID_OUTPUT_FILE;
commandExecutor.addCommand(mockCommand);
diskBenchmark.doBenchmark();
double expectedSpeed = 243;
double actualSpeed = benchmarkResults.getDiskSpeedMbs();
assertEquals(expectedSpeed, actualSpeed, DELTA);
}
@Test
public void doBenchmark_should_store_diskSpeed_as_zero_when_invalid_output() {
String mockCommand = "cat " + INVALID_OUTPUT_FILE;
commandExecutor.addCommand(mockCommand);
diskBenchmark.doBenchmark();
double expectedSpeed = 0;
double actualSpeed = benchmarkResults.getDiskSpeedMbs();
assertEquals(expectedSpeed, actualSpeed, DELTA);
}
@Test
public void parseDiskSpeed_valid_input() throws Exception {
List<String> mockCommandOutput = MockCommandExecutor.readFromFile(VALID_OUTPUT_FILE);
Optional<ParseResult> parseResult = diskBenchmark.parseDiskSpeed(mockCommandOutput);
ParseResult expectedParseResult = new ParseResult("MB/s", "243");
assertEquals(Optional.of(expectedParseResult), parseResult);
}
@Test
public void parseDiskSpeed_invalid_input() throws Exception {
List<String> mockCommandOutput = MockCommandExecutor.readFromFile(INVALID_OUTPUT_FILE);
Optional<ParseResult> parseResult = diskBenchmark.parseDiskSpeed(mockCommandOutput);
assertEquals(Optional.empty(), parseResult);
}
@Test
public void setDiskSpeed_valid_input() {
ParseResult parseResult = new ParseResult("MB/s", "243");
diskBenchmark.setDiskSpeed(Optional.of(parseResult));
double expectedDiskSpeed = 243;
assertEquals(expectedDiskSpeed, benchmarkResults.getDiskSpeedMbs(), DELTA);
}
@Test
public void setDiskSpeed_invalid_input() {
diskBenchmark.setDiskSpeed(Optional.empty());
double expectedDiskSpeed = 0;
assertEquals(expectedDiskSpeed, benchmarkResults.getDiskSpeedMbs(), DELTA);
}
@Test
public void getDiskSpeedInMBs_for_KBs_MBs_and_GBs() {
ParseResult KBsParseResult = new ParseResult("kB/s", "243000");
ParseResult MBsParseResult = new ParseResult("MB/s", "243");
ParseResult GBsParseResult = new ParseResult("GB/s", "0.243");
double expectedMBs = 243;
assertEquals(expectedMBs, diskBenchmark.getDiskSpeedInMBs(KBsParseResult), DELTA);
assertEquals(expectedMBs, diskBenchmark.getDiskSpeedInMBs(MBsParseResult), DELTA);
assertEquals(expectedMBs, diskBenchmark.getDiskSpeedInMBs(GBsParseResult), DELTA);
}
@Test
public void ckeckSpeedValidity_should_return_true_for_valid_format() {
String speed = "123";
assertTrue(diskBenchmark.checkSpeedValidity(speed));
speed = "30000";
assertTrue(diskBenchmark.checkSpeedValidity(speed));
speed = "6";
assertTrue(diskBenchmark.checkSpeedValidity(speed));
}
@Test
public void ckeckSpeedValidity_should_return_false_for_valid_format() {
String speed = "124 GHz";
assertFalse(diskBenchmark.checkSpeedValidity(speed));
speed = null;
assertFalse(diskBenchmark.checkSpeedValidity(speed));
speed = "This should return false as well";
assertFalse(diskBenchmark.checkSpeedValidity(speed));
}
@Test
public void convertToMbs_should_return_properly_converted_disk_speeds() {
String speed = "1234";
double factor = 1000;
double expectedSpeed = 1234000;
assertEquals(expectedSpeed, diskBenchmark.convertToMBs(speed, factor), DELTA);
factor = 1 / 1000.0;
expectedSpeed = 1.234;
assertEquals(expectedSpeed, diskBenchmark.convertToMBs(speed, factor), DELTA);
factor = 1;
expectedSpeed = 1234;
assertEquals(expectedSpeed, diskBenchmark.convertToMBs(speed, factor), DELTA);
}
}
|
