wip

Author: nnqq

Dockerfile

@@ -1,6 +1,7 @@
 FROM golang:1.17-alpine AS build
 WORKDIR /app
 COPY / /app
+RUN go test ./...
 RUN go build -o servicebin cmd/main.go
 
 FROM alpine:latest

go.mod

@@ -3,27 +3,29 @@ module github.com/vahter-robot/backend
 go 1.17
 
 require (
+	github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible
 	github.com/kelseyhightower/envconfig v1.4.0
 	github.com/leaq-ru/lib-graceful v1.0.1
 	github.com/rs/zerolog v1.25.0
+	github.com/stretchr/testify v1.7.0
 	go.mongodb.org/mongo-driver v1.7.3
 	golang.org/x/sync v0.0.0-20210220032951-036812b2e83c
+	gopkg.in/tucnak/telebot.v2 v2.4.0
 )
 
 require (
+	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/go-stack/stack v1.8.0 // indirect
-	github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible // indirect
 	github.com/golang/snappy v0.0.1 // indirect
 	github.com/klauspost/compress v1.13.6 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
-	github.com/stretchr/testify v1.7.0 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/technoweenie/multipartstreamer v1.0.1 // indirect
 	github.com/xdg-go/pbkdf2 v1.0.0 // indirect
 	github.com/xdg-go/scram v1.0.2 // indirect
 	github.com/xdg-go/stringprep v1.0.2 // indirect
 	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
 	golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9 // indirect
 	golang.org/x/text v0.3.6 // indirect
-	gopkg.in/tucnak/telebot.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b // indirect
 )

go.sum

@@ -46,8 +46,10 @@ github.com/klauspost/compress v1.13.6 h1:P76CopJELS0TiO2mebmnzgWaajssP/EszplttgQ
 github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
 github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
+github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
 github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
 github.com/leaq-ru/lib-graceful v1.0.1 h1:8N6Gnz4wCkpLW4nDOTQ39EZU5DRrH/JmtpNdqI9XAyk=
 github.com/leaq-ru/lib-graceful v1.0.1/go.mod h1:F0n189lXn6CKNAM3B6lnuDdTDe5jzXNfoeqDa2ZRC9s=
@@ -142,6 +144,7 @@ golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8T
 golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE=
 golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
 gopkg.in/tucnak/telebot.v2 v2.4.0 h1:nOeqOWnOAD3dzbKW+NRumd8zjj5vrWwSa0WRTxvgfag=

pkg/child_bot/service.go

@@ -28,7 +28,7 @@ type service struct {
 	userRepo             *user.Repo
 	peerRepo             *peer.Repo
 	childBotRepo         *Repo
-	keywordsLimit        uint16
+	keywordsLimitPerBot  uint16
 	inLimitPerKeyword    uint16
 	inLimitChars         uint16
 	outLimitChars        uint16
@@ -59,7 +59,7 @@ func NewService(
 		userRepo:             userRepo,
 		peerRepo:             peerRepo,
 		childBotRepo:         childBotRepo,
-		keywordsLimit:        keywordsLimitPerBot,
+		keywordsLimitPerBot:  keywordsLimitPerBot,
 		inLimitPerKeyword:    inLimitPerKeyword,
 		inLimitChars:         inLimitChars,
 		outLimitChars:        outLimitChars,
@@ -576,7 +576,7 @@ func (s *service) handleOwnerGetKeywords(
 Банить
 %s
 
-%s`, len(bot.Keywords), s.keywordsLimit, keywords, help))
+%s`, len(bot.Keywords), s.keywordsLimitPerBot, keywords, help))
 	if err != nil {
 		return fmt.Errorf("s.reply: %w", err)
 	}
@@ -775,7 +775,7 @@ func ruToBool(s string) (bool, bool) {
 
 func (s *service) parseKeywordsAndMode(in string) ([]Keyword, mode, bool) {
 	words := strings.Split(in, delim)
-	if len(words)-1%3 != 0 || len(words) < 4 {
+	if (len(words)-1)%3 != 0 || len(words) < 4 {
 		return nil, 0, false
 	}
 
@@ -832,7 +832,7 @@ func (s *service) parseKeywordsAndMode(in string) ([]Keyword, mode, bool) {
 			Ban: ban,
 		})
 	}
-	if len(keywords) > int(s.keywordsLimit) {
+	if len(keywords) > int(s.keywordsLimitPerBot) {
 		return nil, 0, false
 	}
 

pkg/child_bot/service_test.go

@@ -0,0 +1,71 @@
+package child_bot
+
+import (
+	"github.com/stretchr/testify/assert"
+	"sort"
+	"testing"
+)
+
+func TestParseKeywordsAndModeOK(t *testing.T) {
+	s := &service{
+		keywordsLimitPerBot: 50,
+		inLimitPerKeyword:   25,
+		inLimitChars:        100,
+		outLimitChars:       1000,
+	}
+
+	kws, m, ok := s.parseKeywordsAndMode(`1
+===
+ваканс
+===
+Спасибо за предложение, но я не в поиске работы
+===
+да
+===
+реклама,прайс
+===
+Прайс на рекламу в канале:
+
+1) Стартапы и бизнес
+100 рублей
+
+Если цена устраивает, отправьте ссылку на ресурс который будем размещать
+===
+нет
+===
+сотруднич,партнер
+===
+Сотрудничество интересно, давайте обсудим
+===
+нет`)
+
+	assert.Len(t, kws, 3)
+
+	in1 := []string{"реклама", "прайс"}
+	sort.Strings(in1)
+	sort.Strings(kws[1].In)
+	in2 := []string{"сотруднич", "партнер"}
+	sort.Strings(in2)
+	sort.Strings(kws[2].In)
+	assert.Equal(t, []Keyword{{
+		In:  []string{"ваканс"},
+		Out: "Спасибо за предложение, но я не в поиске работы",
+		Ban: true,
+	}, {
+		In: in1,
+		Out: `Прайс на рекламу в канале:
+
+1) Стартапы и бизнес
+100 рублей
+
+Если цена устраивает, отправьте ссылку на ресурс который будем размещать`,
+		Ban: false,
+	}, {
+		In:  in2,
+		Out: "Сотрудничество интересно, давайте обсудим",
+		Ban: false,
+	}}, kws)
+
+	assert.Equal(t, OnlyFirst, m)
+	assert.Equal(t, ok, true)
+}