AuthorizationData_test.go 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. package types
  2. import (
  3. "encoding/hex"
  4. "fmt"
  5. "testing"
  6. "github.com/stretchr/testify/assert"
  7. "gopkg.in/jcmturner/gokrb5.v7/iana/adtype"
  8. "gopkg.in/jcmturner/gokrb5.v7/iana/nametype"
  9. "gopkg.in/jcmturner/gokrb5.v7/test/testdata"
  10. )
  11. func TestUnmarshalAuthorizationData(t *testing.T) {
  12. t.Parallel()
  13. var a AuthorizationData
  14. b, err := hex.DecodeString(testdata.MarshaledKRB5authorization_data)
  15. if err != nil {
  16. t.Fatalf("Test vector read error: %v", err)
  17. }
  18. err = a.Unmarshal(b)
  19. if err != nil {
  20. t.Fatalf("Unmarshal error: %v", err)
  21. }
  22. assert.Equal(t, 2, len(a), "Number of authorization data entries not as expected")
  23. for i, entry := range a {
  24. assert.Equal(t, adtype.ADIfRelevant, entry.ADType, fmt.Sprintf("Authorization data type of entry %d not as expected", i+1))
  25. assert.Equal(t, []byte("foobar"), entry.ADData, fmt.Sprintf("Authorization data of entry %d not as expected", i+1))
  26. }
  27. }
  28. func TestUnmarshalAuthorizationData_kdcissued(t *testing.T) {
  29. t.Parallel()
  30. var a ADKDCIssued
  31. b, err := hex.DecodeString(testdata.MarshaledKRB5ad_kdcissued)
  32. if err != nil {
  33. t.Fatalf("Test vector read error: %v", err)
  34. }
  35. err = a.Unmarshal(b)
  36. if err != nil {
  37. t.Fatalf("Unmarshal error: %v", err)
  38. }
  39. assert.Equal(t, int32(1), a.ADChecksum.CksumType, "Checksum type not as expected")
  40. assert.Equal(t, []byte("1234"), a.ADChecksum.Checksum, "Checksum not as expected")
  41. assert.Equal(t, testdata.TEST_REALM, a.IRealm, "Issuing realm not as expected")
  42. assert.Equal(t, nametype.KRB_NT_PRINCIPAL, a.Isname.NameType, "Issuing name type not as expected")
  43. assert.Equal(t, testdata.TEST_PRINCIPALNAME_NAMESTRING, a.Isname.NameString, "Issuing name string entries not as expected")
  44. assert.Equal(t, 2, len(a.Elements), "Number of authorization data elements not as expected")
  45. for i, ele := range a.Elements {
  46. assert.Equal(t, adtype.ADIfRelevant, ele.ADType, fmt.Sprintf("Authorization data type of element %d not as expected", i+1))
  47. assert.Equal(t, []byte(testdata.TEST_AUTHORIZATION_DATA_VALUE), ele.ADData, fmt.Sprintf("Authorization data of element %d not as expected", i+1))
  48. }
  49. }