//---------------------------------------------------------------------------
//
// SCSI Target Emulator PiSCSI
// for Raspberry Pi
//
// Copyright (C) 2022-2023 Uwe Seimet
//
//---------------------------------------------------------------------------

#include "mocks.h"
#include "devices/device_factory.h"
#include "controllers/controller_manager.h"

TEST(ControllerManagerTest, LifeCycle)
{
	const int ID1 = 4;
	const int ID2 = 5;
	const int LUN1 = 0;
	const int LUN2 = 3;

	auto bus = make_shared<MockBus>();
	ControllerManager controller_manager;
	DeviceFactory device_factory;

	auto device = device_factory.CreateDevice(SCHS, -1, "");
	EXPECT_FALSE(controller_manager.AttachToController(*bus, ID1, device));

	device = device_factory.CreateDevice(SCHS, LUN1, "");
	EXPECT_TRUE(controller_manager.AttachToController(*bus, ID1, device));
	EXPECT_TRUE(controller_manager.HasController(ID1));
	auto controller = controller_manager.FindController(ID1);
	EXPECT_NE(nullptr, controller);
	EXPECT_EQ(1, controller->GetLunCount());
	EXPECT_FALSE(controller_manager.HasController(0));
	EXPECT_EQ(nullptr, controller_manager.FindController(0));
	EXPECT_TRUE(controller_manager.HasDeviceForIdAndLun(ID1, LUN1));
	EXPECT_NE(nullptr, controller_manager.GetDeviceForIdAndLun(ID1, LUN1));
	EXPECT_FALSE(controller_manager.HasDeviceForIdAndLun(0, 0));
	EXPECT_EQ(nullptr, controller_manager.GetDeviceForIdAndLun(0, 0));

	device = device_factory.CreateDevice(SCHS, LUN2, "");
	EXPECT_TRUE(controller_manager.AttachToController(*bus, ID1, device));
	EXPECT_TRUE(controller_manager.HasController(ID1));
	controller = controller_manager.FindController(ID1);
	EXPECT_NE(nullptr, controller_manager.FindController(ID1));
	EXPECT_TRUE(controller_manager.DeleteController(*controller));
	EXPECT_EQ(nullptr, controller_manager.FindController(ID1));

	auto disk = make_shared<MockDisk>();
	EXPECT_TRUE(controller_manager.AttachToController(*bus, ID2, disk));
	EXPECT_CALL(*disk, FlushCache);
	controller_manager.DeleteAllControllers();
	EXPECT_FALSE(controller_manager.HasController(ID1));
	EXPECT_EQ(nullptr, controller_manager.FindController(ID1));
	EXPECT_EQ(nullptr, controller_manager.GetDeviceForIdAndLun(ID1, LUN1));
	EXPECT_FALSE(controller_manager.HasDeviceForIdAndLun(ID1, LUN1));
	EXPECT_FALSE(controller_manager.HasController(ID2));
	EXPECT_EQ(nullptr, controller_manager.FindController(ID2));
	EXPECT_EQ(nullptr, controller_manager.GetDeviceForIdAndLun(ID2, LUN1));
	EXPECT_FALSE(controller_manager.HasDeviceForIdAndLun(ID2, LUN1));
}

TEST(ControllerManagerTest, AttachToController)
{
	const int ID = 4;
	const int LUN1 = 3;
	const int LUN2 = 0;

	auto bus = make_shared<MockBus>();
	ControllerManager controller_manager;
	DeviceFactory device_factory;

	auto device1 = device_factory.CreateDevice(SCHS, LUN1, "");
	EXPECT_FALSE(controller_manager.AttachToController(*bus, ID, device1)) << "LUN 0 is missing";

	auto device2 = device_factory.CreateDevice(SCLP, LUN2, "");
	EXPECT_TRUE(controller_manager.AttachToController(*bus, ID, device2));
	EXPECT_TRUE(controller_manager.AttachToController(*bus, ID, device1));
	EXPECT_FALSE(controller_manager.AttachToController(*bus, ID, device1));
}

TEST(ControllerManager, ProcessOnController)
{
	ControllerManager controller_manager;

	EXPECT_EQ(AbstractController::piscsi_shutdown_mode::NONE, controller_manager.ProcessOnController(0));
}