github.com/versent/saml2aws@v2.17.0+incompatible/saml_test.go (about)

     1  package saml2aws
     2  
     3  import (
     4  	"io/ioutil"
     5  	"testing"
     6  
     7  	"github.com/stretchr/testify/assert"
     8  )
     9  
    10  func TestExtractAwsRoles(t *testing.T) {
    11  	data, err := ioutil.ReadFile("testdata/assertion.xml")
    12  	assert.Nil(t, err)
    13  
    14  	roles, err := ExtractAwsRoles(data)
    15  	assert.Nil(t, err)
    16  	assert.Len(t, roles, 2)
    17  }
    18  
    19  func TestExtractSessionDuration(t *testing.T) {
    20  	data, err := ioutil.ReadFile("testdata/assertion.xml")
    21  	assert.Nil(t, err)
    22  
    23  	duration, err := ExtractSessionDuration(data)
    24  	assert.Nil(t, err)
    25  	assert.Equal(t, int64(28800), duration)
    26  }