Paul Hinze 61d2c80813
Tests for the basis service
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.
2022-04-25 12:26:03 -05:00

57 lines
1.3 KiB
Go

package singleprocess
import (
"context"
"github.com/golang/protobuf/ptypes/empty"
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
)
func (s *service) UpsertBasis(
ctx context.Context,
req *vagrant_server.UpsertBasisRequest,
) (*vagrant_server.UpsertBasisResponse, error) {
result := req.Basis
if err := s.state.BasisPut(result); err != nil {
return nil, err
}
return &vagrant_server.UpsertBasisResponse{Basis: result}, nil
}
func (s *service) GetBasis(
ctx context.Context,
req *vagrant_server.GetBasisRequest,
) (*vagrant_server.GetBasisResponse, error) {
result, err := s.state.BasisGet(req.Basis)
if err != nil {
return nil, err
}
return &vagrant_server.GetBasisResponse{Basis: result}, nil
}
func (s *service) FindBasis(
ctx context.Context,
req *vagrant_server.FindBasisRequest,
) (*vagrant_server.FindBasisResponse, error) {
result, err := s.state.BasisFind(req.Basis)
if err != nil {
return nil, err
}
return &vagrant_server.FindBasisResponse{Basis: result}, nil
}
func (s *service) ListBasis(
ctx context.Context,
req *empty.Empty,
) (*vagrant_server.ListBasisResponse, error) {
result, err := s.state.BasisList()
if err != nil {
return nil, err
}
return &vagrant_server.ListBasisResponse{Basis: result}, nil
}