package main

import (
	"testing"

	"github.com/feast-dev/feast/go/internal/feast"
	"github.com/feast-dev/feast/go/internal/feast/server/logging"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
)

// MockServerStarter is a mock of ServerStarter interface for testing
type MockServerStarter struct {
	mock.Mock
}

func (m *MockServerStarter) StartHttpServer(fs *feast.FeatureStore, host string, port int, writeLoggedFeaturesCallback logging.OfflineStoreWriteCallback, loggingOpts *logging.LoggingOptions) error {
	args := m.Called(fs, host, port, writeLoggedFeaturesCallback, loggingOpts)
	return args.Error(0)
}

func (m *MockServerStarter) StartGrpcServer(fs *feast.FeatureStore, host string, port int, writeLoggedFeaturesCallback logging.OfflineStoreWriteCallback, loggingOpts *logging.LoggingOptions) error {
	args := m.Called(fs, host, port, writeLoggedFeaturesCallback, loggingOpts)
	return args.Error(0)
}

// TestStartHttpServer tests the StartHttpServer function
func TestStartHttpServer(t *testing.T) {
	mockServerStarter := new(MockServerStarter)
	fs := &feast.FeatureStore{}
	host := "localhost"
	port := 8080
	var writeLoggedFeaturesCallback logging.OfflineStoreWriteCallback

	loggingOpts := &logging.LoggingOptions{}

	mockServerStarter.On("StartHttpServer", fs, host, port, mock.AnythingOfType("logging.OfflineStoreWriteCallback"), loggingOpts).Return(nil)

	err := mockServerStarter.StartHttpServer(fs, host, port, writeLoggedFeaturesCallback, loggingOpts)
	assert.NoError(t, err)
	mockServerStarter.AssertExpectations(t)
}

// TestStartGrpcServer tests the StartGrpcServer function
func TestStartGrpcServer(t *testing.T) {
	mockServerStarter := new(MockServerStarter)
	fs := &feast.FeatureStore{}
	host := "localhost"
	port := 9090
	var writeLoggedFeaturesCallback logging.OfflineStoreWriteCallback
	loggingOpts := &logging.LoggingOptions{}

	mockServerStarter.On("StartGrpcServer", fs, host, port, mock.AnythingOfType("logging.OfflineStoreWriteCallback"), loggingOpts).Return(nil)

	err := mockServerStarter.StartGrpcServer(fs, host, port, writeLoggedFeaturesCallback, loggingOpts)
	assert.NoError(t, err)
	mockServerStarter.AssertExpectations(t)
}

// TestConstructLoggingService tests the constructLoggingService function
func TestConstructLoggingService(t *testing.T) {
	fs := &feast.FeatureStore{}
	var writeLoggedFeaturesCallback logging.OfflineStoreWriteCallback
	loggingOpts := &logging.LoggingOptions{}

	_, err := constructLoggingService(fs, writeLoggedFeaturesCallback, loggingOpts)
	assert.NoError(t, err)
	// Further assertions can be added here based on the expected behavior of constructLoggingService
}

// Note: Additional tests can be written for other functions and error scenarios.