invocations_test.go

// Copyright (c) The Test Authors
// SPDX-License-Identifier: MPL-2.0

package test

import (
	"strings"
	"testing"
)

type testScript struct {
	label   string
	content string
}

func (ts *testScript) Label() string {
	return ts.label
}

func (ts *testScript) Content() string {
	return ts.content
}

type internalTest struct {
	t       *testing.T
	trigger bool
	helper  bool
	exp     string
	capture string
}

func (it *internalTest) TestPostScript(value string) Setting {
	return func(s *Settings) {
		s.postScripts = append(s.postScripts, &testScript{
			label:   "label: " + value,
			content: "content: " + value,
		})
	}
}

func (it *internalTest) Helper() {
	it.helper = true
}

func (it *internalTest) assert() {
	if !it.helper {
		it.t.Fatal("should be marked as helper")
	}
	if !it.trigger {
		it.t.Fatalf("condition expected to trigger; did not")
	}
	if !strings.Contains(it.capture, it.exp) {
		it.t.Fatalf("expected message %q in output, got %q", it.exp, it.capture)
	}
}

func (it *internalTest) assertNot() {
	if !it.helper {
		it.t.Fatal("should be marked as helper")
	}
	if it.trigger {
		it.t.Fatalf("condition expected not to trigger; it did\ngot message %q in output", it.capture)
	}
}

func (it *internalTest) post() {
	if !strings.Contains(it.capture, "PostScript |") {
		it.t.Fatal("expected post-script output")
	}
}

func newCase(t *testing.T, msg string) *internalTest {
	return &internalTest{
		t:       t,
		trigger: false,
		exp:     msg,
	}
}

func newCapture(t *testing.T) *internalTest {
	return &internalTest{
		t: t,
	}
}