dmaap-mediator-producer/internal/jobs/jobs_test.go

   1 // -
   2 //   ========================LICENSE_START=================================
   3 //   O-RAN-SC
   4 //   %%
   5 //   Copyright (C) 2021: Nordix Foundation
   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 jobs
  22
  23 import (
  24         "os"
  25         "path/filepath"
  26         "testing"
  27
  28         "github.com/stretchr/testify/require"
  29 )
  30
  31 const type1Schema = `{"title": "Type 1"}`
  32
  33 func TestGetTypes_filesOkShouldReturnSliceOfTypesAndProvideSupportedTypes(t *testing.T) {
  34         assertions := require.New(t)
  35         typesDir, err := os.MkdirTemp("", "configs")
  36         if err != nil {
  37                 t.Errorf("Unable to create temporary directory for types due to: %v", err)
  38         }
  39         defer os.RemoveAll(typesDir)
  40         typeDir = typesDir
  41         fname := filepath.Join(typesDir, "type1.json")
  42         if err = os.WriteFile(fname, []byte(type1Schema), 0666); err != nil {
  43                 t.Errorf("Unable to create temporary files for types due to: %v", err)
  44         }
  45         types, err := GetTypes()
  46         wantedType := Type{
  47                 TypeId: "type1",
  48                 Schema: type1Schema,
  49         }
  50         wantedTypes := []*Type{&wantedType}
  51         assertions.EqualValues(wantedTypes, types)
  52         assertions.Nil(err)
  53
  54         supportedTypes := GetSupportedTypes()
  55         assertions.EqualValues([]string{"type1"}, supportedTypes)
  56 }
  57
  58 func TestAddJob_shouldAddJobToAllJobsMap(t *testing.T) {
  59         assertions := require.New(t)
  60         allJobs["type1"] = make(map[string]JobInfo)
  61         jobInfo := JobInfo{
  62                 Owner:            "owner",
  63                 LastUpdated:      "now",
  64                 InfoJobIdentity:  "job1",
  65                 TargetUri:        "target",
  66                 InfoJobData:      "{}",
  67                 InfoTypeIdentity: "type1",
  68         }
  69
  70         err := AddJob(jobInfo)
  71         assertions.Nil(err)
  72         assertions.Equal(1, len(allJobs["type1"]))
  73 }