main_test.go

package main

import (
	"os"
	"testing"

	promlog "github.com/prometheus/common/promlog"
)

func TestScrape(t *testing.T) {
	logger := promlog.New(&promlog.Config{})

	// TestSuccessfulConnection
	exporter, _ := NewExporter(os.Getenv("GOOGLE_PROJECT_ID"), logger)
	projectUp, regionsUp := exporter.scrape()
	if projectUp == nil {
		t.Errorf("TestSuccessfulConnection: projectUp=0, expected=1")
	}
	if regionsUp == nil {
		t.Errorf("TestSuccessfulConnection: regionsUp=0, expected=1")
	}

	// TestFailedConnection
	// Set the project name to "503" since the Google Compute API will append this to the end of the BasePath
	exporter, _ = NewExporter("503", logger)
	exporter.service.BasePath = "http://httpstat.us/"
	projectUp, regionsUp = exporter.scrape()
	if projectUp != nil {
		t.Errorf("TestFailedConnection: projectUp=1, expected=0")
	}
	if regionsUp != nil {
		t.Errorf("TestFailedConnection: regionsUp=1, expected=0")
	}
}