61d2c80813
Filling these out as a part of a sweep to ensure we return decent error messages from our GRPC services. For this service I didn't come up with any error condition that warranted changing the code yet. The get -> notfound case seems to be the main one and the pass through error object seems AOK here.
100 lines
2.5 KiB
Go
100 lines
2.5 KiB
Go
package singleprocess
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
|
|
"github.com/hashicorp/vagrant/internal/server"
|
|
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
func TestServiceBasis(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
t.Run("set and get", func(t *testing.T) {
|
|
require := require.New(t)
|
|
db := testDB(t)
|
|
impl, err := New(WithDB(db))
|
|
require.NoError(err)
|
|
client := server.TestServer(t, impl)
|
|
|
|
resp, err := client.UpsertBasis(ctx, &vagrant_server.UpsertBasisRequest{
|
|
Basis: &vagrant_server.Basis{
|
|
Name: "mybasis",
|
|
},
|
|
})
|
|
require.NoError(err)
|
|
require.NotNil(resp)
|
|
require.NotEmpty(resp.Basis.ResourceId)
|
|
require.Equal("mybasis", resp.Basis.Name)
|
|
|
|
getResp, err := client.GetBasis(ctx, &vagrant_server.GetBasisRequest{
|
|
Basis: &vagrant_plugin_sdk.Ref_Basis{
|
|
ResourceId: resp.Basis.ResourceId,
|
|
},
|
|
})
|
|
require.NoError(err)
|
|
require.NotNil(getResp)
|
|
require.Equal("mybasis", getResp.Basis.Name)
|
|
})
|
|
|
|
t.Run("find and list", func(t *testing.T) {
|
|
require := require.New(t)
|
|
db := testDB(t)
|
|
impl, err := New(WithDB(db))
|
|
require.NoError(err)
|
|
client := server.TestServer(t, impl)
|
|
|
|
// first insert
|
|
_, err = client.UpsertBasis(ctx, &vagrant_server.UpsertBasisRequest{
|
|
Basis: &vagrant_server.Basis{
|
|
Name: "mybasis",
|
|
},
|
|
})
|
|
require.NoError(err)
|
|
|
|
// then find it
|
|
findResp, err := client.FindBasis(ctx, &vagrant_server.FindBasisRequest{
|
|
Basis: &vagrant_server.Basis{
|
|
Name: "mybasis",
|
|
},
|
|
})
|
|
require.NoError(err)
|
|
require.NotNil(findResp)
|
|
require.Equal("mybasis", findResp.Basis.Name)
|
|
|
|
// then ensure it shows up in a list
|
|
listResp, err := client.ListBasis(ctx, &emptypb.Empty{})
|
|
require.NoError(err)
|
|
require.NotNil(listResp)
|
|
require.Len(listResp.Basis, 1)
|
|
})
|
|
|
|
t.Run("reasonable errors: get not found", func(t *testing.T) {
|
|
require := require.New(t)
|
|
db := testDB(t)
|
|
impl, err := New(WithDB(db))
|
|
require.NoError(err)
|
|
client := server.TestServer(t, impl)
|
|
|
|
_, err = client.GetBasis(ctx, &vagrant_server.GetBasisRequest{
|
|
Basis: &vagrant_plugin_sdk.Ref_Basis{
|
|
ResourceId: "idontexist",
|
|
},
|
|
})
|
|
require.Error(err)
|
|
|
|
// we expect this to be a GRPC error with a not found code and a decent
|
|
// message
|
|
st, ok := status.FromError(err)
|
|
require.Equal(ok, true)
|
|
require.Equal(st.Code(), codes.NotFound)
|
|
require.Contains(st.Message(), "not found")
|
|
})
|
|
}
|