Add AlertManager bind address
[ric-plt/vespamgr.git] / cmd / vesmgr / subscribexAPPNotifications_test.go
1 /*
2  *  Copyright (c) 2019 AT&T Intellectual Property.
3  *  Copyright (c) 2018-2019 Nokia.
4  *
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *     http://www.apache.org/licenses/LICENSE-2.0
10  *
11  *  Unless required by applicable law or agreed to in writing, software
12  *  distributed under the License is distributed on an "AS IS" BASIS,
13  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  *  See the License for the specific language governing permissions and
15  *  limitations under the License.
16  *
17  *  This source code is part of the near-RT RIC (RAN Intelligent Controller)
18  *  platform project (RICP).
19  *
20  */
21
22 package main
23
24 import (
25         "bytes"
26         "encoding/json"
27         "fmt"
28         "io/ioutil"
29         "net/http"
30         "net/http/httptest"
31         "testing"
32
33         "github.com/stretchr/testify/suite"
34 )
35
36 type AppmgrHTTPServerTestSuite struct {
37         suite.Suite
38         subscriptions chan subscriptionNotification
39         xappNotifURL  string
40 }
41
42 // suite setup
43 func (suite *AppmgrHTTPServerTestSuite) SetupSuite() {
44         // the url here is not actually used anywhere
45         suite.xappNotifURL = "http://127.0.0.1:8080" + vesmgrXappNotifPath
46         suite.subscriptions = make(chan subscriptionNotification)
47 }
48
49 // test setup
50 func (suite *AppmgrHTTPServerTestSuite) SetupTest() {
51         suite.subscriptions = make(chan subscriptionNotification)
52 }
53
54 func (suite *AppmgrHTTPServerTestSuite) TestSubscribexAppNotifications() {
55         testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
56                 body, _ := ioutil.ReadAll(req.Body)
57                 var result map[string]interface{}
58                 err := json.Unmarshal([]byte(body), &result)
59                 suite.Nil(err)
60                 data := result["Data"].(map[string]interface{})
61                 suite.Equal(5, int(data["maxRetries"].(float64)))
62                 suite.Equal(5, int(data["retryTimer"].(float64)))
63                 suite.Equal("all", data["eventType"].(string))
64                 suite.Equal("POST", req.Method)
65                 res.Header().Add("Content-Type", "application/json")
66                 res.WriteHeader(http.StatusCreated)
67                 res.Write([]byte(`{"id":"deadbeef1234567890", "version":0, "eventType":"all"}`))
68         }))
69         defer testServer.Close()
70
71         go subscribexAppNotifications(suite.xappNotifURL, suite.subscriptions, 1, testServer.URL)
72         isSubscribed := <-suite.subscriptions
73         suite.Nil(isSubscribed.err)
74         suite.Equal("deadbeef1234567890", isSubscribed.subsID)
75 }
76
77 func (suite *AppmgrHTTPServerTestSuite) TestSubscribexAppNotificationsWrongStatus() {
78         testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
79                 res.Header().Add("Content-Type", "application/json")
80                 res.WriteHeader(http.StatusUnauthorized)
81                 res.Write([]byte(`{"id":"deadbeef1234567890", "version":0, "eventType":"all"}`))
82         }))
83         defer testServer.Close()
84
85         requestBody := []byte(fmt.Sprintf(`{"maxRetries": 5, "retryTimer": 5, "eventType":"all", "targetUrl": "%v"}`, suite.xappNotifURL))
86         req, _ := http.NewRequest("POST", testServer.URL, bytes.NewBuffer(requestBody))
87         req.Header.Set("Content-Type", "application/json")
88         client := &http.Client{}
89
90         subsID, err := subscribexAppNotificationsClientDo(req, client)
91         suite.Equal(errWrongStatusCode, err)
92         // after failed POST vesmgr.appmgrSubsId holds an initial values
93         suite.Equal("", subsID)
94 }
95
96 func (suite *AppmgrHTTPServerTestSuite) TestSubscribexAppNotificationsWrongUrl() {
97         // use fake appmgrUrl that is not served in unit test
98         appmgrURL := "/I_do_not_exist/"
99         requestBody := []byte(fmt.Sprintf(`{"maxRetries": 5, "retryTimer": 5, "eventType":"all", "targetUrl": "%v"}`, suite.xappNotifURL))
100         req, _ := http.NewRequest("POST", appmgrURL, bytes.NewBuffer(requestBody))
101         req.Header.Set("Content-Type", "application/json")
102         client := &http.Client{}
103
104         subsID, err := subscribexAppNotificationsClientDo(req, client)
105         suite.Equal(errPostingFailed, err)
106         // after failed POST vesmgr.appmgrSubsId holds an initial values
107         suite.Equal("", subsID)
108 }
109
110 func (suite *AppmgrHTTPServerTestSuite) TestSubscribexAppNotificationsReadBodyFails() {
111         testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
112                 res.Header().Set("Content-Length", "1")
113                 res.Header().Add("Content-Type", "application/json")
114                 res.WriteHeader(http.StatusCreated)
115         }))
116         defer testServer.Close()
117
118         go subscribexAppNotifications(suite.xappNotifURL, suite.subscriptions, 1, testServer.URL)
119         isSubscribed := <-suite.subscriptions
120         suite.Equal("unexpected EOF", isSubscribed.err.Error())
121         suite.Equal("", isSubscribed.subsID)
122 }
123
124 func (suite *AppmgrHTTPServerTestSuite) TestSubscribexAppNotificationsUnMarshalFails() {
125         testServer := httptest.NewServer(http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
126                 res.Header().Add("Content-Type", "application/json")
127                 res.WriteHeader(http.StatusCreated)
128                 res.Write([]byte(`{""dump for UT": make(chan int),"}`))
129         }))
130         defer testServer.Close()
131
132         go subscribexAppNotifications(suite.xappNotifURL, suite.subscriptions, 1, testServer.URL)
133         isSubscribed := <-suite.subscriptions
134         suite.Equal("invalid character 'd' after object key", isSubscribed.err.Error())
135         suite.Equal("", isSubscribed.subsID)
136 }
137
138 func TestAppmgrHttpServerTestSuite(t *testing.T) {
139         suite.Run(t, new(AppmgrHTTPServerTestSuite))
140 }