mapper_workloads_test.go

Documentation: edge-infra.dev/pkg/edge/api/graph/mapper

     1  package mapper
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/assert"
     7  
     8  	appsv1 "k8s.io/api/apps/v1"
     9  	"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
    10  	"k8s.io/apimachinery/pkg/runtime"
    11  )
    12  
    13  // Test DeploymentToLivenessReadinessResponse
    14  func TestDeploymentToLivenessReadinessResponse(t *testing.T) {
    15  	model := DeploymentToLivenessReadinessResponse(testPodlist, testDeployment)
    16  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    17  }
    18  
    19  // Test StatefulSetToLivenessReadinessResponse
    20  func TestStatefulSetToLivenessReadinessResponse(t *testing.T) {
    21  	model := StatefulSetToLivenessReadinessResponse(testPodlist, testStatefulSet)
    22  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    23  }
    24  
    25  // Test DaemonSetToLivenessReadinessResponse
    26  func TestDeploymentToLivenessReadinessResponseDaemonSet(t *testing.T) {
    27  	model := DaemonSetToLivenessReadinessResponse(testPodlist, testDaemonSet)
    28  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    29  }
    30  
    31  // Test DaemonSetToLivenessReadinessResponse unstructured
    32  func TestUnstructuredToPodsList(t *testing.T) {
    33  	pod, err := runtime.DefaultUnstructuredConverter.
    34  		ToUnstructured(&testPodlist.Items[0])
    35  	assert.Equal(t, err, nil)
    36  	podList, err := UnstructuredToPodsList(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: pod}}})
    37  	assert.Equal(t, err, nil)
    38  	model := DaemonSetToLivenessReadinessResponse(podList, testDaemonSet)
    39  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    40  }
    41  
    42  // Test DeploymentToLivenessReadinessResponse unstructured
    43  func TestUnstructuredToDeploymentList(t *testing.T) {
    44  	dep, err := runtime.DefaultUnstructuredConverter.
    45  		ToUnstructured(testDeployment)
    46  	assert.Equal(t, err, nil)
    47  	depList, err := UnstructuredToDeploymentList(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: dep}}})
    48  	assert.Equal(t, err, nil)
    49  	model := DeploymentToLivenessReadinessResponse(testPodlist, &depList.Items[0])
    50  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    51  }
    52  
    53  // Test StatefulSetToLivenessReadinessResponse unstructured
    54  func TestUnstructuredToStatefulSetList(t *testing.T) {
    55  	ss, err := runtime.DefaultUnstructuredConverter.
    56  		ToUnstructured(testStatefulSet)
    57  	assert.Equal(t, err, nil)
    58  	ssList, err := UnstructuredToStatefulSetList(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: ss}}})
    59  	assert.Equal(t, err, nil)
    60  	model := StatefulSetToLivenessReadinessResponse(testPodlist, &ssList.Items[0])
    61  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    62  }
    63  
    64  // Test DaemonSetToLivenessReadinessResponse unstructured
    65  func TestUnstructuredToDaemonSetList(t *testing.T) {
    66  	ds, err := runtime.DefaultUnstructuredConverter.
    67  		ToUnstructured(testDeployment)
    68  	assert.Equal(t, err, nil)
    69  	dsList, err := UnstructuredToDaemonSetList(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: ds}}})
    70  	assert.Equal(t, err, nil)
    71  	model := DaemonSetToLivenessReadinessResponse(testPodlist, &dsList.Items[0])
    72  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    73  }
    74  
    75  // Test DeploymentToLivenessReadinessResponse unstructured object deployment
    76  func TestUnstructuredToObjectDeployment(t *testing.T) {
    77  	dep, err := runtime.DefaultUnstructuredConverter.
    78  		ToUnstructured(testDeployment)
    79  	assert.Equal(t, err, nil)
    80  	depRes := &appsv1.Deployment{}
    81  	err = UnstructuredToObject(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: dep}}}, depRes)
    82  	assert.Equal(t, err, nil)
    83  	model := DeploymentToLivenessReadinessResponse(testPodlist, depRes)
    84  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    85  }
    86  
    87  // Test StatefulSetToLivenessReadinessResponse unstructured object stateful set
    88  func TestUnstructuredToObjectStatefulSet(t *testing.T) {
    89  	ss, err := runtime.DefaultUnstructuredConverter.
    90  		ToUnstructured(testStatefulSet)
    91  	assert.Equal(t, err, nil)
    92  	ssRes := &appsv1.StatefulSet{}
    93  	err = UnstructuredToObject(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: ss}}}, ssRes)
    94  	assert.Equal(t, err, nil)
    95  	model := StatefulSetToLivenessReadinessResponse(testPodlist, ssRes)
    96  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
    97  }
    98  
    99  // Test DaemonSetToLivenessReadinessResponse unstructured daemon set
   100  func TestUnstructuredToObjectDaemonSet(t *testing.T) {
   101  	ds, err := runtime.DefaultUnstructuredConverter.
   102  		ToUnstructured(testDaemonSet)
   103  	assert.Equal(t, err, nil)
   104  	dsRes := &appsv1.DaemonSet{}
   105  	err = UnstructuredToObject(&unstructured.UnstructuredList{Items: []unstructured.Unstructured{{Object: ds}}}, dsRes)
   106  	assert.Equal(t, err, nil)
   107  	model := DaemonSetToLivenessReadinessResponse(testPodlist, dsRes)
   108  	ComapreLivenessReadinessModel(model, httpPort, conditionTime, t)
   109  }
   110
View as plain text