135 lines
2.6 KiB
Go
135 lines
2.6 KiB
Go
package output_test
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
//"git.jamestombleson.com/jtom38/newsbot-api/internal/database"
|
|
"git.jamestombleson.com/jtom38/newsbot-api/internal/entity"
|
|
"git.jamestombleson.com/jtom38/newsbot-api/internal/services/output"
|
|
"github.com/joho/godotenv"
|
|
)
|
|
|
|
var (
|
|
article entity.ArticleEntity = entity.ArticleEntity{
|
|
ID: 999,
|
|
SourceID: 1,
|
|
Tags: "unit, testing",
|
|
Title: "Demo",
|
|
Url: "https://github.com/jtom38/newsbot.collector.api",
|
|
Description: "Hello World",
|
|
}
|
|
blank string = ""
|
|
)
|
|
|
|
func TestDiscordMessageContainsTitle(t *testing.T) {
|
|
d := output.NewDiscordWebHookMessage(article)
|
|
msg, err := d.GeneratePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
for _, i := range *msg.Embeds {
|
|
if i.Title == &blank {
|
|
t.Error("title missing")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDiscordMessageContainsDescription(t *testing.T) {
|
|
d := output.NewDiscordWebHookMessage(article)
|
|
msg, err := d.GeneratePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
for _, i := range *msg.Embeds {
|
|
if i.Description == &blank {
|
|
t.Error("description missing")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDiscordMessageFooter(t *testing.T) {
|
|
d := output.NewDiscordWebHookMessage(article)
|
|
msg, err := d.GeneratePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
for _, i := range *msg.Embeds {
|
|
blank := ""
|
|
if i.Footer.Value == &blank {
|
|
t.Error("missing footer vlue")
|
|
}
|
|
if i.Footer.IconUrl == &blank {
|
|
t.Error("missing footer url")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDiscordMessageFields(t *testing.T) {
|
|
header := "Link"
|
|
d := output.NewDiscordWebHookMessage(article)
|
|
msg, err := d.GeneratePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
for _, embed := range *msg.Embeds {
|
|
for _, field := range embed.Fields {
|
|
var fName string
|
|
if field.Name != nil {
|
|
fName = *field.Name
|
|
} else {
|
|
t.Error("missing link field value")
|
|
}
|
|
|
|
if fName != header {
|
|
t.Error("missing link field key")
|
|
}
|
|
|
|
var fValue string
|
|
if field.Value != nil {
|
|
fValue = *field.Value
|
|
}
|
|
|
|
if fValue == blank {
|
|
t.Error("missing link field value")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// This test requires a env value to be present to work
|
|
func TestDiscordMessagePost(t *testing.T) {
|
|
_, err := os.Open(".env")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = godotenv.Load()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
res := os.Getenv("TESTS_DISCORD_WEBHOOK")
|
|
if res == "" {
|
|
t.Error("TESTS_DISCORD_WEBHOOK is missing")
|
|
}
|
|
endpoints := strings.Split(res, " ")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
d := output.NewDiscordWebHookMessage(article)
|
|
msg, err := d.GeneratePayload()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = d.SendPayload(msg, endpoints[0])
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|