2 * Copyright 2016-2017, Nokia Corporation
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 package org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.restapi;
19 import org.junit.Before;
20 import org.junit.Test;
21 import org.mockito.InjectMocks;
22 import org.mockito.Mockito;
23 import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.TestBase;
24 import org.springframework.http.HttpHeaders;
25 import org.springframework.http.MediaType;
27 import javax.servlet.ServletOutputStream;
29 import static org.mockito.Mockito.verify;
30 import static org.mockito.Mockito.when;
31 import static org.springframework.test.util.ReflectionTestUtils.setField;
34 public class TestSwaggerApi extends TestBase {
37 private SwaggerApi swaggerApi;
41 public void initMocks() throws Exception {
42 setField(SwaggerApi.class, "logger", logger);
46 * test swagger definition get
49 public void testSwaggerRetrieval() throws Exception {
50 byte[] bytes = new byte[]{1, 2};
51 when(selfRegistrationManager.getSwaggerApiDefinition()).thenReturn(bytes);
52 ServletOutputStream os = Mockito.mock(ServletOutputStream.class);
53 when(httpResponse.getOutputStream()).thenReturn(os);
55 swaggerApi.getSwaggerApiDefinition(httpResponse);
57 verify(httpResponse).addHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE);
58 verify(httpResponse).addHeader(HttpHeaders.CONTENT_LENGTH, Integer.toString(bytes.length));
59 verify(os).write(bytes);
60 verify(logger).info("REST: get swagger definition");