APRep_test.go 2.3 KB

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