■ ■ ■ ■ ■ ■
plugin/src/test/java/com/google/tsunami/plugin/RemoteVulnDetectorImplTest.java
| skipped 82 lines |
83 | 83 | | @Test |
84 | 84 | | public void detect_withServingServer_returnsSuccessfulDetectionReportList() throws Exception { |
85 | 85 | | registerHealthCheckWithStatus(ServingStatus.SERVING); |
| 86 | + | registerSuccessfulRunService(); |
86 | 87 | | |
87 | 88 | | RemoteVulnDetector pluginToTest = getNewRemoteVulnDetectorInstance(); |
88 | 89 | | var endpointToTest = NetworkEndpointUtils.forIpAndPort("1.1.1.1", 80); |
| skipped 125 lines |
214 | 215 | | public void check( |
215 | 216 | | HealthCheckRequest request, StreamObserver<HealthCheckResponse> responseObserver) { |
216 | 217 | | responseObserver.onNext(HealthCheckResponse.newBuilder().setStatus(status).build()); |
| 218 | + | responseObserver.onCompleted(); |
| 219 | + | } |
| 220 | + | }); |
| 221 | + | } |
| 222 | + | |
| 223 | + | private void registerSuccessfulRunService() { |
| 224 | + | serviceRegistry.addService( |
| 225 | + | new PluginServiceImplBase() { |
| 226 | + | @Override |
| 227 | + | public void run(RunRequest request, StreamObserver<RunResponse> responseObserver) { |
| 228 | + | DetectionReportList.Builder reportListBuilder = DetectionReportList.newBuilder(); |
| 229 | + | for (MatchedPlugin plugin : request.getPluginsList()) { |
| 230 | + | reportListBuilder.addDetectionReports( |
| 231 | + | DetectionReport.newBuilder() |
| 232 | + | .setTargetInfo(request.getTarget()) |
| 233 | + | .setNetworkService(plugin.getServices(0))); |
| 234 | + | } |
| 235 | + | responseObserver.onNext(RunResponse.newBuilder().setReports(reportListBuilder).build()); |
217 | 236 | | responseObserver.onCompleted(); |
218 | 237 | | } |
219 | 238 | | }); |
| skipped 15 lines |