More tests

Signed-off-by: Yogesh Deshpande <[email protected]>

comid/integregisters.go

@@ -37,6 +37,10 @@ func (i *IntegReg) AddDigests(index IRegisterIndex, digests Digests) error {
 
 // AddDigest allows inserting a Digest at a specific RegisterIndex
 func (i *IntegReg) AddDigest(index IRegisterIndex, digest swid.HashEntry) error {
+
+	if i.Register == nil {
+		return fmt.Errorf("no register to add digest")
+	}
 	switch t := index.(type) {
 	case string, uint, uint64:
 		i.Register[t] = append(i.Register[t], digest)

comid/integregisters_test.go

@@ -51,14 +51,26 @@ func TestIntegRegister_AddDigest_OK(t *testing.T) {
 }
 
 func TestIntegRegister_AddDigest_NOK(t *testing.T) {
-	expectedErr := `unexpected type for index: bool`
+	expectedErr := `no register to add digest`
+	register := IntegReg{}
+	err := register.AddDigest(uint(0), swid.HashEntry{})
+	assert.EqualError(t, err, expectedErr)
+
+	expectedErr = `unexpected type for index: bool`
 	var k bool
 	reg, err := prepareRegister("uint")
 	require.NoError(t, err)
 	err = reg.AddDigest(k, swid.HashEntry{})
 	assert.EqualError(t, err, expectedErr)
 }
 
+func TestIntegRegister_AddDigests_NOK(t *testing.T) {
+	expectedErr := `no digests to add`
+	register := IntegReg{}
+	err := register.AddDigests(uint(0), []swid.HashEntry{})
+	assert.EqualError(t, err, expectedErr)
+}
+
 func TestIntegRegister_UInt_MarshalCBOR_OK(t *testing.T) {
 	reg, err := prepareRegister("uint")
 	require.NoError(t, err)