jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 1 | package com.android.tradefed.device.metric; |
| 2 | |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 3 | import static org.mockito.Mockito.times; |
| 4 | import static org.mockito.Mockito.verify; |
| 5 | |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 6 | import com.android.tradefed.config.OptionSetter; |
| 7 | import com.android.tradefed.device.ITestDevice; |
Julien Desprez | 1ea5eba | 2019-09-26 10:37:54 -0700 | [diff] [blame] | 8 | import com.android.tradefed.device.StubDevice; |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 9 | import com.android.tradefed.invoker.IInvocationContext; |
| 10 | import com.android.tradefed.invoker.InvocationContext; |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 11 | import com.android.tradefed.metrics.proto.MetricMeasurement.Metric; |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 12 | import com.android.tradefed.result.FileInputStreamSource; |
| 13 | import com.android.tradefed.result.ITestInvocationListener; |
| 14 | import com.android.tradefed.result.LogDataType; |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 15 | import com.android.tradefed.result.TestDescription; |
gopinath | d60f7dd | 2018-01-19 13:56:19 -0800 | [diff] [blame] | 16 | import com.android.tradefed.util.FileUtil; |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 17 | import com.android.tradefed.util.proto.TfMetricProtoUtil; |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 18 | |
| 19 | import org.junit.Before; |
| 20 | import org.junit.Test; |
| 21 | import org.junit.runner.RunWith; |
| 22 | import org.junit.runners.JUnit4; |
| 23 | import org.mockito.Mock; |
| 24 | import org.mockito.Mockito; |
| 25 | import org.mockito.MockitoAnnotations; |
| 26 | |
| 27 | import java.io.File; |
| 28 | import java.util.HashMap; |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 29 | |
| 30 | /** Unit tests for {@link FilePullerDeviceMetricCollector}. */ |
| 31 | @RunWith(JUnit4.class) |
| 32 | public class FilePullerDeviceMetricCollectorTest { |
| 33 | private FilePullerDeviceMetricCollector mFilePuller; |
| 34 | @Mock private ITestInvocationListener mMockListener; |
| 35 | @Mock private ITestDevice mMockDevice; |
Julien Desprez | 1ea5eba | 2019-09-26 10:37:54 -0700 | [diff] [blame] | 36 | @Mock private ITestDevice mStubDevice; |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 37 | private IInvocationContext mContext; |
| 38 | |
| 39 | @Before |
| 40 | public void setUp() { |
| 41 | MockitoAnnotations.initMocks(this); |
| 42 | mContext = new InvocationContext(); |
| 43 | mContext.addAllocatedDevice("default", mMockDevice); |
Julien Desprez | 1ea5eba | 2019-09-26 10:37:54 -0700 | [diff] [blame] | 44 | mContext.addAllocatedDevice("stub", mStubDevice); |
| 45 | Mockito.when(mStubDevice.getIDevice()).thenReturn(new StubDevice("serial-stub")); |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 46 | mFilePuller = |
| 47 | new FilePullerDeviceMetricCollector() { |
| 48 | @Override |
| 49 | public void processMetricFile( |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 50 | String key, File metricFile, DeviceMetricData data) { |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 51 | try (FileInputStreamSource source = new FileInputStreamSource(metricFile)) { |
| 52 | testLog(key, LogDataType.TEXT, source); |
| 53 | } |
| 54 | } |
gopinath | d60f7dd | 2018-01-19 13:56:19 -0800 | [diff] [blame] | 55 | @Override |
| 56 | public void processMetricDirectory( |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 57 | String key, File metricDirectory, DeviceMetricData data) { |
gopinath | d60f7dd | 2018-01-19 13:56:19 -0800 | [diff] [blame] | 58 | try (FileInputStreamSource source = new FileInputStreamSource( |
| 59 | metricDirectory)) { |
| 60 | testLog(key, LogDataType.TEXT, source); |
| 61 | } finally { |
| 62 | FileUtil.deleteFile(metricDirectory); |
| 63 | } |
| 64 | } |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 65 | }; |
| 66 | mFilePuller.init(mContext, mMockListener); |
| 67 | } |
| 68 | |
| 69 | /** Test when no keys have been requested, nothing should be queried anywhere. */ |
| 70 | @Test |
| 71 | public void testNoMatchingKey() { |
| 72 | mFilePuller.testRunStarted("fakeRun", 5); |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 73 | mFilePuller.testRunEnded(500, new HashMap<String, Metric>()); |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 74 | } |
| 75 | |
| 76 | /** |
| 77 | * Test when a file is found matching the key, then pulled and {@link |
| 78 | * FilePullerDeviceMetricCollector#processMetricFile(String, File, DeviceMetricData)} is called. |
| 79 | */ |
| 80 | @Test |
| 81 | public void testPullMatchingKey() throws Exception { |
| 82 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 83 | setter.setOptionValue("pull-pattern-keys", "coverageFile"); |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 84 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 85 | currentMetrics.put("coverageFile", TfMetricProtoUtil.stringToMetric("/data/coverage")); |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 86 | |
| 87 | Mockito.when(mMockDevice.pullFile(Mockito.eq("/data/coverage"))) |
| 88 | .thenReturn(new File("fake")); |
| 89 | |
| 90 | mFilePuller.testRunStarted("fakeRun", 5); |
| 91 | mFilePuller.testRunEnded(500, currentMetrics); |
| 92 | |
| 93 | Mockito.verify(mMockListener) |
| 94 | .testLog(Mockito.eq("coverageFile"), Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 95 | } |
| 96 | |
| 97 | /** |
Gopinath Elanchezhian | cfedcaf | 2020-04-27 13:08:23 -0700 | [diff] [blame^] | 98 | * Test when a multiple file is found matching the matching key, then pulled and {@link |
| 99 | * FilePullerDeviceMetricCollector#processMetricFile(String, File, DeviceMetricData)} is called |
| 100 | * multiple times. |
| 101 | */ |
| 102 | @Test |
| 103 | public void testPullMultipleMatchingKeyInMetrics() throws Exception { |
| 104 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 105 | setter.setOptionValue("pull-pattern-keys", "coverageFile"); |
| 106 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 107 | currentMetrics.put("coverageFile", TfMetricProtoUtil.stringToMetric("/data/coverage1")); |
| 108 | currentMetrics.put("coverageFileAnother", |
| 109 | TfMetricProtoUtil.stringToMetric("/data/coverage2")); |
| 110 | |
| 111 | Mockito.when(mMockDevice.pullFile(Mockito.eq("/data/coverage1"))) |
| 112 | .thenReturn(new File("fake1")); |
| 113 | Mockito.when(mMockDevice.pullFile(Mockito.eq("/data/coverage2"))) |
| 114 | .thenReturn(new File("fake2")); |
| 115 | |
| 116 | mFilePuller.testRunStarted("fakeRun", 5); |
| 117 | mFilePuller.testRunEnded(500, currentMetrics); |
| 118 | |
| 119 | Mockito.verify(mMockListener) |
| 120 | .testLog(Mockito.eq("coverageFile"), Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 121 | Mockito.verify(mMockListener) |
| 122 | .testLog(Mockito.eq("coverageFileAnother"), Mockito.eq(LogDataType.TEXT), |
| 123 | Mockito.any()); |
| 124 | } |
| 125 | |
| 126 | /** |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 127 | * Test when a file is found matching the key using a pattern matching, then pulled and {@link |
| 128 | * FilePullerDeviceMetricCollector#processMetricFile(String, File, DeviceMetricData)} is called. |
| 129 | */ |
| 130 | @Test |
| 131 | public void testPullMatchingKeyPattern() throws Exception { |
| 132 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 133 | // Using a pattern to find the file |
| 134 | setter.setOptionValue("pull-pattern-keys", "coverage.*"); |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 135 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 136 | currentMetrics.put("coverageFile", TfMetricProtoUtil.stringToMetric("/data/coverage")); |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 137 | |
| 138 | Mockito.when(mMockDevice.pullFile(Mockito.eq("/data/coverage"))) |
| 139 | .thenReturn(new File("fake")); |
| 140 | |
| 141 | mFilePuller.testRunStarted("fakeRun", 5); |
| 142 | mFilePuller.testRunEnded(500, currentMetrics); |
| 143 | |
| 144 | Mockito.verify(mMockListener) |
| 145 | .testLog(Mockito.eq("coverageFile"), Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 146 | } |
| 147 | |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 148 | /** |
| 149 | * Test {@link FilePullerDeviceMetricCollector#processMetricFile(String, File, |
Julien Desprez | 3e64ada | 2019-03-28 17:33:23 -0700 | [diff] [blame] | 150 | * DeviceMetricData)} is called on test case end and test run ended. |
gopinath | 2b4bd4c | 2018-12-03 11:15:32 -0800 | [diff] [blame] | 151 | */ |
| 152 | @Test |
| 153 | public void testMetricFileProcessingFlow() throws Exception { |
| 154 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 155 | setter.setOptionValue("pull-pattern-keys", "coverageFile"); |
| 156 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 157 | currentMetrics.put("coverageFile", TfMetricProtoUtil.stringToMetric("/data/coverage")); |
| 158 | |
| 159 | Mockito.when(mMockDevice.pullFile(Mockito.eq("/data/coverage"))) |
| 160 | .thenReturn(new File("fake")); |
| 161 | |
| 162 | TestDescription testDesc = new TestDescription("xyz", "abc"); |
| 163 | |
| 164 | mFilePuller.testRunStarted("fakeRun", 5); |
| 165 | mFilePuller.testStarted(testDesc); |
| 166 | mFilePuller.testEnded(testDesc, currentMetrics); |
| 167 | Mockito.verify(mMockListener) |
| 168 | .testLog(Mockito.eq("coverageFile"), Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 169 | verify(mMockListener, times(1)).testLog(Mockito.eq("coverageFile"), |
| 170 | Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 171 | mFilePuller.testRunEnded(500, currentMetrics); |
| 172 | verify(mMockListener, times(2)).testLog(Mockito.eq("coverageFile"), |
| 173 | Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 174 | |
| 175 | } |
| 176 | |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 177 | /** Test when a file exists in the metrics but the pattern searching does not match it. */ |
| 178 | @Test |
| 179 | public void testPatternNotMatching() throws Exception { |
| 180 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 181 | // Using a pattern to find the file but no file matches |
| 182 | setter.setOptionValue("pull-pattern-keys", "wrongPattern.*"); |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 183 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 184 | currentMetrics.put("coverageFile", TfMetricProtoUtil.stringToMetric("/data/coverage")); |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 185 | |
| 186 | mFilePuller.testRunStarted("fakeRun", 5); |
| 187 | mFilePuller.testRunEnded(500, currentMetrics); |
| 188 | // Nothing gets pulled from the device and logged. |
| 189 | Mockito.verify(mMockDevice, Mockito.times(0)).pullFile(Mockito.eq("/data/coverage")); |
| 190 | Mockito.verify(mMockListener, Mockito.times(0)) |
| 191 | .testLog(Mockito.eq("coverageFile"), Mockito.eq(LogDataType.TEXT), Mockito.any()); |
| 192 | } |
gopinath | d60f7dd | 2018-01-19 13:56:19 -0800 | [diff] [blame] | 193 | |
| 194 | /** |
| 195 | * Test when a directory is found matching the key, then pulled and {@link |
| 196 | * FilePullerDeviceMetricCollector#processMetricDirectory(String key, |
| 197 | * File metricDirectory, DeviceMetricData runData)} is called. |
| 198 | */ |
| 199 | @Test |
| 200 | public void testPullMatchingDirectory() throws Exception { |
| 201 | OptionSetter setter = new OptionSetter(mFilePuller); |
| 202 | setter.setOptionValue("directory-keys", "coverageDirectory"); |
Julien Desprez | 74089e1 | 2018-05-31 01:47:53 -0700 | [diff] [blame] | 203 | HashMap<String, Metric> currentMetrics = new HashMap<>(); |
| 204 | currentMetrics.put("coverageDirectory", TfMetricProtoUtil.stringToMetric("/data/coverage")); |
gopinath | d60f7dd | 2018-01-19 13:56:19 -0800 | [diff] [blame] | 205 | |
| 206 | Mockito.when(mMockDevice.pullDir(Mockito.eq("coverageDirectory"), |
| 207 | Mockito.any(File.class))).thenReturn(true); |
| 208 | |
| 209 | mFilePuller.testRunStarted("fakeRun", 5); |
| 210 | mFilePuller.testRunEnded(500, currentMetrics); |
| 211 | |
| 212 | Mockito.verify(mMockListener) |
| 213 | .testLog(Mockito.eq("coverageDirectory"), Mockito.eq(LogDataType.TEXT), |
| 214 | Mockito.any()); |
| 215 | } |
jdesprez | 7154090 | 2017-10-16 12:53:36 -0700 | [diff] [blame] | 216 | } |