statping/core/services_checkin_test.go

// Statping
// Copyright (C) 2018.  Hunter Long and the project contributors
// Written by Hunter Long <info@socialeck.com> and the project contributors
//
// https://github.com/hunterlong/statping
//
// The licenses for most software and other practical works are designed
// to take away your freedom to share and change the works.  By contrast,
// the GNU General Public License is intended to guarantee your freedom to
// share and change all versions of a program--to make sure it remains free
// software for all its users.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package core

import (
	"github.com/hunterlong/statping/types"
	"github.com/hunterlong/statping/utils"
	"github.com/stretchr/testify/assert"
	"testing"
	"time"
)

var (
	testCheckin *Checkin
)

func TestCreateCheckin(t *testing.T) {
	service := SelectService(1)
	testCheckin = ReturnCheckin(&types.Checkin{
		ServiceId:   service.Id,
		Interval:    10,
		GracePeriod: 5,
		ApiKey:      utils.RandomString(7),
	})
	id, err := testCheckin.Create()
	assert.Nil(t, err)
	assert.NotZero(t, id)
	assert.NotEmpty(t, testCheckin.ApiKey)
	assert.Equal(t, int64(10), testCheckin.Interval)
	assert.Equal(t, int64(5), testCheckin.GracePeriod)
	assert.True(t, testCheckin.Expected().Minutes() < 0)
}

func TestSelectCheckin(t *testing.T) {
	service := SelectService(1)
	checkins := service.AllCheckins()
	assert.NotNil(t, checkins)
	assert.Equal(t, 1, len(checkins))
	testCheckin = checkins[0]
	assert.Equal(t, int64(10), testCheckin.Interval)
	assert.Equal(t, int64(5), testCheckin.GracePeriod)
	assert.Equal(t, 7, len(testCheckin.ApiKey))
}

func TestUpdateCheckin(t *testing.T) {
	testCheckin.Interval = 60
	testCheckin.GracePeriod = 15
	id, err := testCheckin.Update()
	assert.Nil(t, err)
	assert.NotZero(t, id)
	assert.NotEmpty(t, testCheckin.ApiKey)
	service := SelectService(1)
	checkin := service.AllCheckins()[0]
	assert.Equal(t, int64(60), checkin.Interval)
	assert.Equal(t, int64(15), checkin.GracePeriod)
	t.Log(testCheckin.Expected())
	assert.True(t, testCheckin.Expected().Minutes() < 0)
}

func TestCreateCheckinHits(t *testing.T) {
	service := SelectService(1)
	checkins := service.AllCheckins()
	assert.Equal(t, 1, len(checkins))
	created := time.Now().UTC().Add(-60 * time.Second)
	hit := ReturnCheckinHit(&types.CheckinHit{
		Checkin:   testCheckin.Id,
		From:      "192.168.1.1",
		CreatedAt: created,
	})
	hit.Create()
	hits := testCheckin.AllHits()
	assert.Equal(t, 1, len(hits))
}

func TestSelectCheckinMethods(t *testing.T) {
	time.Sleep(5 * time.Second)
	service := SelectService(1)
	checkins := service.AllCheckins()
	assert.NotNil(t, checkins)
	lastHit := testCheckin.Last()
	assert.Equal(t, float64(60), testCheckin.Period().Seconds())
	assert.Equal(t, float64(15), testCheckin.Grace().Seconds())
	t.Log(testCheckin.Expected())
	assert.True(t, testCheckin.Expected().Seconds() < -5)
	assert.False(t, lastHit.CreatedAt.IsZero())
	assert.Equal(t, "A minute ago", lastHit.Ago())
}