Use RestTemplate in AaiRestClient
[aai/model-loader.git] / src / test / java / org / onap / aai / modelloader / service / TestModelLoaderService.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 European Software Marketing Ltd.
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21 package org.onap.aai.modelloader.service;
22
23 import static org.hamcrest.CoreMatchers.is;
24 import static org.hamcrest.MatcherAssert.assertThat;
25 import static org.junit.Assert.assertTrue;
26
27 import java.io.IOException;
28 import java.util.Base64;
29
30 import javax.ws.rs.core.Response;
31
32 import org.junit.After;
33 import org.junit.Test;
34 import org.junit.runner.RunWith;
35 import org.onap.aai.modelloader.util.ArtifactTestUtils;
36 import org.springframework.beans.factory.annotation.Autowired;
37 import org.springframework.boot.test.context.SpringBootTest;
38 import org.springframework.test.context.TestPropertySource;
39 import org.springframework.test.context.junit4.SpringRunner;
40
41 /**
42  * Tests for the ModelLoaderService class.
43  *
44  */
45 @RunWith(SpringRunner.class)
46 @SpringBootTest(classes = {ModelLoaderService.class, MockBabelServiceClientFactory.class})
47 @TestPropertySource(properties = {"CONFIG_HOME=src/test/resources",})
48 public class TestModelLoaderService {
49
50     @Autowired
51     private ModelLoaderService service;
52
53     @After
54     public void shutdown() {
55         service.preShutdownOperations();
56     }
57
58     @Test
59     public void testMissingConfig() {
60         new ModelLoaderService().start();
61         assertTrue(true);
62     }
63
64     @Test
65     public void testLoadModel() {
66         Response response = service.loadModel("");
67         assertThat(response.getStatus(), is(Response.Status.OK.getStatusCode()));
68     }
69
70     @Test
71     public void testSaveModel() {
72         Response response = service.saveModel("", "");
73         assertThat(response.getStatus(), is(Response.Status.OK.getStatusCode()));
74     }
75
76     @Test
77     public void testIngestModel() throws IOException {
78         byte[] csarPayload = new ArtifactTestUtils().loadResource("compressedArtifacts/service-VscpaasTest-csar.csar");
79         Response response = service.ingestModel("model-name", "", Base64.getEncoder().encodeToString(csarPayload));
80         assertThat(response.getStatus(), is(Response.Status.OK.getStatusCode()));
81     }
82
83     @Test
84     public void testIngestModelMissingName() throws IOException {
85         byte[] csarPayload = new ArtifactTestUtils().loadResource("compressedArtifacts/service-VscpaasTest-csar.csar");
86         Response response = service.ingestModel("", "", Base64.getEncoder().encodeToString(csarPayload));
87         assertThat(response.getStatus(), is(Response.Status.INTERNAL_SERVER_ERROR.getStatusCode()));
88     }
89
90 }