Function pollForLineageMappings

lineage_integration_test.go:18–39

pollForLineageMappings polls the database until the expected number of lineage mappings appear

(ctx context.Context, ds database.IDataSource, balanceID string, expectedCount int, pollInterval, timeout time.Duration)

// pollForLineageMappings polls the database until the expected number of lineage mappings appear
func pollForLineageMappings(ctx context.Context, ds database.IDataSource, balanceID string, expectedCount int, pollInterval, timeout time.Duration) ([]model.LineageMapping, error) {
	timeoutCtx, cancel := context.WithTimeout(ctx, timeout)
	defer cancel()

	ticker := time.NewTicker(pollInterval)
	defer ticker.Stop()

	for {
		select {
		case <-timeoutCtx.Done():
			return nil, fmt.Errorf("timed out waiting for %d lineage mappings for balance %s: %w", expectedCount, balanceID, timeoutCtx.Err())
		case <-ticker.C:
			mappings, err := ds.GetLineageMappings(timeoutCtx, balanceID)
			if err != nil {
				continue
			}
			if len(mappings) >= expectedCount {
				return mappings, nil
			}
		}
	}
}

// pollForBalance polls until the balance reaches the expected value
func pollForBalance(ctx context.Context, ds database.IDataSource, balanceID string, expectedBalance *big.Int, pollInterval, timeout time.Duration) (*model.Balance, error) {