t.Run("UpdateKey_WithError", func(t *testing.T) {
        softwareKeyManager.EXPECT().UpdateKey(gomock.Any(), gomock.Eq("set1"), gomock.Any()).Return(errors.New("test"))
        err := keyManager.UpdateKey(context.TODO(), "set1", nil)
        assert.Error(t, err, "test")
    })