main_test.go

package concourse_tfe_resource

import (
	"bytes"
	"encoding/json"
	"os"
	"path"
	"strings"
	"testing"
)

func TestStartup(t *testing.T) {
	input := inputJSON{
		Source:  sourceJSON{},
		Params:  paramsJSON{},
		Version: version{},
	}

	err := startup(input)
	if err == nil || !strings.Contains(err.Error(), "creating tfe client") {
		t.Errorf("no/bad error creating client with empty config: %s", err)
	}

	input.Source.Token = os.Getenv("ATLAS_TOKEN")
	err = startup(input)

	if err == nil || !strings.Contains(err.Error(), "getting workspace") {
		t.Errorf("no/bad error without org/workspace set: %s", err)
	}

	input.Source.Workspace = os.Getenv("TFE_WORKSPACE")
	input.Source.Organization = os.Getenv("TFE_ORGANIZATION")

	err = startup(input)
	if err != nil {
		t.Errorf("startup failed with valid config: %s", err)
	}
}

func TestRealMain(t *testing.T) {
	input := inputJSON{
		Source: sourceJSON{
			Workspace:    os.Getenv("TFE_WORKSPACE"),
			Organization: os.Getenv("TFE_ORGANIZATION"),
			Token:        os.Getenv("ATLAS_TOKEN"),
			Address:      os.Getenv("TFE_ADDRESS"),
		},
		Params: paramsJSON{
			PollingPeriod: 5,
		},
	}

	args := []string{"check"}
	byteInput, _ := json.Marshal(input)
	output, err := realMain(args, bytes.NewReader(byteInput))

	if err != nil {
		t.Errorf("check failed: %s", err)
	}

	wd, _ := os.Getwd()
	_ = os.Mkdir(path.Join(wd, "test_output", "test_main_in"), os.FileMode(0755))
	run := make([]version, 1)
	_ = json.Unmarshal(output, &run)
	args = []string{"in", path.Join("test_output", "test_main_in")}
	input.Version = version{Ref: run[0].Ref}
	byteInput, _ = json.Marshal(input)
	_, err = realMain(args, bytes.NewReader(byteInput))

	if err != nil {
		t.Errorf("in on checked run failed: %s", err)
	}

	args[0] = "out"
	input.Params.Message = "TestRealMain out test"
	byteInput, _ = json.Marshal(input)
	_, err = realMain(args, bytes.NewReader(byteInput))

	if err != nil {
		t.Errorf("out failed: %s", err)
	}
}