test_exporter.go

package main

import (
	"bytes"
	"io/ioutil"
	"os"
	"path"
	"path/filepath"

	"github.com/bitrise-io/go-steputils/testresultexport"
	"github.com/bitrise-io/go-steputils/tools"
	"github.com/bitrise-io/go-utils/log"
)

const (
	testName               = "Flutter test results"
	testResultJSONFileName = "flutter_json_test_results.json"
	coverageFileName       = "flutter_coverage_lcov.info"
	coverageRelativePath   = "./coverage/lcov.info"
)

type testExporter interface {
	copyBufferToDeployPath(jsonBuffer bytes.Buffer) string
	exportDeployPath(testResultDeployPath string)
	exportTestResultsToResultPath(cfg config, testResultPath string)
	exportCoverage(projectLocation string)
}

type realTestExporter struct {
	interrupt interrupt
}

func (r realTestExporter) copyBufferToDeployPath(jsonBuffer bytes.Buffer) string {
	return copyBufferToDeployDir(jsonBuffer.Bytes(), testResultJSONFileName, r.interrupt)
}

func (r realTestExporter) exportDeployPath(testResultDeployPath string) {
	if err := tools.ExportEnvironmentWithEnvman("BITRISE_FLUTTER_TESTRESULT_PATH", testResultDeployPath); err != nil {
		r.interrupt.failWithMessage("Failed to export: BITRISE_FLUTTER_TESTRESULT_PATH, error: %s", err)
	}
	log.Donef("Test results exported in JUnit format as $BITRISE_FLUTTER_TESTRESULT_PATH")
}

func (r realTestExporter) exportTestResultsToResultPath(cfg config, testResultPath string) {
	exporter := testresultexport.NewExporter(cfg.TestResultsDir)
	if err := exporter.ExportTest(testName, testResultPath); err != nil {
		r.interrupt.failWithMessage("Export outputs: failed to export test result: %s", err)
	}
}

func (r realTestExporter) exportCoverage(projectLocation string) {
	covData, err := ioutil.ReadFile(path.Join(projectLocation, coverageRelativePath))
	if err != nil {
		r.interrupt.failWithMessage("Export outputs: failed to open %s", coverageRelativePath)
	}

	covDeployPath := copyBufferToDeployDir(covData, coverageFileName, r.interrupt)

	if err := tools.ExportEnvironmentWithEnvman("BITRISE_FLUTTER_COVERAGE_PATH", covDeployPath); err != nil {
		r.interrupt.failWithMessage("Export outputs: failed to export $BITRISE_FLUTTER_COVERAGE_PATH: %s", err)
	}

	log.Donef("Test coverage file exported as $BITRISE_FLUTTER_COVERAGE_PATH")
}

func copyBufferToDeployDir(buffer []byte, logFileName string, interrupt interrupt) string {
	deployDir := os.Getenv("BITRISE_DEPLOY_DIR")
	if deployDir == "" {
		interrupt.failWithMessage("Export outputs: no $BITRISE_DEPLOY_DIR found")
	}
	deployPth := filepath.Join(deployDir, logFileName)

	if err := ioutil.WriteFile(deployPth, buffer, 0664); err != nil {
		interrupt.failWithMessage("Export outputs: failed to write buffer to %s: %s", deployPth, err)
	}
	return deployPth
}