mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-12 11:15:12 +00:00
Groups Service (#40)
This commit is contained in:
ben-toogood
288
groups/handler/handler_test.go
Normal file
288
groups/handler/handler_test.go
Normal file
@@ -0,0 +1,288 @@
|
||||
package handler_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sort"
|
||||
"testing"
|
||||
|
||||
"github.com/google/uuid"
|
||||
"github.com/micro/services/groups/handler"
|
||||
pb "github.com/micro/services/groups/proto"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func testHandler(t *testing.T) *handler.Groups {
|
||||
// connect to the database
|
||||
db, err := gorm.Open(postgres.Open("postgresql://postgres@localhost:5432/groups?sslmode=disable"), &gorm.Config{})
|
||||
if err != nil {
|
||||
t.Fatalf("Error connecting to database: %v", err)
|
||||
}
|
||||
|
||||
// migrate the database
|
||||
if err := db.AutoMigrate(&handler.Group{}, &handler.Membership{}); err != nil {
|
||||
t.Fatalf("Error migrating database: %v", err)
|
||||
}
|
||||
|
||||
// clean any data from a previous run
|
||||
if err := db.Exec("TRUNCATE TABLE groups CASCADE").Error; err != nil {
|
||||
t.Fatalf("Error cleaning database: %v", err)
|
||||
}
|
||||
|
||||
return &handler.Groups{DB: db}
|
||||
}
|
||||
func TestCreate(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
t.Run("MissingName", func(t *testing.T) {
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{}, &pb.CreateResponse{})
|
||||
assert.Equal(t, handler.ErrMissingName, err)
|
||||
})
|
||||
|
||||
t.Run("Valid", func(t *testing.T) {
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Doe Family Group",
|
||||
}, &pb.CreateResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestUpdate(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
t.Run("MissingID", func(t *testing.T) {
|
||||
err := h.Update(context.TODO(), &pb.UpdateRequest{
|
||||
Name: "Doe Family Group",
|
||||
}, &pb.UpdateResponse{})
|
||||
assert.Equal(t, handler.ErrMissingID, err)
|
||||
})
|
||||
|
||||
t.Run("MissingName", func(t *testing.T) {
|
||||
err := h.Update(context.TODO(), &pb.UpdateRequest{
|
||||
Id: uuid.New().String(),
|
||||
}, &pb.UpdateResponse{})
|
||||
assert.Equal(t, handler.ErrMissingName, err)
|
||||
})
|
||||
|
||||
t.Run("NotFound", func(t *testing.T) {
|
||||
err := h.Update(context.TODO(), &pb.UpdateRequest{
|
||||
Id: uuid.New().String(),
|
||||
Name: "Bar Family Group",
|
||||
}, &pb.UpdateResponse{})
|
||||
assert.Equal(t, handler.ErrNotFound, err)
|
||||
})
|
||||
|
||||
t.Run("Valid", func(t *testing.T) {
|
||||
// create a demo group
|
||||
var cRsp pb.CreateResponse
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Doe Family Group",
|
||||
}, &cRsp)
|
||||
assert.NoError(t, err)
|
||||
|
||||
err = h.Update(context.TODO(), &pb.UpdateRequest{
|
||||
Id: cRsp.Group.Id,
|
||||
Name: "Bar Family Group",
|
||||
}, &pb.UpdateResponse{})
|
||||
assert.NoError(t, err)
|
||||
|
||||
var rRsp pb.ReadResponse
|
||||
err = h.Read(context.TODO(), &pb.ReadRequest{
|
||||
Ids: []string{cRsp.Group.Id},
|
||||
}, &rRsp)
|
||||
assert.NoError(t, err)
|
||||
|
||||
g := rRsp.Groups[cRsp.Group.Id]
|
||||
if g == nil {
|
||||
t.Errorf("Group not returned")
|
||||
} else {
|
||||
assert.Equal(t, "Bar Family Group", g.Name)
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestDelete(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
t.Run("MissingID", func(t *testing.T) {
|
||||
err := h.Delete(context.TODO(), &pb.DeleteRequest{}, &pb.DeleteResponse{})
|
||||
assert.Equal(t, handler.ErrMissingID, err)
|
||||
})
|
||||
|
||||
t.Run("NotFound", func(t *testing.T) {
|
||||
err := h.Delete(context.TODO(), &pb.DeleteRequest{
|
||||
Id: uuid.New().String(),
|
||||
}, &pb.DeleteResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
|
||||
// create a demo group
|
||||
var cRsp pb.CreateResponse
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Doe Family Group",
|
||||
}, &cRsp)
|
||||
assert.NoError(t, err)
|
||||
|
||||
t.Run("Valid", func(t *testing.T) {
|
||||
err := h.Delete(context.TODO(), &pb.DeleteRequest{
|
||||
Id: cRsp.Group.Id,
|
||||
}, &pb.DeleteResponse{})
|
||||
assert.NoError(t, err)
|
||||
|
||||
var rRsp pb.ReadResponse
|
||||
err = h.Read(context.TODO(), &pb.ReadRequest{
|
||||
Ids: []string{cRsp.Group.Id},
|
||||
}, &rRsp)
|
||||
assert.Nil(t, rRsp.Groups[cRsp.Group.Id])
|
||||
})
|
||||
}
|
||||
func TestList(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
// create two demo groups
|
||||
var cRsp1 pb.CreateResponse
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Alpha Group",
|
||||
}, &cRsp1)
|
||||
assert.NoError(t, err)
|
||||
|
||||
var cRsp2 pb.CreateResponse
|
||||
err = h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Bravo Group",
|
||||
}, &cRsp2)
|
||||
assert.NoError(t, err)
|
||||
|
||||
// add a member to the first group
|
||||
uid := uuid.New().String()
|
||||
err = h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
GroupId: cRsp1.Group.Id, MemberId: uid,
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.NoError(t, err)
|
||||
|
||||
t.Run("Unscoped", func(t *testing.T) {
|
||||
var rsp pb.ListResponse
|
||||
err = h.List(context.TODO(), &pb.ListRequest{}, &rsp)
|
||||
assert.NoError(t, err)
|
||||
assert.Lenf(t, rsp.Groups, 2, "Two groups should be returned")
|
||||
if len(rsp.Groups) != 2 {
|
||||
return
|
||||
}
|
||||
|
||||
sort.Slice(rsp.Groups, func(i, j int) bool {
|
||||
return rsp.Groups[i].Name < rsp.Groups[j].Name
|
||||
})
|
||||
assert.Equal(t, cRsp1.Group.Id, rsp.Groups[0].Id)
|
||||
assert.Equal(t, cRsp1.Group.Name, rsp.Groups[0].Name)
|
||||
assert.Len(t, rsp.Groups[0].MemberIds, 1)
|
||||
assert.Contains(t, rsp.Groups[0].MemberIds, uid)
|
||||
assert.Equal(t, cRsp2.Group.Id, rsp.Groups[1].Id)
|
||||
assert.Equal(t, cRsp2.Group.Name, rsp.Groups[1].Name)
|
||||
assert.Len(t, rsp.Groups[1].MemberIds, 0)
|
||||
})
|
||||
|
||||
t.Run("Scoped", func(t *testing.T) {
|
||||
var rsp pb.ListResponse
|
||||
err = h.List(context.TODO(), &pb.ListRequest{MemberId: uid}, &rsp)
|
||||
assert.NoError(t, err)
|
||||
assert.Lenf(t, rsp.Groups, 1, "One group should be returned")
|
||||
if len(rsp.Groups) != 1 {
|
||||
return
|
||||
}
|
||||
|
||||
assert.Equal(t, cRsp1.Group.Id, rsp.Groups[0].Id)
|
||||
assert.Equal(t, cRsp1.Group.Name, rsp.Groups[0].Name)
|
||||
assert.Len(t, rsp.Groups[0].MemberIds, 1)
|
||||
assert.Contains(t, rsp.Groups[0].MemberIds, uid)
|
||||
})
|
||||
}
|
||||
|
||||
func TestAddMember(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
t.Run("MissingGroupID", func(t *testing.T) {
|
||||
err := h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.Equal(t, handler.ErrMissingGroupID, err)
|
||||
})
|
||||
|
||||
t.Run("MissingMemberID", func(t *testing.T) {
|
||||
err := h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
GroupId: uuid.New().String(),
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.Equal(t, handler.ErrMissingMemberID, err)
|
||||
})
|
||||
|
||||
t.Run("GroupNotFound", func(t *testing.T) {
|
||||
err := h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
GroupId: uuid.New().String(),
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.Equal(t, handler.ErrNotFound, err)
|
||||
})
|
||||
|
||||
// create a test group
|
||||
var cRsp pb.CreateResponse
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Alpha Group",
|
||||
}, &cRsp)
|
||||
assert.NoError(t, err)
|
||||
|
||||
t.Run("Valid", func(t *testing.T) {
|
||||
err := h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
GroupId: cRsp.Group.Id,
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
|
||||
t.Run("Retry", func(t *testing.T) {
|
||||
err := h.AddMember(context.TODO(), &pb.AddMemberRequest{
|
||||
GroupId: cRsp.Group.Id,
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.AddMemberResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
}
|
||||
|
||||
func TestRemoveMember(t *testing.T) {
|
||||
h := testHandler(t)
|
||||
|
||||
t.Run("MissingGroupID", func(t *testing.T) {
|
||||
err := h.RemoveMember(context.TODO(), &pb.RemoveMemberRequest{
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.RemoveMemberResponse{})
|
||||
assert.Equal(t, handler.ErrMissingGroupID, err)
|
||||
})
|
||||
|
||||
t.Run("MissingMemberID", func(t *testing.T) {
|
||||
err := h.RemoveMember(context.TODO(), &pb.RemoveMemberRequest{
|
||||
GroupId: uuid.New().String(),
|
||||
}, &pb.RemoveMemberResponse{})
|
||||
assert.Equal(t, handler.ErrMissingMemberID, err)
|
||||
})
|
||||
|
||||
// create a test group
|
||||
var cRsp pb.CreateResponse
|
||||
err := h.Create(context.TODO(), &pb.CreateRequest{
|
||||
Name: "Alpha Group",
|
||||
}, &cRsp)
|
||||
assert.NoError(t, err)
|
||||
|
||||
t.Run("Valid", func(t *testing.T) {
|
||||
err := h.RemoveMember(context.TODO(), &pb.RemoveMemberRequest{
|
||||
GroupId: cRsp.Group.Id,
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.RemoveMemberResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
|
||||
t.Run("Retry", func(t *testing.T) {
|
||||
err := h.RemoveMember(context.TODO(), &pb.RemoveMemberRequest{
|
||||
GroupId: cRsp.Group.Id,
|
||||
MemberId: uuid.New().String(),
|
||||
}, &pb.RemoveMemberResponse{})
|
||||
assert.NoError(t, err)
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user