package qemu

import "sync"

type DriverMock struct {
	sync.Mutex

	StopCalled bool
	StopErr    error

	QemuCalls [][]string
	QemuErrs  []error

	WaitForShutdownCalled bool
	WaitForShutdownState  bool

	QemuImgCalls [][]string
	QemuImgErrs  []error

	VerifyCalled bool
	VerifyErr    error

	VersionCalled bool
	VersionResult string
	VersionErr    error
}

func (d *DriverMock) Stop() error {
	d.StopCalled = true
	return d.StopErr
}

func (d *DriverMock) Qemu(args ...string) error {
	d.QemuCalls = append(d.QemuCalls, args)

	if len(d.QemuErrs) >= len(d.QemuCalls) {
		return d.QemuErrs[len(d.QemuCalls)-1]
	}
	return nil
}

func (d *DriverMock) WaitForShutdown(cancelCh <-chan struct{}) bool {
	d.WaitForShutdownCalled = true
	return d.WaitForShutdownState
}

func (d *DriverMock) QemuImg(args ...string) error {
	d.QemuImgCalls = append(d.QemuImgCalls, args)

	if len(d.QemuImgErrs) >= len(d.QemuImgCalls) {
		return d.QemuImgErrs[len(d.QemuImgCalls)-1]
	}
	return nil
}

func (d *DriverMock) Verify() error {
	d.VerifyCalled = true
	return d.VerifyErr
}

func (d *DriverMock) Version() (string, error) {
	d.VersionCalled = true
	return d.VersionResult, d.VersionErr
}