Method Or

testing/generation/utils/generator_stack.go:66–104

Or will take all items from the current depth and add them to a parent OrGen. Either the previous depth is an OrGen, or the stack will increment the sub depth to insert an OrGen.

()

// Or will take all items from the current depth and add them to a parent OrGen. Either the previous depth is an OrGen,
// or the stack will increment the sub depth to insert an OrGen.
func (sgs *StatementGeneratorStack) Or() error {
	if sgs.stack.Empty() {
		return errors.Errorf("cannot apply Or to an empty stack")
	}
	parentElement := sgs.stack.PeekDepth(1)
	current := sgs.aggregate(sgs.stack.Pop().generators)
	if parentElement == nil {
		// We're the root, so we put an OrGen as the root, and make the current a child of the new OrGen
		sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth, Or(current)))
		sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth))
	} else if len(parentElement.generators) == 0 {
		if parentElement.depth == sgs.depth {
			// We're still at the same depth, so it's safe to append an OrGen
			parentElement.Append(Or(current))
		} else {
			// We should retain the depth, so we need to create a new element since there's a depth boundary
			sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth, Or(current)))
		}
		sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth))
	} else {
		if parentElement.depth == sgs.depth {
			// We're still at the same depth, so we need to check if the parent is an OrGen or not
			if orGen, ok := parentElement.LastGenerator().(*OrGen); ok {
				// The parent is an OrGen, so we can simply add this as another option
				if err := orGen.AddChildren(current); err != nil {
					return err
				}
			} else {
				// The parent is not an OrGen, so we need to add to the depth
				sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth, Or(current)))
			}
		} else {
			// We should retain the depth, so we need to create a new element since there's a depth boundary
			sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth, Or(current)))
		}
		sgs.stack.Push(NewStatementGeneratorStackElement(sgs.depth))
	}
	return nil
}

// NewScope increases the depth.
func (sgs *StatementGeneratorStack) NewScope() {