package executor

import "time"

var _ Service = &Mock{}

type Mock struct {
	SlurmQueueFunc            func() []string
	SlurmJobsFunc             func() []string
	NodeUptimeFunc            func(hostname string) time.Time
	TemperatureFunc           func(hostname string) float64
	MemoryUsageFunc           func(hostname string) int64
	StorageUsageFunc          func(hostname string) int64
	NetworkUploadDownloadFunc func(hostname string) (int64, int64)
}

func (ex *Mock) SlurmQueue() []string {
	return ex.SlurmQueueFunc()
}

func (ex *Mock) SlurmJobs() []string {
	return ex.SlurmJobsFunc()
}

func (ex *Mock) NodeUptime(hostname string) time.Time {
	return ex.NodeUptimeFunc(hostname)
}

func (ex *Mock) Temperature(hostname string) float64 {
	return ex.TemperatureFunc(hostname)
}

func (ex *Mock) MemoryUsage(hostname string) int64 {
	return ex.MemoryUsageFunc(hostname)
}

func (ex *Mock) StorageUsage(hostname string) int64 {
	return ex.StorageUsageFunc(hostname)
}

func (ex *Mock) NetworkUploadDownload(hostname string) (int64, int64) {
	return ex.NetworkUploadDownloadFunc(hostname)
}