plugin_test.go

package main

import (
	"github.com/stretchr/testify/assert"

	"net/http"
	"net/http/httptest"
	"os"
	"testing"
)

func TestMissingLineConfig(t *testing.T) {
	var plugin Plugin

	err := plugin.Webhook()

	assert.NotNil(t, err)
}

func TestMissingChannelID(t *testing.T) {
	var plugin Plugin

	plugin.Config.ChannelToken = ""
	plugin.Config.ChannelSecret = ""

	err := plugin.Exec()

	assert.NotNil(t, err)
}

func TestMissingUserConfig(t *testing.T) {
	plugin := Plugin{
		Config: Config{
			ChannelToken:  "123456789",
			ChannelSecret: "test wrong id",
		},
	}

	err := plugin.Exec()

	assert.NotNil(t, err)
}

func TestSendTextError(t *testing.T) {
	plugin := Plugin{
		Repo: Repo{
			Name:     "go-hello",
			FullName: "appleboy",
		},
		Commit: Commit{
			Author:  "Bo-Yi Wu",
			Branch:  "master",
			Message: "update by drone line plugin.",
		},
		Build: Build{
			Number: 101,
			Status: "success",
			Link:   "https://github.com/appleboy/go-hello",
		},
		Config: Config{
			ChannelToken:  "1465486347",
			ChannelSecret: "ChannelSecret",
			To:            []string{"1234567890"},
			Message:       []string{"Test"},
		},
	}

	// enable message
	err := plugin.Exec()
	assert.Nil(t, err)
}

func TestSendRoomAndGroup(t *testing.T) {
	plugin := Plugin{
		Repo: Repo{
			Name:     "go-hello",
			FullName: "appleboy",
		},
		Commit: Commit{
			Author:  "Bo-Yi Wu",
			Branch:  "master",
			Message: "support send message to room or to group.",
		},
		Build: Build{
			Number: 102,
			Status: "failed",
			Link:   "https://github.com/appleboy/go-hello",
		},
		Config: Config{
			ChannelToken:  "1465486347",
			ChannelSecret: "ChannelSecret",
			ToRoom:        "1234567890",
			ToGroup:       "0123456789",
			Message:       []string{"Test"},
		},
	}

	// enable message
	err := plugin.Exec()
	assert.Nil(t, err)
}

func TestDefaultMessageFormat(t *testing.T) {
	plugin := Plugin{
		Repo: Repo{
			Name:     "go-hello",
			FullName: "appleboy",
		},
		Commit: Commit{
			Author:  "Bo-Yi Wu",
			Branch:  "master",
			Message: "update by drone line plugin.",
		},
		Build: Build{
			Number: 101,
			Status: "success",
			Link:   "https://github.com/appleboy/go-hello",
		},
	}

	message := plugin.Message(plugin.Repo, plugin.Build, plugin.Commit)

	assert.Equal(t, []string{"[success] <https://github.com/appleboy/go-hello> (master)『update by drone line plugin.』by Bo-Yi Wu"}, message)
}

func TestErrorSendMessage(t *testing.T) {
	plugin := Plugin{
		Repo: Repo{
			Name:     "go-hello",
			FullName: "appleboy",
		},
		Commit: Commit{
			Author:  "Bo-Yi Wu",
			Branch:  "master",
			Message: "update by drone line plugin.",
		},
		Build: Build{
			Number: 101,
			Status: "success",
			Link:   "https://github.com/appleboy/go-hello",
		},

		Config: Config{
			ChannelToken:  os.Getenv("LINE_CHANNEL_TOKEN"),
			ChannelSecret: os.Getenv("LINE_CHANNEL_SECRET"),
			ToRoom:        os.Getenv("LINE_TO"),
			Delimiter:     "::",
			Message:       []string{"Test Line Bot From Travis or Local", "commit message: 『{{ build.message }}』", " "},
			Image:         []string{"https://cdn3.iconfinder.com/data/icons/picons-social/57/16-apple-128.png"},
			Video:         []string{"https://www.sample-videos.com/video/mp4/480/big_buck_bunny_480p_5mb.mp4"},
			Audio:         []string{"https://feeds-tmp.soundcloud.com/stream/270161326-gotimefm-5-sarah-adams-on-test2doc-and-women-who-go.mp3::2920000", "http://feeds-tmp.soundcloud.com/stream/270161326-gotimefm-5-sarah-adams-on-test2doc-and-women-who-go.mp3"},
			Sticker:       []string{"1::1::100", "1"},
			Location:      []string{"竹北體育館::新竹縣竹北市::24.834687::120.993368", "1::1"},
		},
	}

	err := plugin.Exec()
	assert.Nil(t, err)

	plugin.Config.Message = []string{}
	err = plugin.Exec()
	assert.Nil(t, err)
}

func TestTrimElement(t *testing.T) {
	var input, result []string

	input = []string{"1", "     ", "3"}
	result = []string{"1", "3"}

	assert.Equal(t, result, trimElement(input))

	input = []string{"1", "2"}
	result = []string{"1", "2"}

	assert.Equal(t, result, trimElement(input))
}

func TestConvertImage(t *testing.T) {
	var input string
	var result []string

	input = "http://example.com/1.png"
	result = []string{"http://example.com/1.png", "http://example.com/1.png"}

	assert.Equal(t, result, convertImage(input, "::"))

	input = "http://example.com/1.png::http://example.com/2.png"
	result = []string{"http://example.com/1.png", "http://example.com/2.png"}

	assert.Equal(t, result, convertImage(input, "::"))

	input = "http://example.com/1.png@@http://example.com/2.png"
	result = []string{"http://example.com/1.png", "http://example.com/2.png"}

	assert.Equal(t, result, convertImage(input, "@@"))
}

