chatwork_notifier_test.go

package main

import (
	"errors"
	"github.com/joho/godotenv"
	"github.com/stretchr/testify/assert"
	"os"
	"testing"
)

func NewTestChatworkNotifier() *ChatworkNotifier {
	godotenv.Load()

	apiToken := os.Getenv("TEST_CHATWORK_API_TOKEN")
	roomID := os.Getenv("TEST_CHATWORK_ROOM_ID")

	if len(apiToken) == 0 || len(roomID) == 0 {
		return nil
	}
	return NewChatworkNotifier(apiToken, roomID)
}

func TestChatworkNotifier_PostStatus_True(t *testing.T) {
	notifier := NewTestChatworkNotifier()
	if notifier == nil {
		return
	}

	param := PostStatusParam{
		CheckURL:          "https://www.google.co.jp/",
		BeforeStatusCode:  500,
		CurrentStatusCode: 200,
		HTTPError:         nil,
	}

	err := notifier.PostStatus(&param)
	assert.NoError(t, err)
}

func TestChatworkNotifier_PostStatus_False(t *testing.T) {
	notifier := NewTestChatworkNotifier()
	if notifier == nil {
		return
	}

	param := PostStatusParam{
		CheckURL:          "https://www.google.co.jp/aaa",
		BeforeStatusCode:  0,
		CurrentStatusCode: 404,
		HTTPError:         nil,
	}
	err := notifier.PostStatus(&param)
	assert.NoError(t, err)
}

func TestChatworkNotifier_PostStatus_HasError(t *testing.T) {
	notifier := NewTestChatworkNotifier()
	if notifier == nil {
		return
	}

	param := PostStatusParam{
		CheckURL:          "https://aaaaaaaaa/",
		BeforeStatusCode:  0,
		CurrentStatusCode: 0,
		HTTPError:         errors.New("Test"),
	}
	err := notifier.PostStatus(&param)
	assert.NoError(t, err)
}