AuthorizationData_test.go 2.1 KB

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