func TestConvertVideo(t *testing.T) {
	var input string
	var result []string

	input = "http://example.com/1.mp4"
	result = []string{"http://example.com/1.mp4", defaultPreviewImageURL}

	assert.Equal(t, result, convertVideo(input, "::"))

	input = "http://example.com/1.mp4::http://example.com/2.png"
	result = []string{"http://example.com/1.mp4", "http://example.com/2.png"}

	assert.Equal(t, result, convertVideo(input, "::"))

	input = "http://example.com/1.mp4@@http://example.com/2.png"
	result = []string{"http://example.com/1.mp4", "http://example.com/2.png"}

	assert.Equal(t, result, convertVideo(input, "@@"))
}

func TestConvertAudio(t *testing.T) {
	var input string
	var result Audio
	var empty bool

	input = "http://example.com/1.mp3"
	result, empty = convertAudio(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, Audio{}, result)

	// strconv.ParseInt: parsing "我": invalid syntax
	input = "http://example.com/1.mp3::我"
	result, empty = convertAudio(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, Audio{}, result)

	input = "http://example.com/1.mp3::1000"
	result, empty = convertAudio(input, "::")

	assert.Equal(t, false, empty)
	assert.Equal(t, Audio{
		URL:      "http://example.com/1.mp3",
		Duration: 1000,
	}, result)
}

func TestConvertSticker(t *testing.T) {
	var input string
	var result []string
	var empty bool

	input = "1,1"
	result, empty = convertSticker(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, []string{}, result)

	input = "1::1::100"
	result, empty = convertSticker(input, "::")

	assert.Equal(t, false, empty)
	assert.Equal(t, []string{"1", "1", "100"}, result)
}

func TestConvertLocation(t *testing.T) {
	var input string
	var result Location
	var empty bool

	input = "1::2::3"
	result, empty = convertLocation(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, Location{}, result)

	// strconv.ParseInt: parsing "測試": invalid syntax
	input = "竹北體育館::新竹縣竹北市::測試::139.704051"
	result, empty = convertLocation(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, Location{}, result)

	// strconv.ParseInt: parsing "測試": invalid syntax
	input = "竹北體育館::新竹縣竹北市::35.661777::測試"
	result, empty = convertLocation(input, "::")

	assert.Equal(t, true, empty)
	assert.Equal(t, Location{}, result)

	input = "竹北體育館::新竹縣竹北市::35.661777::139.704051"
	result, empty = convertLocation(input, "::")

	assert.Equal(t, false, empty)
	assert.Equal(t, Location{
		Title:     "竹北體育館",
		Address:   "新竹縣竹北市",
		Latitude:  float64(35.661777),
		Longitude: float64(139.704051),
	}, result)
}

func TestParseTo(t *testing.T) {
	input := []string{"a", "b::1@gmail.com", "c::2@gmail.com", "d::3@gmail.com", "e", "f"}

	ids := parseTo(input, "1@gmail.com", false, "::")
	assert.Equal(t, []string{"a", "e", "f", "b"}, ids)

	ids = parseTo(input, "1@gmail.com", true, "::")
	assert.Equal(t, []string{"b"}, ids)

	ids = parseTo(input, "a@gmail.com", false, "::")
	assert.Equal(t, []string{"a", "e", "f"}, ids)

	ids = parseTo(input, "a@gmail.com", true, "::")
	assert.Equal(t, []string{"a", "e", "f"}, ids)

	// test empty ids
	ids = parseTo([]string{"", " ", "   "}, "a@gmail.com", true, "::")
	assert.Equal(t, 0, len(ids))
}

func TestTunnelDomain(t *testing.T) {
	plugin := Plugin{
		Config: Config{
			Domain: "abc",
		},
	}

	domain, err := plugin.getTunnelDomain()
	assert.Empty(t, domain)
	assert.Error(t, err)

	plugin.Config.Domain = "newtunnel"

	domain, err = plugin.getTunnelDomain()
	assert.Equal(t, "newtunnel", domain)
	assert.Nil(t, err)

	plugin.Config.Domain = ""

	domain, err = plugin.getTunnelDomain()
	assert.Equal(t, 10, len(domain))
	assert.Nil(t, err)
}

func performRequest(r http.Handler, method, url string) *httptest.ResponseRecorder {
	req, _ := http.NewRequest(method, url, nil)
	w := httptest.NewRecorder()
	r.ServeHTTP(w, req)
	return w
}

func TestDefaultRouter(t *testing.T) {
	p := Plugin{
		Config: Config{
			ChannelToken:  os.Getenv("LINE_CHANNEL_TOKEN"),
			ChannelSecret: os.Getenv("LINE_CHANNEL_SECRET"),
			ToRoom:        os.Getenv("LINE_TO"),
		},
	}

	bot, err := p.Bot()

	if err != nil {
		t.Fatal(err)
	}

	router := p.Handler(bot)
	w := performRequest(router, "GET", "/")
	assert.Equal(t, "Welcome to Line webhook page.\n", w.Body.String())

	w = performRequest(router, "GET", "/metrics")
	assert.Equal(t, 200, w.Code)
}