github.com/filecoin-project/bacalhau@v0.3.23-0.20230228154132-45c989550ace/pkg/compute/capacity/parser_test.go (about)

     1  package capacity
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/filecoin-project/bacalhau/pkg/model"
     7  	"github.com/stretchr/testify/require"
     8  )
     9  
    10  func TestResourceUsageConfigParser(t *testing.T) {
    11  	testCases := []struct {
    12  		usageConfig  model.ResourceUsageConfig
    13  		expectedData model.ResourceUsageData
    14  	}{
    15  		{
    16  			model.ResourceUsageConfig{
    17  				CPU:    "100m",
    18  				Memory: "100Mi",
    19  				Disk:   "",
    20  				GPU:    "",
    21  			},
    22  			model.ResourceUsageData{
    23  				CPU:    0.1,               // 100m
    24  				Memory: 100 * 1024 * 1024, // 100Mi
    25  				Disk:   0,
    26  				GPU:    0,
    27  			},
    28  		},
    29  	}
    30  
    31  	for _, tc := range testCases {
    32  		data := ParseResourceUsageConfig(tc.usageConfig)
    33  		require.Equal(t, tc.expectedData, data)
    34  	}
    35  }