add teams tests

internal/tests/client.go

@@ -3,10 +3,14 @@ package tests
 import (
 	"context"
 	"log"
+	"testing"
 	"time"
 
+	"evening_detective/internal/config"
+	"evening_detective/internal/services"
 	pb "evening_detective/proto"
 
+	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
 )
 
@@ -21,3 +25,33 @@ func getClient() (pb.EveningDetectiveClient, func() error) {
 func getContext() (context.Context, context.CancelFunc) {
 	return context.WithTimeout(context.Background(), time.Second)
 }
+
+func deleteTeams(t *testing.T) {
+	dbFilepath := config.GetDBFilepath()
+	repository, err := services.NewRepository(dbFilepath)
+	if err != nil {
+		assert.Nil(t, err, "подключение к базе")
+	}
+	defer repository.Close()
+
+	err = repository.DeleteAllTeams(context.Background())
+	assert.Nil(t, err, "команды удалены")
+}
+
+func createTeam(client pb.EveningDetectiveClient, name string) (*pb.AddTeamsRsp, error) {
+	ctx, cancel := getContext()
+	defer cancel()
+	req := &pb.AddTeamsReq{
+		Teams: []*pb.Team{
+			{Name: name},
+		},
+	}
+	return client.AddTeams(ctx, req)
+}
+
+func getTeams(client pb.EveningDetectiveClient) (*pb.GetTeamsRsp, error) {
+	ctx, cancel := getContext()
+	defer cancel()
+	req := &pb.GetTeamsReq{}
+	return client.GetTeams(ctx, req)
+}

internal/tests/teams_test.go

@@ -0,0 +1,79 @@
+package tests
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCreateTeam(t *testing.T) {
+	client, close := getClient()
+	defer close()
+
+	createTeamResp, err := createTeam(client, "Тестовая команда")
+	assert.Nil(t, err, "запрос отправлен успешно")
+	assert.Equal(t, len(createTeamResp.Teams), 1, "количество команд равно 1")
+	assert.Equal(t, createTeamResp.Teams[0].Name, "Тестовая команда")
+
+	getTeamsResp, err := getTeams(client)
+	assert.Nil(t, err, "запрос отправлен успешно")
+	assert.Equal(t, len(getTeamsResp.Teams), 1, "количество команд равно 1")
+	assert.Equal(t, getTeamsResp.Teams[0].Name, "Тестовая команда")
+
+	deleteTeams(t)
+}
+
+func TestCreateTeamWithEmptyName(t *testing.T) {
+	client, close := getClient()
+	defer close()
+
+	_, err := createTeam(client, "")
+	assert.NotNil(t, err, "запрос не удался")
+
+	deleteTeams(t)
+}
+
+func TestCreateTwoTeam(t *testing.T) {
+	client, close := getClient()
+	defer close()
+
+	_, err := createTeam(client, "Тестовая команда 1")
+	assert.Nil(t, err, "запрос отправлен успешно")
+
+	_, err = createTeam(client, "Тестовая команда 2")
+	assert.Nil(t, err, "запрос отправлен успешно")
+
+	getTeamsResp, err := getTeams(client)
+	assert.Nil(t, err, "запрос отправлен успешно")
+	assert.Equal(t, len(getTeamsResp.Teams), 2, "количество команд равно 2")
+	assert.Equal(t, getTeamsResp.Teams[0].Name, "Тестовая команда 1")
+	assert.Equal(t, getTeamsResp.Teams[1].Name, "Тестовая команда 2")
+
+	deleteTeams(t)
+}
+
+func TestCreateTwoEqTeam(t *testing.T) {
+	client, close := getClient()
+	defer close()
+
+	_, err := createTeam(client, "Тестовая команда")
+	assert.Nil(t, err, "запрос отправлен успешно")
+
+	_, err = createTeam(client, "Тестовая команда")
+	assert.NotNil(t, err, "запрос не удался")
+
+	deleteTeams(t)
+}
+
+func TestCreateTwoBadTeam(t *testing.T) {
+	client, close := getClient()
+	defer close()
+
+	_, err := createTeam(client, "Тестовая команда")
+	assert.Nil(t, err, "запрос отправлен успешно")
+
+	_, err = createTeam(client, "Тестовая команда ")
+	assert.NotNil(t, err, "запрос не удался")
+
+	deleteTeams(t)
+}