github.com/Venafi/vcert/v5@v5.10.2/pkg/venafi/firefly/fireflyUtil_test.go (about)

     1  package firefly
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/Venafi/vcert/v5/pkg/certificate"
     7  	"github.com/stretchr/testify/assert"
     8  )
     9  
    10  func TestGetRSASize(t *testing.T) {
    11  	t.Run("default", func(t *testing.T) {
    12  		rsaSizeGotten, err := GetRSASize(0)
    13  		assert.Nil(t, err)
    14  		assert.Equal(t, certificate.DefaultRSAlength, rsaSizeGotten)
    15  	})
    16  	t.Run("2048", func(t *testing.T) {
    17  		rsaSizeGotten, err := GetRSASize(2048)
    18  		assert.Nil(t, err)
    19  		assert.Equal(t, certificate.DefaultRSAlength, rsaSizeGotten)
    20  	})
    21  	t.Run("3072", func(t *testing.T) {
    22  		rsaSizeGotten, err := GetRSASize(3072)
    23  		assert.Nil(t, err)
    24  		assert.Equal(t, 3072, rsaSizeGotten)
    25  	})
    26  	t.Run("4096", func(t *testing.T) {
    27  		rsaSizeGotten, err := GetRSASize(4096)
    28  		assert.Nil(t, err)
    29  		assert.Equal(t, 4096, rsaSizeGotten)
    30  	})
    31  	t.Run("unsupported", func(t *testing.T) {
    32  		_, err := GetRSASize(1024)
    33  		if assert.Errorf(t, err, "I was expected an error but is nil") {
    34  			assert.ErrorContains(t, err, "key size 1024 is not supported. Valid RSA sizes for Firefly are ")
    35  		}
    36  	})
    37  }