59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package msg_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"gitea.dwysokinski.me/twhelp/core/internal/msg/internal/model"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/ThreeDotsLabs/watermill/message/subscriber"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"gitea.dwysokinski.me/twhelp/core/internal/domain"
|
|
|
|
"gitea.dwysokinski.me/twhelp/core/internal/msg"
|
|
)
|
|
|
|
func TestServerPublisher_CmdRefresh(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
marshaler := msg.GobMarshaler{}
|
|
pubSub := newPubSub(t)
|
|
|
|
payloads := []domain.RefreshServersCmdPayload{
|
|
{
|
|
URL: "https://host1.com",
|
|
VersionCode: "vc1",
|
|
},
|
|
{
|
|
URL: "https://host2.com",
|
|
VersionCode: "vc2",
|
|
},
|
|
}
|
|
|
|
msgs, err := pubSub.Subscribe(context.Background(), "servers.cmd.refresh")
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, msg.NewServerPublisher(pubSub, marshaler).CmdRefresh(context.Background(), payloads...))
|
|
|
|
receivedMsgs, _ := subscriber.BulkRead(msgs, len(payloads), time.Second)
|
|
require.Len(t, receivedMsgs, len(payloads))
|
|
for _, m := range receivedMsgs {
|
|
var received model.RefreshServersCmdPayload
|
|
assert.NoError(t, marshaler.Unmarshal(m, &received))
|
|
|
|
found := false
|
|
for _, payload := range payloads {
|
|
if payload.VersionCode == received.VersionCode && payload.URL == received.URL {
|
|
found = true
|
|
break
|
|
}
|
|
}
|
|
assert.True(t, found)
|
|
}
|
|
}
|