How to use Context method of modulestest Package

Best K6 code snippet using modulestest.Context

queuer_test.go

Source:queuer_test.go

1package taskqueue2import (3	"context"4	"testing"5	"time"6	"github.com/dop251/goja"7	"github.com/stretchr/testify/require"8	"go.k6.io/k6/js/common"9	"go.k6.io/k6/js/eventloop"10	"go.k6.io/k6/js/modulestest"11)12func TestTaskQueue(t *testing.T) {13	// really basic test14	t.Parallel()15	rt := goja.New()16	vu := &modulestest.VU{17		RuntimeField: rt,18		InitEnvField: &common.InitEnvironment{},19		CtxField:     context.Background(),20		StateField:   nil,21	}22	loop := eventloop.New(vu)23	fq := New(loop.RegisterCallback)24	var i int25	require.NoError(t, rt.Set("a", func() {26		fq.Queue(func() error {27			fq.Queue(func() error {28				fq.Queue(func() error {29					i++30					fq.Close()31					return nil32				})33				i++34				return nil35			})36			i++37			return nil38		})39	}))40	err := loop.Start(func() error {41		_, err := vu.Runtime().RunString(`a()`)42		return err43	})44	require.NoError(t, err)45	require.Equal(t, i, 3)46}47func TestTwoTaskQueues(t *testing.T) {48	// try to find any kind of races through running multiple queues and having them race with each other49	t.Parallel()50	rt := goja.New()51	ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*100)52	t.Cleanup(cancel)53	vu := &modulestest.VU{54		RuntimeField: rt,55		CtxField:     ctx,56	}57	loop := eventloop.New(vu)58	fq := New(loop.RegisterCallback)59	fq2 := New(loop.RegisterCallback)60	var i int61	incrimentI := func() { i++ }62	var j int63	incrimentJ := func() { j++ }64	var k int65	incrimentK := func() { k++ }66	require.NoError(t, rt.Set("a", func() {67		for s := 0; s < 5; s++ { // make multiple goroutines68			go func() {69				for p := 0; p < 1000000; p++ {70					fq.Queue(func() error { // queue a task to increment integers71						incrimentI()72						incrimentJ()73						return nil74					})75					time.Sleep(time.Millisecond) // this is here mostly to not get a goroutine that just loops76					select {77					case <-ctx.Done():78						return79					default:80					}81				}82			}()83			go func() { // same as above but with the other queue84				for p := 0; p < 1000000; p++ {85					fq2.Queue(func() error {86						incrimentI()87						incrimentK()88						return nil89					})90					time.Sleep(time.Millisecond)91					select {92					case <-ctx.Done():93						return94					default:95					}96				}97			}()98		}99	}))100	go func() {101		<-ctx.Done()102		fq.Close()103		fq2.Close()104	}()105	err := loop.Start(func() error {106		_, err := vu.Runtime().RunString(`a()`)107		return err108	})109	require.NoError(t, err)110	loop.WaitOnRegistered()111	require.Equal(t, i, k+j)112	require.Greater(t, k, 100)113	require.Greater(t, j, 100)114}...

timers_test.go

Source:timers_test.go

1package timers2import (3	"context"4	"testing"5	"time"6	"github.com/dop251/goja"7	"github.com/stretchr/testify/require"8	"go.k6.io/k6/js/common"9	"go.k6.io/k6/js/eventloop"10	"go.k6.io/k6/js/modulestest"11)12func TestSetTimeout(t *testing.T) {13	t.Parallel()14	rt := goja.New()15	vu := &modulestest.VU{16		RuntimeField: rt,17		InitEnvField: &common.InitEnvironment{},18		CtxField:     context.Background(),19		StateField:   nil,20	}21	m, ok := New().NewModuleInstance(vu).(*Timers)22	require.True(t, ok)23	var log []string24	require.NoError(t, rt.Set("timers", m.Exports().Named))25	require.NoError(t, rt.Set("print", func(s string) { log = append(log, s) }))26	loop := eventloop.New(vu)27	vu.RegisterCallbackField = loop.RegisterCallback28	err := loop.Start(func() error {29		_, err := vu.Runtime().RunString(`30      timers.setTimeout(()=> {31        print("in setTimeout")32      })33      print("outside setTimeout")34      `)35		return err36	})37	require.NoError(t, err)38	require.Equal(t, []string{"outside setTimeout", "in setTimeout"}, log)39}40func TestSetInterval(t *testing.T) {41	t.Parallel()42	rt := goja.New()43	vu := &modulestest.VU{44		RuntimeField: rt,45		InitEnvField: &common.InitEnvironment{},46		CtxField:     context.Background(),47		StateField:   nil,48	}49	m, ok := New().NewModuleInstance(vu).(*Timers)50	require.True(t, ok)51	var log []string52	require.NoError(t, rt.Set("timers", m.Exports().Named))53	require.NoError(t, rt.Set("print", func(s string) { log = append(log, s) }))54	require.NoError(t, rt.Set("sleep10", func() { time.Sleep(10 * time.Millisecond) }))55	loop := eventloop.New(vu)56	vu.RegisterCallbackField = loop.RegisterCallback57	err := loop.Start(func() error {58		_, err := vu.Runtime().RunString(`59      var i = 0;60      let s = timers.setInterval(()=> {61        sleep10();62        if (i>1) {63          print("in setInterval");64          timers.clearInterval(s);65        }66        i++;67      }, 1);68      print("outside setInterval")69      `)70		return err71	})72	require.NoError(t, err)73	require.True(t, len(log) > 2)74	require.Equal(t, "outside setInterval", log[0])75	for i, l := range log[1:] {76		require.Equal(t, "in setInterval", l, i)77	}78}...

batch_test.go

Source:batch_test.go

1package loki2import (3	"context"4	"testing"5	gofakeit "github.com/brianvoe/gofakeit/v6"6	"go.k6.io/k6/js/modulestest"7	"go.k6.io/k6/lib"8	"go.k6.io/k6/metrics"9)10func BenchmarkNewBatch(b *testing.B) {11	samples := make(chan metrics.SampleContainer)12	state := &lib.State{13		Samples: samples,14		VUID:    15,15	}16	ctx, cancel := context.WithCancel(context.Background())17	vu := &modulestest.VU{18		CtxField:   ctx,19		StateField: state,20	}21	defer cancel()22	defer close(samples)23	go func() { // this is so that we read the send samples24		for range samples {25		}26	}()27	faker := gofakeit.New(12345)28	cardinalities := map[string]int{29		"app":       5,30		"namespace": 10,31		"pod":       100,32	}33	streams, minBatchSize, maxBatchSize := 5, 500, 100034	labels := newLabelPool(faker, cardinalities)35	c := Client{36		vu: vu,37	}38	b.ResetTimer()39	b.ReportAllocs()40	for i := 0; i < b.N; i++ {41		_ = c.newBatch(labels, streams, minBatchSize, maxBatchSize)42	}43}44func BenchmarkEncode(b *testing.B) {45	samples := make(chan metrics.SampleContainer)46	state := &lib.State{47		Samples: samples,48		VUID:    15,49	}50	ctx, cancel := context.WithCancel(context.Background())51	vu := &modulestest.VU{52		CtxField:   ctx,53		StateField: state,54	}55	defer cancel()56	defer close(samples)57	go func() { // this is so that we read the send samples58		for range samples {59		}60	}()61	faker := gofakeit.New(12345)62	cardinalities := map[string]int{63		"app":       5,64		"namespace": 10,65		"pod":       100,66	}67	streams, minBatchSize, maxBatchSize := 5, 500, 100068	labels := newLabelPool(faker, cardinalities)69	c := Client{70		vu: vu,71	}72	b.ReportAllocs()73	batch := c.newBatch(labels, streams, minBatchSize, maxBatchSize)74	b.Run("encode protobuf", func(b *testing.B) {75		b.ResetTimer()76		for i := 0; i < b.N; i++ {77			_, _ = batch.createPushRequest()78		}79	})80	b.Run("encode json", func(b *testing.B) {81		b.ResetTimer()82		for i := 0; i < b.N; i++ {83			_, _ = batch.createJSONPushRequest()84		}85	})86}...

Context

Using AI Code Generation

1import (2func main() {3	ctx.BuildTags = []string{"dev"}4	prog, pkgs, err := load(&ctx)5	if err != nil {6		log.Fatal(err)7	}8	mainFunc := mainPkg.Func("main")9	mainPkg.Build()10	fmt.Println(mainFunc)11	for _, b := range mainFunc.Blocks {12		for _, instr := range b.Instrs {13			if call, ok := instr.(*ssa.Call); ok {14				fmt.Println(call.Call)15			}16		}17	}18	fmt.Println(mainFunc.Signature.Params())19	fmt.Println(mainFunc.Signature.Results())20	for _, b := range mainFunc.Blocks {21		for _, instr := range b.Instrs {22			if call, ok := instr.(*ssa.Call); ok {23				fmt.Println(call.Call.Signature())24			}25		}26	}27	for _, mem := range mainPkg.Members {28		if global, ok := mem.(*ssa.Global); ok {29			fmt.Println(global.Type())30		}31	}32	for _, b := range mainFunc.Blocks {33		for _, instr := range b.Instrs {34			if alloc, ok := instr.(*ssa.Alloc); ok

Context

Using AI Code Generation

1import (2func main() {3    conf := loader.Config{4    }5    conf.Import("github.com/kr/pretty")6    prog, err := conf.Load()7    if err != nil {8        panic(err)9    }10    ssaProg := ssautil.CreateProgram(prog, 0)11    ssaProg.Build()12    for _, mem := range prog.Created[0].Pkg.Scope().Names() {13        fmt.Println(mem)14    }15}16golang.org/x/tools/go/ssa.(*Program).Package(0xc4200b8000, 0xc4200b8000, 0xc4200b8000)17golang.org/x/tools/go/ssa.(*Package).Type(0xc4200b8000, 0xc4200b8000, 0xc4200b8000)18golang.org/x/tools/go/ssa.(*Program).TypeOf(0xc4200b8000, 0xc4200b8000, 0xc4200b8000)

Context

Using AI Code Generation

1import (2func main() {3	var (4	fset := token.NewFileSet()5	pkgs, err := parser.ParseDir(fset, pkgPath, nil, 0)6	if err != nil {7		log.Fatal(err)8	}9	packages := make([]string, 0, len(pkgs))10	for pkg := range pkgs {11		packages = append(packages, pkg)12	}13	cfg := &packages.Config{14	}15	pkgs, err = packages.Load(cfg, packages)16	if err != nil {17		log.Fatal(err)18	}19	ctx := &build.Context{20		GOROOT:      os.Getenv("GOROOT"),21		GOPATH:      os.Getenv("GOPATH"),22		BuildTags:   []string{},23		ReleaseTags: []string{"go1.9"},24	}25	ctx.Setenv("GO111MODULE", "on")26	ctx.Setenv("GOPROXY", "direct")27	ctx.Setenv("GOSUMDB", "off")28	ctx.Setenv("GOPRIVATE", "")29	ctx.Setenv("GOINSECURE", "")30	ctx.Setenv("GOPRIVATE", "")31	ctx.BuildTags = append(ctx.BuildTags, "module_"+module)32	cfg = &packages.Config{33	}

Context

Using AI Code Generation

1import (2func main() {3	fmt.Println(modules.Context())4}5import (6func main() {7	fmt.Println(modules.Context())8}9import (10func main() {11	fmt.Println(modules.Context())12}13import (14func main() {15	fmt.Println(modules.Context())16}17import (18func main() {19	fmt.Println(modules.Context())20}21import (22func main() {23	fmt.Println(modules.Context())24}25import (26func main() {27	fmt.Println(modules.Context())28}29import (30func main() {31	fmt.Println(modules.Context())32}33import (34func main() {35	fmt.Println(modules.Context())36}

Context

Using AI Code Generation

1import (2func main() {3	ctx := context.Background()4	svc, err := translate.NewService(ctx, option.WithoutAuthentication())5	if err != nil {6		log.Fatalf("Unable to create translate service: %v", err)7	}8	resp, err := svc.Languages.List().Target("en").Do()9	if err != nil {10		log.Fatalf("Unable to retrieve supported languages: %v", err)11	}12	for _, language := range resp.Languages {13		log.Printf("Name: %s, Code: %s", language.Name, language.Language)14	}15}162021/01/22 11:48:43 Name: Chinese (Simplified), Code: zh-CN172021/01/22 11:48:43 Name: Chinese (Traditional), Code: zh-TW

Context

Using AI Code Generation

1import (2func TestExample(t *testing.T) {3	ctrl := gomock.NewController(t)4	defer ctrl.Finish()5	mockFoo := modulestest.NewMockFoo(ctrl)6	mockFoo.EXPECT().Bar(gomock.Any(), gomock.Any()).Return(nil)7	mockFoo.Bar(context.Background(), "Hello, world!")8}9import (10type Foo interface {11	Bar(context.Context, string) error12}13type MockFoo struct {14}15type MockFooMockRecorder struct {16}17func NewMockFoo(ctrl *gomock.Controller) *MockFoo {18	mock := &MockFoo{ctrl: ctrl}19	mock.recorder = &MockFooMockRecorder{mock}20}21func (m *MockFoo) EXPECT() *MockFooMockRecorder {22}23func (m *MockFoo) Bar(arg0 context.Context, arg1 string) error {24	m.ctrl.T.Helper()25	ret := m.ctrl.Call(m, "Bar", arg0, arg1)26	ret0, _ := ret[0].(error)27}28func (mr *MockFooMockRecorder) Bar(arg0, arg1 interface{}) *gomock.Call {29	mr.mock.ctrl.T.Helper()30	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Bar", reflect.TypeOf((*MockFoo)(nil).Bar), arg0, arg1)31}32import (33type Foo interface {34	Bar(context.Context, string) error35}36func Bar(ctx context.Context, s string) error {37	fmt.Println("Hello, world!")

Context

Using AI Code Generation

1import (2func main() {3 ctx = testing.Context(ctx)4 fmt.Println(ctx)5}6{[go] [/usr/local/go/src] [map[go:1.9.3]]

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.