// Licensed under the Apache License, Version 2.0 (the "License"); you may not
// use this file except in compliance with the License. You may obtain a copy of
// the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
// License for the specific language governing permissions and limitations under
// the License.

package mock

import "github.com/go-kivik/kivik/v4/driver"

// Attachments mocks driver.Attachments
type Attachments struct {
	// ID identifies a specific Attachments instance
	ID        string
	NextFunc  func(*driver.Attachment) error
	CloseFunc func() error
}

var _ driver.Attachments = &Attachments{}

// Next calls a.NextFunc
func (a *Attachments) Next(att *driver.Attachment) error {
	return a.NextFunc(att)
}

// Close calls a.CloseFunc
func (a *Attachments) Close() error {
	if a.CloseFunc != nil {
		return a.CloseFunc()
	}
	return nil
}