beam: github.com/apache/beam/sdks/go/pkg/beam/testing/ptest Index | Files

package ptest

import "github.com/apache/beam/sdks/go/pkg/beam/testing/ptest"

Package ptest contains utilities for pipeline unit testing.

Index

Package Files

ptest.go

Variables

var (
    // expansionAddr is the endpoint for an expansion service for cross-language
    // transforms.
    ExpansionAddr = flag.String("expansion_addr", "", "Address of Expansion Service")
)
var (
    Runner = flag.String("runner", "", "Pipeline runner.")
)

Runner is a flag that sets which runner pipelines under test will use.

The test file must have a TestMain that calls Main or MainWithDefault to function.

func Create Uses

func Create(values []interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection)

Create creates a pipeline and a PCollection with the given values.

func Create2 Uses

func Create2(a, b []interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection, beam.PCollection)

Create2 creates a pipeline and 2 PCollections with the given values.

func CreateList Uses

func CreateList(values interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection)

CreateList creates a pipeline and a PCollection with the given values.

func CreateList2 Uses

func CreateList2(a, b interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection, beam.PCollection)

CreateList2 creates a pipeline and 2 PCollections with the given values.

func DefaultRunner Uses

func DefaultRunner() string

func Main Uses

func Main(m *testing.M)

Main is an implementation of testing's TestMain to permit testing pipelines on runners other than the direct runner.

To enable this behavior, _ import the desired runner, and set the flag accordingly.

func TestMain(m *testing.M) {
	ptest.Main(m)
}

func MainWithDefault Uses

func MainWithDefault(m *testing.M, runner string)

MainWithDefault is an implementation of testing's TestMain to permit testing pipelines on runners other than the direct runner, while setting the default runner to use.

func Run Uses

func Run(p *beam.Pipeline) error

Run runs a pipeline for testing. The semantics of the pipeline is expected to be verified through passert.

func RunAndValidate Uses

func RunAndValidate(t *testing.T, p *beam.Pipeline)

RunAndValidate runs a pipeline for testing and validates the result, failing the test if the pipeline fails.

Package ptest imports 9 packages (graph) and is imported by 1 packages. Updated 2021-01-07. Refresh now. Tools for package owners.