package ptest

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

Package ptest contains utilities for pipeline unit testing.

Index ¶

• Variables
• func Create(values []interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection)
• func Create2(a, b []interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection, beam.PCollection)
• func CreateList(values interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection)
• func CreateList2(a, b interface{}) (*beam.Pipeline, beam.Scope, beam.PCollection, beam.PCollection)
• func DefaultRunner() string
• func Main(m *testing.M)
• func MainWithDefault(m *testing.M, runner string)
• func Run(p *beam.Pipeline) error
• func RunAndValidate(t *testing.T, p *beam.Pipeline)

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.