APRep_test.go 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. package messages
  2. import (
  3. "encoding/hex"
  4. "testing"
  5. "time"
  6. "github.com/stretchr/testify/assert"
  7. "gopkg.in/jcmturner/gokrb5.v6/iana"
  8. "gopkg.in/jcmturner/gokrb5.v6/iana/msgtype"
  9. "gopkg.in/jcmturner/gokrb5.v6/testdata"
  10. )
  11. func TestUnmarshalAPRep(t *testing.T) {
  12. t.Parallel()
  13. var a APRep
  14. v := "encode_krb5_ap_rep"
  15. b, err := hex.DecodeString(testdata.TestVectors[v])
  16. if err != nil {
  17. t.Fatalf("Test vector read error of %s: %v\n", v, err)
  18. }
  19. err = a.Unmarshal(b)
  20. if err != nil {
  21. t.Fatalf("Unmarshal error of %s: %v\n", v, err)
  22. }
  23. assert.Equal(t, iana.PVNO, a.PVNO, "PVNO not as expected")
  24. assert.Equal(t, msgtype.KRB_AP_REP, a.MsgType, "MsgType is not as expected")
  25. assert.Equal(t, testdata.TEST_ETYPE, a.EncPart.EType, "Ticket encPart etype not as expected")
  26. assert.Equal(t, iana.PVNO, a.EncPart.KVNO, "Ticket encPart KVNO not as expected")
  27. assert.Equal(t, []byte(testdata.TEST_CIPHERTEXT), a.EncPart.Cipher, "Ticket encPart cipher not as expected")
  28. }
  29. func TestUnmarshalEncAPRepPart(t *testing.T) {
  30. t.Parallel()
  31. var a EncAPRepPart
  32. v := "encode_krb5_ap_rep_enc_part"
  33. b, err := hex.DecodeString(testdata.TestVectors[v])
  34. if err != nil {
  35. t.Fatalf("Test vector read error of %s: %v\n", v, err)
  36. }
  37. err = a.Unmarshal(b)
  38. if err != nil {
  39. t.Fatalf("Unmarshal error of %s: %v\n", v, err)
  40. }
  41. //Parse the test time value into a time.Time type
  42. tt, _ := time.Parse(testdata.TEST_TIME_FORMAT, testdata.TEST_TIME)
  43. assert.Equal(t, tt, a.CTime, "CTime not as expected")
  44. assert.Equal(t, 123456, a.Cusec, "Client microseconds not as expected")
  45. assert.Equal(t, int32(1), a.Subkey.KeyType, "Subkey type not as expected")
  46. assert.Equal(t, []byte("12345678"), a.Subkey.KeyValue, "Subkey value not as expected")
  47. assert.Equal(t, int64(17), a.SequenceNumber, "Sequence number not as expected")
  48. }
  49. func TestUnmarshalEncAPRepPart_optionalsNULL(t *testing.T) {
  50. t.Parallel()
  51. var a EncAPRepPart
  52. v := "encode_krb5_ap_rep_enc_part(optionalsNULL)"
  53. b, err := hex.DecodeString(testdata.TestVectors[v])
  54. if err != nil {
  55. t.Fatalf("Test vector read error of %s: %v\n", v, err)
  56. }
  57. err = a.Unmarshal(b)
  58. if err != nil {
  59. t.Fatalf("Unmarshal error of %s: %v\n", v, err)
  60. }
  61. //Parse the test time value into a time.Time type
  62. tt, _ := time.Parse(testdata.TEST_TIME_FORMAT, testdata.TEST_TIME)
  63. assert.Equal(t, tt, a.CTime, "CTime not as expected")
  64. assert.Equal(t, 123456, a.Cusec, "Client microseconds not as expected")
  65. }