How to use WaitUntilReady method of wait Package

Best Testcontainers-go code snippet using wait.WaitUntilReady

api_test.go

Source:api_test.go

1package internal2import (3	"fmt"4	"testing"5	"time"6	"github.com/stretchr/testify/suite"7	"github.com/diamnet/go/ingest/ledgerbackend"8	"github.com/diamnet/go/support/log"9	"github.com/diamnet/go/xdr"10)11func TestAPITestSuite(t *testing.T) {12	suite.Run(t, new(APITestSuite))13}14type APITestSuite struct {15	suite.Suite16	ledgerBackend *ledgerbackend.MockDatabaseBackend17	api           CaptiveCoreAPI18}19func (s *APITestSuite) SetupTest() {20	s.ledgerBackend = &ledgerbackend.MockDatabaseBackend{}21	s.api = NewCaptiveCoreAPI(s.ledgerBackend, log.New())22}23func (s *APITestSuite) TearDownTest() {24	s.ledgerBackend.AssertExpectations(s.T())25}26func (s *APITestSuite) TestLatestSeqActiveRequestInvalid() {27	_, err := s.api.GetLatestLedgerSequence()28	s.Assert().Equal(err, ErrMissingPrepareRange)29}30func (s *APITestSuite) TestGetLedgerActiveRequestInvalid() {31	_, err := s.api.GetLedger(64)32	s.Assert().Equal(err, ErrMissingPrepareRange)33}34func (s *APITestSuite) runBeforeReady(prepareRangeErr error, f func()) {35	waitChan := make(chan time.Time)36	ledgerRange := ledgerbackend.UnboundedRange(63)37	s.ledgerBackend.On("PrepareRange", ledgerRange).38		WaitUntil(waitChan).39		Return(prepareRangeErr).Once()40	response, err := s.api.PrepareRange(ledgerRange)41	s.Assert().NoError(err)42	s.Assert().False(response.Ready)43	s.Assert().Equal(response.LedgerRange, ledgerRange)44	f()45	close(waitChan)46	s.api.wg.Wait()47}48func (s *APITestSuite) TestLatestSeqActiveRequestNotReady() {49	s.runBeforeReady(nil, func() {50		_, err := s.api.GetLatestLedgerSequence()51		s.Assert().Equal(err, ErrPrepareRangeNotReady)52	})53}54func (s *APITestSuite) TestGetLedgerNotReady() {55	s.runBeforeReady(nil, func() {56		_, err := s.api.GetLedger(64)57		s.Assert().Equal(err, ErrPrepareRangeNotReady)58	})59}60func (s *APITestSuite) waitUntilReady(ledgerRange ledgerbackend.Range) {61	s.ledgerBackend.On("PrepareRange", ledgerRange).62		Return(nil).Once()63	response, err := s.api.PrepareRange(ledgerRange)64	s.Assert().NoError(err)65	s.Assert().False(response.Ready)66	s.Assert().Equal(response.LedgerRange, ledgerRange)67	s.api.wg.Wait()68}69func (s *APITestSuite) TestLatestSeqError() {70	s.waitUntilReady(ledgerbackend.UnboundedRange(63))71	expectedErr := fmt.Errorf("test error")72	s.ledgerBackend.On("GetLatestLedgerSequence").Return(uint32(0), expectedErr).Once()73	_, err := s.api.GetLatestLedgerSequence()74	s.Assert().Equal(err, expectedErr)75}76func (s *APITestSuite) TestGetLedgerError() {77	s.waitUntilReady(ledgerbackend.UnboundedRange(63))78	expectedErr := fmt.Errorf("test error")79	s.ledgerBackend.On("GetLedger", uint32(64)).80		Return(false, xdr.LedgerCloseMeta{}, expectedErr).Once()81	_, err := s.api.GetLedger(64)82	s.Assert().Equal(err, expectedErr)83}84func (s *APITestSuite) TestLatestSeqSucceeds() {85	s.waitUntilReady(ledgerbackend.UnboundedRange(63))86	expectedSeq := uint32(100)87	s.ledgerBackend.On("GetLatestLedgerSequence").Return(expectedSeq, nil).Once()88	seq, err := s.api.GetLatestLedgerSequence()89	s.Assert().NoError(err)90	s.Assert().Equal(seq, ledgerbackend.LatestLedgerSequenceResponse{Sequence: expectedSeq})91}92func (s *APITestSuite) TestGetLedgerSucceeds() {93	s.waitUntilReady(ledgerbackend.UnboundedRange(63))94	expectedLedger := xdr.LedgerCloseMeta{95		V0: &xdr.LedgerCloseMetaV0{96			LedgerHeader: xdr.LedgerHeaderHistoryEntry{97				Header: xdr.LedgerHeader{98					LedgerSeq: 64,99				},100			},101		},102	}103	s.ledgerBackend.On("GetLedger", uint32(64)).104		Return(true, expectedLedger, nil).Once()105	seq, err := s.api.GetLedger(64)106	s.Assert().NoError(err)107	s.Assert().Equal(seq, ledgerbackend.LedgerResponse{108		Present: true,109		Ledger:  ledgerbackend.Base64Ledger(expectedLedger),110	})111}112func (s *APITestSuite) TestShutDownBeforePrepareRange() {113	s.ledgerBackend.On("Close").Return(nil).Once()114	s.api.Shutdown()115	_, err := s.api.PrepareRange(ledgerbackend.UnboundedRange(63))116	s.Assert().EqualError(err, "Cannot prepare range when shut down")117}118func (s *APITestSuite) TestShutDownDuringPrepareRange() {119	s.runBeforeReady(nil, func() {120		s.api.cancel()121	})122	s.Assert().False(s.api.activeRequest.ready)123}124func (s *APITestSuite) TestPrepareRangeInvalidActiveRequest() {125	s.runBeforeReady(nil, func() {126		s.Assert().True(s.api.activeRequest.valid)127		s.api.activeRequest.valid = false128	})129	s.Assert().False(s.api.activeRequest.ready)130	s.api.activeRequest = &rangeRequest{}131	s.runBeforeReady(fmt.Errorf("with error"), func() {132		s.Assert().True(s.api.activeRequest.valid)133		s.api.activeRequest.valid = false134	})135	s.Assert().False(s.api.activeRequest.ready)136}137func (s *APITestSuite) TestPrepareRangeDoesNotMatchActiveRequestRange() {138	s.runBeforeReady(nil, func() {139		s.Assert().Equal(ledgerbackend.UnboundedRange(63), s.api.activeRequest.ledgerRange)140		s.api.activeRequest.ledgerRange = ledgerbackend.UnboundedRange(1000)141	})142	s.Assert().False(s.api.activeRequest.ready)143	s.Assert().Equal(ledgerbackend.UnboundedRange(1000), s.api.activeRequest.ledgerRange)144	s.api.activeRequest = &rangeRequest{}145	s.runBeforeReady(fmt.Errorf("with error"), func() {146		s.Assert().Equal(ledgerbackend.UnboundedRange(63), s.api.activeRequest.ledgerRange)147		s.api.activeRequest.ledgerRange = ledgerbackend.UnboundedRange(10)148	})149	s.Assert().False(s.api.activeRequest.ready)150	s.Assert().Equal(ledgerbackend.UnboundedRange(10), s.api.activeRequest.ledgerRange)151}152func (s *APITestSuite) TestPrepareRangeActiveRequestReady() {153	s.runBeforeReady(nil, func() {154		s.api.activeRequest.ready = true155	})156	s.Assert().True(s.api.activeRequest.ready)157	s.Assert().True(s.api.activeRequest.valid)158	s.Assert().Equal(0, s.api.activeRequest.readyDuration)159	s.api.activeRequest = &rangeRequest{}160	s.runBeforeReady(fmt.Errorf("with error"), func() {161		s.api.activeRequest.ready = true162	})163	s.Assert().True(s.api.activeRequest.ready)164	s.Assert().True(s.api.activeRequest.valid)165	s.Assert().Equal(0, s.api.activeRequest.readyDuration)166}167func (s *APITestSuite) TestPrepareRangeError() {168	s.runBeforeReady(fmt.Errorf("with error"), func() {169		s.Assert().False(s.api.activeRequest.ready)170		s.Assert().True(s.api.activeRequest.valid)171	})172	s.Assert().False(s.api.activeRequest.ready)173	s.Assert().False(s.api.activeRequest.valid)174	s.api.activeRequest = &rangeRequest{}175}176func (s *APITestSuite) TestRangeAlreadyPrepared() {177	superSetRange := ledgerbackend.UnboundedRange(63)178	s.waitUntilReady(superSetRange)179	for _, ledgerRange := range []ledgerbackend.Range{180		superSetRange,181		ledgerbackend.UnboundedRange(100),182		ledgerbackend.BoundedRange(63, 70),183	} {184		response, err := s.api.PrepareRange(ledgerRange)185		s.Assert().NoError(err)186		s.Assert().True(response.Ready)187		s.Assert().Equal(superSetRange, response.LedgerRange)188	}189}190func (s *APITestSuite) TestNewPrepareRange() {191	s.waitUntilReady(ledgerbackend.UnboundedRange(63))192	s.waitUntilReady(ledgerbackend.UnboundedRange(50))193	s.waitUntilReady(ledgerbackend.BoundedRange(45, 50))194	s.waitUntilReady(ledgerbackend.UnboundedRange(46))195}...

