Function getBatchResults

backend/plugin/parser/tsql/batch/batch_test.go:96–122

(a *require.Assertions, input string)

}

func getBatchResults(a *require.Assertions, input string) []batchResult {
	batch := NewBatcher(input)
	var batchResults []batchResult
	for {
		command, err := batch.Next()
		if err != nil {
			if err == io.EOF {
				if v := batch.Batch(); v != nil && len(v.Text) > 0 {
					// If meet the end of file, get the last batch.
					batchResults = append(batchResults, batchResult{
						Statements: *v,
					})
				}
				batch.Reset(nil)
				return batchResults
			}
			a.NoError(err)
		}
		if command != nil {
			batchResults = append(batchResults, batchResult{
				Statements: *batch.Batch(),
				Command:    command.String(),
			})
			batch.Reset(nil)
		}
	}
}