Hello everyone, I'm relatively new to Go development and currently facing challenges with testing.

I'm struggling to mock the libraries in the google/genai SDK. I tried to create a wrapper for abstraction.

package clients
import (
    "context"
    "google.golang.org/genai"
    "io"
    "iter"
)

type GenaiClientWrapper struct {
    *genai.Client
}

func NewGenaiClientWrapper(client *genai.Client) *GenaiClientWrapper {
    return &GenaiClientWrapper{Client: client}
}

func (c GenaiClientWrapper) GenerateContent(ctx context.Context, model string, contents []*genai.Content, config *genai.GenerateContentConfig) (*genai.GenerateContentResponse, error) {
    return c.Client.Models.GenerateContent(
       ctx,
       model,
       contents,
       config,
    )
}

func (c GenaiClientWrapper) GenerateContentStream(ctx context.Context, model string, contents []*genai.Content, config *genai.GenerateContentConfig) iter.Seq2[*genai.GenerateContentResponse, error] {
    return c.Client.Models.GenerateContentStream(
       ctx,
       model,
       contents,
       config,
    )
}

func (c GenaiClientWrapper) Upload(ctx context.Context, r io.Reader, config *genai.UploadFileConfig) (*genai.File, error) {
    return c.Client.Files.Upload(
       ctx,
       r,
       config,
    )
}

But i can't seem to find a way to mock the iter.Seq2 response. Has anyone tried to use the genai sdk in their projects? Is there a better way to implement the abstraction?