hook_test.go

// Copyright 2019 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package stackrus

import (
	"bytes"
	"context"
	"os"
	"testing"
	"time"

	"cloud.google.com/go/logging"

	"github.com/sirupsen/logrus"
)

var levelTests = []struct {
	logrusLevel      logrus.Level
	stackdriverLevel logging.Severity
}{
	{logrus.DebugLevel, logging.Debug},
	{logrus.InfoLevel, logging.Info},
	{logrus.WarnLevel, logging.Warning},
	{logrus.ErrorLevel, logging.Error},
	{logrus.FatalLevel, logging.Critical},
	{logrus.PanicLevel, logging.Emergency},
}

func TestEntryConversion(t *testing.T) {
	testtime := time.Now()
	logger := logrus.New()
	logger.Out = &bytes.Buffer{}
	entry := logrus.NewEntry(logger)
	entry = entry.WithFields(logrus.Fields{
		"animal": "walrus",
		"size":   10,
	})
	entry.Time = testtime
	entry.Message = "test payload"
	for _, ltest := range levelTests {
		entry.Level = ltest.logrusLevel
		result := logrusToStackdriverEntry(entry)
		if result.Severity != ltest.stackdriverLevel {
			t.Errorf("incorrectlevel, expected %v, got %v", ltest.logrusLevel, ltest.stackdriverLevel)
		}

		if result.Timestamp != testtime {
			t.Errorf("timestamp mismatch")
		}

		if result.Payload != "test payload" {
			t.Errorf("payload mismatch")
		}

		if result.Labels["animal"] != "walrus" {
			t.Errorf("string label mismatch")
		}

		if result.Labels["size"] != "10" {
			t.Errorf("int conversion label mismatch")
		}

	}

}

// this is a live integration test and will ping the stackdriver
// client in the hook, will only run if it finds a STACKRUS_TEST_PROJECT
// environment variable set. If you are @google - you can use
// STACKRUS_TEST_PROJECT='stackrus-test-project' go test
func TestPingStackdriver(t *testing.T) {
	testProject := os.Getenv("STACKRUS_TEST_PROJECT")
	ctx := context.Background()
	if testProject == "" {
		t.Skip("skipping live integration test")
	}
	testhook := NewHook(ctx, testProject, "dummy", logrus.DebugLevel)
	err := testhook.client.Ping(ctx)
	if err != nil {
		t.Errorf("Failed to ping stackdriver client: %s", err.Error())
	}
}