Add UT for REST interface & code refactor

[ric-plt/alarm-go.git] / adapter / cmd / restapi_test.go

diff --git a/adapter/cmd/restapi_test.go b/adapter/cmd/restapi_test.go
new file mode 100755 (executable)
index 0000000..0ef1d19
--- /dev/null
+++ b/adapter/cmd/restapi_test.go
@@ -0,0 +1,87 @@
+/*
+ *  Copyright (c) 2020 AT&T Intellectual Property.
+ *  Copyright (c) 2020 Nokia.
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ *
+ * This source code is part of the near-RT RIC (RAN Intelligent Controller)
+ * platform project (RICP).
+ */
+
+package main
+
+import (
+       "bytes"
+       "encoding/json"
+       "github.com/stretchr/testify/assert"
+       "net/http"
+       "net/http/httptest"
+       "testing"
+
+       "gerrit.o-ran-sc.org/r/ric-plt/alarm-go/alarm"
+)
+
+// Test cases
+func TestGetActiveAlarmsRESTInterface(t *testing.T) {
+       req, err := http.NewRequest("GET", "/ric/v1/alarms", nil)
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       rr := httptest.NewRecorder()
+       handler := http.HandlerFunc(alarmAdapter.GetActiveAlarms)
+       handler.ServeHTTP(rr, req)
+
+       assert.Equal(t, true, rr != nil)
+       assert.Equal(t, rr.Code, http.StatusOK)
+}
+
+func TestRaiseAlarmRESTInterface(t *testing.T) {
+       a := alarmer.NewAlarm(alarm.RIC_RT_DISTRIBUTION_FAILED, alarm.SeverityMajor, "Some App data", "eth 0 1")
+       b, err := json.Marshal(&a)
+       if err != nil {
+               t.Errorf("Unexpected error %v", err)
+       }
+
+       req, err := http.NewRequest("POST", "/ric/v1/alarms", bytes.NewBuffer(b))
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       rr := httptest.NewRecorder()
+       handler := http.HandlerFunc(alarmAdapter.RaiseAlarm)
+       handler.ServeHTTP(rr, req)
+
+       assert.True(t, rr != nil)
+       assert.Equal(t, rr.Code, http.StatusOK)
+}
+
+func TestClearAlarmRESTInterface(t *testing.T) {
+       a := alarmer.NewAlarm(alarm.RIC_RT_DISTRIBUTION_FAILED, alarm.SeverityMajor, "Some App data", "eth 0 1")
+       b, err := json.Marshal(&a)
+       if err != nil {
+               t.Errorf("Unexpected error %v", err)
+       }
+
+       req, err := http.NewRequest("POST", "/ric/v1/alarms", bytes.NewBuffer(b))
+       if err != nil {
+               t.Fatal(err)
+       }
+
+       rr := httptest.NewRecorder()
+       handler := http.HandlerFunc(alarmAdapter.ClearAlarm)
+       handler.ServeHTTP(rr, req)
+
+       assert.Equal(t, true, rr != nil)
+       assert.Equal(t, rr.Code, http.StatusOK)
+}