AuthorizationData_test.go 2.0 KB

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