Skip to content

Commit b283ce1

Browse files
Merge pull request #397 from iExecBlockchainComputing/feature/computing-tasks-endpoint
Feature/computing tasks endpoint
2 parents cc426a6 + e2d90b6 commit b283ce1

File tree

4 files changed

+37
-6
lines changed

4 files changed

+37
-6
lines changed

src/main/java/com/iexec/core/worker/WorkerController.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -165,13 +165,13 @@ public ResponseEntity<PublicConfiguration> getPublicConfiguration() {
165165
}
166166

167167

168-
@GetMapping(path = "/workers/currenttasks")
169-
public ResponseEntity<List<String>> getTasksInProgress(@RequestHeader("Authorization") String bearerToken) {
168+
@GetMapping(path = "/workers/computing")
169+
public ResponseEntity<List<String>> getComputingTasks(@RequestHeader("Authorization") String bearerToken) {
170170
String workerWalletAddress = jwtTokenProvider.getWalletAddressFromBearerToken(bearerToken);
171171
if (workerWalletAddress.isEmpty()) {
172172
return ResponseEntity.status(HttpStatus.UNAUTHORIZED.value()).build();
173173
}
174-
return ok(workerService.getChainTaskIds(workerWalletAddress));
174+
return ok(workerService.getComputingTaskIds(workerWalletAddress));
175175
}
176176

177177
}

src/main/java/com/iexec/core/worker/WorkerService.java

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,15 @@ public List<String> getChainTaskIds(String walletAddress) {
137137
return Collections.emptyList();
138138
}
139139

140+
public List<String> getComputingTaskIds(String walletAddress) {
141+
Optional<Worker> optional = workerRepository.findByWalletAddress(walletAddress);
142+
if (optional.isPresent()) {
143+
Worker worker = optional.get();
144+
return worker.getComputingChainTaskIds();
145+
}
146+
return Collections.emptyList();
147+
}
148+
140149
public Optional<Worker> removeChainTaskIdFromWorker(String chainTaskId, String walletAddress) {
141150
Optional<Worker> optional = workerRepository.findByWalletAddress(walletAddress);
142151
if (optional.isPresent()) {

src/test/java/com/iexec/core/worker/WorkerControllerTests.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -250,9 +250,9 @@ public void shouldGetTasksInProgress() {
250250
List<String> list = List.of("t1", "t2");
251251
when(jwtTokenProvider.getWalletAddressFromBearerToken(TOKEN))
252252
.thenReturn(WALLET);
253-
when(workerService.getChainTaskIds(WALLET)).thenReturn(list);
253+
when(workerService.getComputingTaskIds(WALLET)).thenReturn(list);
254254
ResponseEntity<List<String>> response =
255-
workerController.getTasksInProgress(TOKEN);
255+
workerController.getComputingTasks(TOKEN);
256256
assertThat(response.getStatusCode()).isEqualTo(HttpStatus.OK);
257257
assertThat(response.getBody()).isEqualTo(list);
258258
}
@@ -261,7 +261,7 @@ public void shouldGetTasksInProgress() {
261261
public void shouldNotGetTasksInProgressSinceUnauthorized() {
262262
when(jwtTokenProvider.getWalletAddressFromBearerToken(TOKEN)).thenReturn("");
263263
ResponseEntity<List<String>> response =
264-
workerController.getTasksInProgress(TOKEN);
264+
workerController.getComputingTasks(TOKEN);
265265
assertThat(response.getStatusCode()).isEqualTo(HttpStatus.UNAUTHORIZED);
266266
}
267267

src/test/java/com/iexec/core/worker/WorkerServiceTests.java

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,28 @@ public void shouldGetEmptyChainTaskIdListSinceWorkerNotFound() {
305305
when(workerRepository.findByWalletAddress(wallet)).thenReturn(Optional.empty());
306306
assertThat(workerService.getChainTaskIds(wallet)).isEmpty();
307307
}
308+
309+
// Computing task IDs
310+
311+
@Test
312+
public void shouldGetComputingTaskIds() {
313+
String wallet = "wallet";
314+
List<String> list = List.of("t1", "t1");
315+
Worker worker = Worker.builder()
316+
.computingChainTaskIds(list)
317+
.build();
318+
when(workerRepository.findByWalletAddress(wallet)).thenReturn(Optional.of(worker));
319+
320+
assertThat(workerService.getComputingTaskIds(wallet)).isEqualTo(list);
321+
}
322+
323+
@Test
324+
public void shouldNotGetComputingTaskIdsSinceNoWorker() {
325+
String wallet = "wallet";
326+
when(workerRepository.findByWalletAddress(wallet)).thenReturn(Optional.empty());
327+
328+
assertThat(workerService.getComputingTaskIds(wallet)).isEmpty();
329+
}
308330

309331
// removeChainTaskIdFromWorker
310332

0 commit comments

Comments
 (0)