WaitUntilReady

Using AI Code Generation

1import (2func main() {3    wg.Add(2)4    go func() {5        defer wg.Done()6        time.Sleep(5 * time.Second)7        fmt.Println("Hello")8    }()9    go func() {10        defer wg.Done()11        time.Sleep(1 * time.Second)12        fmt.Println("World")13    }()14    wg.Wait()15    fmt.Println("Done")16}17import (18func main() {19    ch := make(chan struct{})20    go func() {21        time.Sleep(5 * time.Second)22        fmt.Println("Hello")23        ch <- struct{}{}24    }()25    go func() {26        time.Sleep(1 * time.Second)27        fmt.Println("World")28        ch <- struct{}{}29    }()30    fmt.Println("Done")31}32import (33func main() {

WaitUntilReady

Using AI Code Generation

1import (  2type Wait struct {  3    ch chan struct{}4}5func (w *Wait) WaitUntilReady() {  6}7func (w *Wait) SetReady() {  8    close(w.ch)9}10func main() {  11    w := Wait{12        ch: make(chan struct{}),13    }14    go func() {  15        time.Sleep(1 * time.Second)16        w.SetReady()17    }()18    fmt.Println(“Waiting for ready…”)19    w.WaitUntilReady()20    fmt.Println(“Ready and done!”)21}22import (  23func main() {  24    for i := 1; i <= 5; i++ {25        wg.Add(1)26        go worker(i, &wg)27    }28    wg.Wait()29}30func worker(id int, wg *sync.WaitGroup) {  31    defer wg.Done()32    fmt.Printf(“Worker %d starting33    time.Sleep(time.Second)34    fmt.Printf(“Worker %d done35}

WaitUntilReady

Using AI Code Generation

1import (2func main() {3	fmt.Println("main started")4	wait := NewWait()5	go wait.WaitUntilReady()6	fmt.Println("main stopped")7}8type Wait struct {9}10func NewWait() Wait {11	return Wait{12		ready: make(chan bool),13	}14}15func (w *Wait) WaitUntilReady() {16	fmt.Println("WaitUntilReady started")17	time.Sleep(2 * time.Second)18	fmt.Println("WaitUntilReady stopped")19}20import (21func main() {22	fmt.Println("main started")23	wait := NewWait()24	go wait.WaitUntilReady()25	fmt.Println("main stopped")26}27type Wait struct {28}29func NewWait() Wait {30	return Wait{31		ready: make(chan bool, 1),32	}33}34func (w *Wait) WaitUntilReady() {35	fmt.Println("WaitUntilReady started")36	time.Sleep(2 * time.Second)37	fmt.Println("WaitUntilReady stopped")

WaitUntilReady

Using AI Code Generation

1import (2type Wait struct {3	c chan struct{}4}5func (w *Wait) WaitUntilReady() {6	fmt.Println("Ready")7}8func main() {9	w := &Wait{10		c: make(chan struct{}),11	}12	go func() {13		time.Sleep(time.Second * 2)14		w.c <- struct{}{}15	}()16	w.WaitUntilReady()17}18import (19type Wait struct {20	c chan struct{}21}22func (w *Wait) WaitUntilReady() {23	fmt.Println("Ready")24}25func main() {26	w := &Wait{27		c: make(chan struct{}),28	}29	go func() {30		time.Sleep(time.Second * 2)31		w.c <- struct{}{}32	}()33	w.WaitUntilReady()34}35import (36type Wait struct {37	c chan struct{}38}39func (w *Wait) WaitUntilReady() {40	fmt.Println("Ready")41}42func main() {43	w := &Wait{44		c: make(chan struct{}),45	}46	go func() {47		time.Sleep(time.Second * 2)

WaitUntilReady

Using AI Code Generation

1import (2func main() {3	wait := new(wait)4	wait.WaitUntilReady()5	fmt.Println("WaitUntilReady method finished")6}7import (8func main() {9	wait := new(wait)10	wait.WaitUntilReady()11	fmt.Println("WaitUntilReady method finished")12}

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.