pkg/resthooks/resthooks_test.go

   1 //==================================================================================
   2 //  Copyright (c) 2022 Samsung
   3 //
   4 //   Licensed under the Apache License, Version 2.0 (the "License");
   5 //   you may not use this file except in compliance with the License.
   6 //   You may obtain a copy of the License at
   7 //
   8 //       http://www.apache.org/licenses/LICENSE-2.0
   9 //
  10 //   Unless required by applicable law or agreed to in writing, software
  11 //   distributed under the License is distributed on an "AS IS" BASIS,
  12 //   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 //   See the License for the specific language governing permissions and
  14 //   limitations under the License.
  15 //
  16 //   This source code is part of the near-RT RIC (RAN Intelligent Controller)
  17 //   platform project (RICP).
  18 //==================================================================================
  19 //
  20 package resthooks
  21
  22 import (
  23         "os"
  24         "testing"
  25
  26         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/health"
  27         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/models"
  28         h "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/restapi/operations/health"
  29         "gerrit.o-ran-sc.org/r/ric-plt/ricdms/pkg/ricdms"
  30         "github.com/stretchr/testify/assert"
  31         "github.com/stretchr/testify/mock"
  32 )
  33
  34 var rh *Resthook
  35 var successStatus *models.Status
  36
  37 func TestMain(m *testing.M) {
  38
  39         successStatus = &models.Status{
  40                 Status: &health.HEALTHY,
  41         }
  42         ricdms.Init()
  43         rh = &Resthook{
  44                 HealthChecker: HealthCheckerMock{},
  45         }
  46         code := m.Run()
  47         os.Exit(code)
  48 }
  49
  50 func TestHealth(t *testing.T) {
  51         resp := rh.GetDMSHealth()
  52         switch resp.(type) {
  53         case *h.GetHealthCheckOK:
  54                 assert.Equal(t, successStatus, resp.(*h.GetHealthCheckOK).Payload)
  55
  56         case *h.GetHealthCheckInternalServerError:
  57                 assert.Fail(t, "Internal Server generated: %v", resp)
  58
  59         default:
  60                 assert.Fail(t, "Unknown type of resp : %v", resp)
  61         }
  62 }
  63
  64 type HealthCheckerMock struct {
  65         mock.Mock
  66 }
  67
  68 func (h HealthCheckerMock) GetStatus() *models.Status {
  69         return successStatus
  70 }