5cba783555d84b470da53fad2c8a9be57666e991
[policy/models.git] / models-sim / models-sim-dmaap / src / test / java / org / onap / policy / models / sim / dmaap / rest / BaseRestControllerV1Test.java
1 /*
2  * ============LICENSE_START=======================================================
3  * ONAP PAP
4  * ================================================================================
5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.models.sim.dmaap.rest;
22
23 import static org.junit.Assert.assertEquals;
24 import static org.junit.Assert.assertNotNull;
25
26 import java.util.UUID;
27 import javax.ws.rs.core.Response;
28 import javax.ws.rs.core.Response.ResponseBuilder;
29 import org.junit.Before;
30 import org.junit.Test;
31
32 public class BaseRestControllerV1Test {
33
34     private BaseRestControllerV1 ctlr;
35     private ResponseBuilder bldr;
36
37     @Before
38     public void setUp() {
39         ctlr = new BaseRestControllerV1();
40         bldr = Response.status(Response.Status.OK);
41     }
42
43     @Test
44     public void testAddVersionControlHeaders() {
45         Response resp = ctlr.addVersionControlHeaders(bldr).build();
46         assertEquals("0", resp.getHeaderString(BaseRestControllerV1.VERSION_MINOR_NAME));
47         assertEquals("0", resp.getHeaderString(BaseRestControllerV1.VERSION_PATCH_NAME));
48         assertEquals("1.0.0", resp.getHeaderString(BaseRestControllerV1.VERSION_LATEST_NAME));
49     }
50
51     @Test
52     public void testAddLoggingHeaders_Null() {
53         Response resp = ctlr.addLoggingHeaders(bldr, null).build();
54         assertNotNull(resp.getHeaderString(BaseRestControllerV1.REQUEST_ID_NAME));
55     }
56
57     @Test
58     public void testAddLoggingHeaders_NonNull() {
59         UUID uuid = UUID.randomUUID();
60         Response resp = ctlr.addLoggingHeaders(bldr, uuid).build();
61         assertEquals(uuid.toString(), resp.getHeaderString(BaseRestControllerV1.REQUEST_ID_NAME));
62     }
63 }