Newer
Older
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
/*
* This file is part of vospace-rest
* Copyright (C) 2021 Istituto Nazionale di Astrofisica
* SPDX-License-Identifier: GPL-3.0-or-later
*/
package it.inaf.oats.vospace;
import it.inaf.ia2.aa.data.User;
import it.inaf.oats.vospace.datamodel.Views;
import java.io.ByteArrayOutputStream;
import java.net.URI;
import java.util.Arrays;
import javax.servlet.http.HttpServletRequest;
import net.ivoa.xml.vospace.v2.Transfer;
import net.ivoa.xml.vospace.v2.View;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.fail;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import org.mockito.junit.jupiter.MockitoExtension;
import org.mockito.junit.jupiter.MockitoSettings;
import org.mockito.quality.Strictness;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.client.ClientHttpRequest;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.test.util.ReflectionTestUtils;
import org.springframework.web.client.RequestCallback;
import org.springframework.web.client.ResponseExtractor;
import org.springframework.web.client.RestTemplate;
@ExtendWith(MockitoExtension.class)
@MockitoSettings(strictness = Strictness.LENIENT)
public class FileServiceClientTest {
@Mock
private RestTemplate restTemplate;
@Mock
private HttpServletRequest request;
@InjectMocks
private FileServiceClient fileServiceClient;
@BeforeEach
public void setUp() {
ReflectionTestUtils.setField(fileServiceClient, "authority", "example.com!vospace");
ReflectionTestUtils.setField(fileServiceClient, "fileServiceUrl", "http://file-service");
}
@Test
public void testTarArchiveJob() {
testStartArchiveJob(Views.TAR_VIEW_URI);
}
@Test
public void testZipArchiveJob() {
testStartArchiveJob(Views.ZIP_VIEW_URI);
}
@Test
public void testInvalidArchiveJob() {
try {
testStartArchiveJob("foo");
fail("IllegalArgumentException was expected");
} catch (IllegalArgumentException ex) {
}
}
private void testStartArchiveJob(String viewUri) {
Transfer transfer = new Transfer();
transfer.setDirection("pullFromVoSpace");
transfer.setTarget(Arrays.asList("vos://example.com!vospace/file1", "vos://example.com!vospace/file2"));
View view = new View();
view.setUri(viewUri);
transfer.setView(view);
User user = mock(User.class);
when(user.getAccessToken()).thenReturn("<token>");
when(request.getUserPrincipal()).thenReturn(user);
doAnswer(invocation -> {
RequestCallback requestCallback = invocation.getArgument(2);
ClientHttpRequest mockedRequest = mock(ClientHttpRequest.class);
HttpHeaders mockedRequestHeaders = mock(HttpHeaders.class);
when(mockedRequest.getBody()).thenReturn(new ByteArrayOutputStream());
when(mockedRequest.getHeaders()).thenReturn(mockedRequestHeaders);
requestCallback.doWithRequest(mockedRequest);
ResponseExtractor responseExtractor = invocation.getArgument(3);
ClientHttpResponse mockedResponse = mock(ClientHttpResponse.class);
HttpHeaders mockedResponseHeaders = mock(HttpHeaders.class);
when(mockedResponseHeaders.getLocation()).thenReturn(new URI("http://file-service/archive/result"));
when(mockedResponse.getHeaders()).thenReturn(mockedResponseHeaders);
responseExtractor.extractData(mockedResponse);
return mockedResponseHeaders.getLocation().toString();
}).when(restTemplate).execute(eq("http://file-service/archive"), eq(HttpMethod.POST),
any(RequestCallback.class), any(ResponseExtractor.class), any(Object[].class));
String redirect = fileServiceClient.startArchiveJob(transfer, "job123");
assertEquals("http://file-service/archive/result", redirect);
}
}