fleg
/
hswro-alarm-bot
mirror of https://git.sr.ht/~michalr/hswro-alarm-bot
1
0
Fork 0
Code

Small simplifcation

This commit is contained in:
Michał Rudowicz 2024-02-18 12:38:24 +01:00
parent 7a9a973904
commit 1e89fea274
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -45,7 +45,7 @@ func (self TgSender) Send(msg GenericMessage) error {
} }
func sendTgMessage(tgEvents chan GenericMessage, msg string, chatId int64) { func sendTgMessage(tgEvents chan GenericMessage, msg string, chatId int64) {
tgEvents <- GenericMessage{TgChatId{chatId}, []MsgContent{MsgContent{msg}}} tgEvents <- GenericMessage{TgChatId{chatId}, []MsgContent{{msg}}}
} }
type RealSleeper struct { type RealSleeper struct {

12
sender_worker_test.go
View File

@ -37,18 +37,18 @@ func TestMessageThrottling(t *testing.T) {
mockSleeper := MockSleeper{nil, 0} mockSleeper := MockSleeper{nil, 0}
Consume(SendToTg(tgSenderWorker(testEvents, &wg, &mockSleeper, log.New(io.Discard, "", log.Ltime)), Consume(SendToTg(tgSenderWorker(testEvents, &wg, &mockSleeper, log.New(io.Discard, "", log.Ltime)),
&mockSender, &wg, log.New(io.Discard, "", log.Ltime))) &mockSender, &wg, log.New(io.Discard, "", log.Ltime)))
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{MsgContent{"test1"}}} testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test1"}}}
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{MsgContent{"test3"}}} testEvents <- GenericMessage{TgChatId{124}, []MsgContent{{"test3"}}}
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{MsgContent{"test2"}}} testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test2"}}}
testEvents <- GenericMessage{TgChatId{124}, []MsgContent{MsgContent{"test4"}}} testEvents <- GenericMessage{TgChatId{124}, []MsgContent{{"test4"}}}
assert.Equal(t, 1, mockSleeper.callCount) assert.Equal(t, 1, mockSleeper.callCount)
*mockSleeper.ch <- nil *mockSleeper.ch <- nil
assert.Equal(t, 1, mockSleeper.callCount) assert.Equal(t, 1, mockSleeper.callCount)
testEvents <- GenericMessage{TgChatId{123}, []MsgContent{MsgContent{"test5"}}} testEvents <- GenericMessage{TgChatId{123}, []MsgContent{{"test5"}}}
close(testEvents) close(testEvents)
wg.Wait() wg.Wait()
assert.Equal(t, 2, mockSleeper.callCount) assert.Equal(t, 2, mockSleeper.callCount)
assert.Len(t, mockSender.messages, 3) assert.Len(t, mockSender.messages, 3)
assert.Contains(t, mockSender.messages, GenericMessage{TgChatId{123}, []MsgContent{MsgContent{"test1"}, MsgContent{"test2"}}}) assert.Contains(t, mockSender.messages, GenericMessage{TgChatId{123}, []MsgContent{{"test1"}, {"test2"}}})
} }