manager/cmd/restapi_test.go

   1 /*
   2  *  Copyright (c) 2020 AT&T Intellectual Property.
   3  *  Copyright (c) 2020 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 package main
  22
  23 import (
  24         "bytes"
  25         "encoding/json"
  26         "net/http"
  27         "net/http/httptest"
  28         "testing"
  29
  30         "github.com/stretchr/testify/assert"
  31
  32         "gerrit.o-ran-sc.org/r/ric-plt/alarm-go/alarm"
  33 )
  34
  35 // Test cases
  36 func TestGetActiveAlarmsRESTInterface(t *testing.T) {
  37         req, err := http.NewRequest("GET", "/ric/v1/alarms", nil)
  38         if err != nil {
  39                 t.Fatal(err)
  40         }
  41
  42         rr := httptest.NewRecorder()
  43         handler := http.HandlerFunc(alarmManager.GetActiveAlarms)
  44         handler.ServeHTTP(rr, req)
  45
  46         assert.Equal(t, true, rr != nil)
  47         assert.Equal(t, rr.Code, http.StatusOK)
  48 }
  49
  50 func TestRaiseAlarmRESTInterface(t *testing.T) {
  51         a := alarmer.NewAlarm(alarm.RIC_RT_DISTRIBUTION_FAILED, alarm.SeverityMajor, "Some App data", "eth 0 1")
  52         b, err := json.Marshal(&a)
  53         if err != nil {
  54                 t.Errorf("Unexpected error %v", err)
  55         }
  56
  57         req, err := http.NewRequest("POST", "/ric/v1/alarms", bytes.NewBuffer(b))
  58         if err != nil {
  59                 t.Fatal(err)
  60         }
  61
  62         rr := httptest.NewRecorder()
  63         handler := http.HandlerFunc(alarmManager.RaiseAlarm)
  64         handler.ServeHTTP(rr, req)
  65
  66         assert.True(t, rr != nil)
  67         assert.Equal(t, rr.Code, http.StatusOK)
  68 }
  69
  70 func TestClearAlarmRESTInterface(t *testing.T) {
  71         a := alarmer.NewAlarm(alarm.RIC_RT_DISTRIBUTION_FAILED, alarm.SeverityMajor, "Some App data", "eth 0 1")
  72         b, err := json.Marshal(&a)
  73         if err != nil {
  74                 t.Errorf("Unexpected error %v", err)
  75         }
  76
  77         req, err := http.NewRequest("POST", "/ric/v1/alarms", bytes.NewBuffer(b))
  78         if err != nil {
  79                 t.Fatal(err)
  80         }
  81
  82         rr := httptest.NewRecorder()
  83         handler := http.HandlerFunc(alarmManager.ClearAlarm)
  84         handler.ServeHTTP(rr, req)
  85
  86         assert.Equal(t, true, rr != nil)
  87         assert.Equal(t, rr.Code, http.StatusOK)
  88 }
  89
  90 func TestSymptomDataHandler(t *testing.T) {
  91         req, err := http.NewRequest("POST", "/ric/v1/symptomdata", nil)
  92         if err != nil {
  93                 t.Fatal(err)
  94         }
  95
  96         rr := httptest.NewRecorder()
  97         handler := http.HandlerFunc(alarmManager.SymptomDataHandler)
  98         handler.ServeHTTP(rr, req)
  99
 100         assert.Equal(t, true, rr != nil)
 101         assert.Equal(t, rr.Code, http.StatusOK)
 102 }