fix a test

This commit is contained in:
idk
2021-05-26 08:15:06 -04:00
parent a3b83b5e1e
commit cfc3cc97ca
3 changed files with 3 additions and 1 deletions

View File

@ -179,6 +179,7 @@ func ReadCertificate(data []byte) (certificate Certificate, remainder []byte, er
if err != nil { if err != nil {
switch err.Error() { switch err.Error() {
case "error parsing certificate length: certificate is too short": case "error parsing certificate length: certificate is too short":
certificate.CertLen = 0
return return
case "certificate parsing warning: certificate data is shorter than specified by length": case "certificate parsing warning: certificate data is shorter than specified by length":
//err = nil //err = nil

View File

@ -240,6 +240,7 @@ func ReadKeyCertificate(data []byte) (key_certificate KeyCertificate, err error)
"reason": "not enough data", "reason": "not enough data",
}).Error("error parsing key certificate") }).Error("error parsing key certificate")
err = errors.New("error parsing key certificate: not enough data") err = errors.New("error parsing key certificate: not enough data")
key_certificate.SPKType = Integer(data[:])
return return
} }
key_certificate.SPKType = Integer(data[:2]) key_certificate.SPKType = Integer(data[:2])

View File

@ -12,7 +12,7 @@ func TestCertificateWithMissingData(t *testing.T) {
data := make([]byte, 128+256) data := make([]byte, 128+256)
data = append(data, cert_data...) data = append(data, cert_data...)
keys_and_cert, _, err := ReadKeysAndCert(data) keys_and_cert, _, err := ReadKeysAndCert(data)
assert.Nil(err)
cert, err := keys_and_cert.GetCertificate() cert, err := keys_and_cert.GetCertificate()
if assert.NotNil(err) { if assert.NotNil(err) {
assert.Equal("certificate parsing warning: certificate data is shorter than specified by length", err.Error()) assert.Equal("certificate parsing warning: certificate data is shorter than specified by length", err.Error())