object: implement SignedData() and AddSignKey() methods on IntegrityHeader

This commit is contained in:
Leonard Lyubich 2020-05-18 13:44:32 +03:00
parent ab796b81d2
commit 96a6bb4842
4 changed files with 64 additions and 0 deletions

View file

@ -219,3 +219,21 @@ func testData(t *testing.T, sz int) []byte {
return data
}
func TestIntegrityHeaderSignMethods(t *testing.T) {
// create new IntegrityHeader
s := new(IntegrityHeader)
// set test headers checksum
s.SetHeadersChecksum([]byte{1, 2, 3})
data, err := s.SignedData()
require.NoError(t, err)
require.Equal(t, data, s.GetHeadersChecksum())
// add signature
sig := []byte{4, 5, 6}
s.AddSignKey(sig, nil)
require.Equal(t, sig, s.GetSignature())
}