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

#include "mocks.h"
#include "shared/scsi.h"
#include "shared/piscsi_exceptions.h"
#include "devices/disk.h"
#include "devices/scsi_command_util.h"

using namespace scsi_defs;
using namespace scsi_command_util;

pair<shared_ptr<MockAbstractController>, shared_ptr<MockDisk>> CreateDisk()
{
	auto controller = make_shared<NiceMock<MockAbstractController>>(0);
	auto disk = make_shared<MockDisk>();
	EXPECT_TRUE(disk->Init({}));
	EXPECT_TRUE(controller->AddDevice(disk));

	return { controller, disk };
}

TEST(DiskTest, Dispatch)
{
	auto [controller, disk] = CreateDisk();

	disk->SetRemovable(true);
	disk->SetMediumChanged(false);
	disk->SetReady(true);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdTestUnitReady);
	EXPECT_EQ(status::good, controller->GetStatus());

	disk->SetMediumChanged(true);
	EXPECT_CALL(*controller, Error);
	disk->Dispatch(scsi_command::eCmdTestUnitReady);
	EXPECT_FALSE(disk->IsMediumChanged());
}

TEST(DiskTest, Rezero)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdRezero); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "REZERO must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdRezero);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, FormatUnit)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdFormatUnit); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "FORMAT UNIT must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdFormatUnit);
	EXPECT_EQ(status::good, controller->GetStatus());

	controller->SetCmdByte(1, 0x10);
	controller->SetCmdByte(4, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdFormatUnit); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))));
}

TEST(DiskTest, ReassignBlocks)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReassignBlocks); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "REASSIGN must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdReassignBlocks);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, Seek6)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdSeek6); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "SEEK(6) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	// Block count
	controller->SetCmdByte(4, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdSeek6); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "SEEK(6) must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdSeek6);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, Seek10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdSeek10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "SEEK(10) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	// Block count
	controller->SetCmdByte(5, 1);

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdSeek10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "SEEK(10) must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdSeek10);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, ReadCapacity10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "READ CAPACITY(10) must fail because drive is not ready";

	disk->SetReady(true);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "READ CAPACITY(10) must fail because the medium has no capacity";

	disk->SetBlockCount(0x12345678);
	EXPECT_CALL(*controller, DataIn);
	disk->Dispatch(scsi_command::eCmdReadCapacity10);
	auto& buf = controller->GetBuffer();
	EXPECT_EQ(0x1234, GetInt16(buf, 0));
	EXPECT_EQ(0x5677, GetInt16(buf, 2));

	disk->SetBlockCount(0x1234567887654321);
	EXPECT_CALL(*controller, DataIn);
	disk->Dispatch(scsi_command::eCmdReadCapacity10);
	buf = controller->GetBuffer();
	EXPECT_EQ(0xffff, GetInt16(buf, 0));
	EXPECT_EQ(0xffff, GetInt16(buf, 2));
}

TEST(DiskTest, ReadCapacity16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	controller->SetCmdByte(1, 0x00);

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))))
		<< "Neither READ CAPACITY(16) nor READ LONG(16)";

	// READ CAPACITY(16), not READ LONG(16)
	controller->SetCmdByte(1, 0x10);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "READ CAPACITY(16) must fail because drive is not ready";

	disk->SetReady(true);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "READ CAPACITY(16) must fail because the medium has no capacity";

	disk->SetBlockCount(0x1234567887654321);
	disk->SetSectorSizeInBytes(1024);
	EXPECT_CALL(*controller, DataIn);
	disk->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16);
	const auto& buf = controller->GetBuffer();
	EXPECT_EQ(0x1234, GetInt16(buf, 0));
	EXPECT_EQ(0x5678, GetInt16(buf, 2));
	EXPECT_EQ(0x8765, GetInt16(buf, 4));
	EXPECT_EQ(0x4320, GetInt16(buf, 6));
	EXPECT_EQ(0x0000, GetInt16(buf, 8));
	EXPECT_EQ(0x0400, GetInt16(buf, 10));
}

TEST(DiskTest, Read6)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdRead6); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "READ(6) must fail for a medium with 0 blocks";

	// Further testing requires filesystem access
}

TEST(DiskTest, Read10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdRead10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "READ(10) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdRead10);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, Read16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdRead16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "READ(16) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdRead16);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, Write6)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWrite6); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "WRITE(6) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	disk->SetReady(true);
	disk->SetProtectable(true);
	disk->SetProtected(true);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWrite6); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::data_protect),
			Property(&scsi_exception::get_asc, asc::write_protected))));

	// Further testing requires filesystem access
}

TEST(DiskTest, Write10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWrite10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "WRITE(10) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdWrite10);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, Write16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWrite16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "WRITE(16) must fail for a medium with 0 blocks";

	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdWrite16);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, Verify10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdVerify10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "VERIFY(10) must fail for a medium with 0 blocks";

	disk->SetReady(true);
	// Verify 0 sectors
	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdVerify10);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Verify 1 sector with BytChk=0
	controller->SetCmdByte(8, 1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdVerify10);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, Verify16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdVerify16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "VERIFY(16) must fail for a medium with 0 blocks";

	disk->SetReady(true);
	// Verify 0 sectors
	disk->SetBlockCount(1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdVerify16);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Verify 1 sector with BytChk=0
	controller->SetCmdByte(13, 1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdVerify16);
	EXPECT_EQ(status::good, controller->GetStatus());

	// Further testing requires filesystem access
}

TEST(DiskTest, ReadLong10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdReadLong10);
	EXPECT_EQ(status::good, controller->GetStatus());

	controller->SetCmdByte(2, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadLong10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "READ LONG(10) must fail because the capacity is exceeded";
	controller->SetCmdByte(2, 0);

	controller->SetCmdByte(7, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadLong10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))))
		<< "READ LONG(10) must fail because it currently only supports 0 bytes transfer length";
}

TEST(DiskTest, ReadLong16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	// READ LONG(16), not READ CAPACITY(16)
	controller->SetCmdByte(1, 0x11);
	controller->SetCmdByte(2, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "READ LONG(16) must fail because the capacity is exceeded";
	controller->SetCmdByte(2, 0);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16);
	EXPECT_EQ(status::good, controller->GetStatus());

	controller->SetCmdByte(13, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdReadCapacity16_ReadLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))))
		<< "READ LONG(16) must fail because it currently only supports 0 bytes transfer length";
}

TEST(DiskTest, WriteLong10)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdWriteLong10);
	EXPECT_EQ(status::good, controller->GetStatus());

	controller->SetCmdByte(2, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWriteLong10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "WRITE LONG(10) must fail because the capacity is exceeded";
	controller->SetCmdByte(2, 0);

	controller->SetCmdByte(7, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWriteLong10); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))))
		<< "WRITE LONG(10) must fail because it currently only supports 0 bytes transfer length";
}

TEST(DiskTest, WriteLong16)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	controller->SetCmdByte(2, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWriteLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::lba_out_of_range))))
		<< "WRITE LONG(16) must fail because the capacity is exceeded";
	controller->SetCmdByte(2, 0);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdWriteLong16);
	EXPECT_EQ(status::good, controller->GetStatus());

	controller->SetCmdByte(13, 1);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdWriteLong16); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::invalid_field_in_cdb))))
		<< "WRITE LONG(16) must fail because it currently only supports 0 bytes transfer length";
}

TEST(DiskTest, StartStopUnit)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	disk->SetRemovable(true);

	// Stop/Unload
	disk->SetReady(true);
	EXPECT_CALL(*controller, Status);
	EXPECT_CALL(*disk, FlushCache);
	disk->Dispatch(scsi_command::eCmdStartStop);
	EXPECT_EQ(status::good, controller->GetStatus());
	EXPECT_TRUE(disk->IsStopped());

	// Stop/Load
	controller->SetCmdByte(4, 0x02);
	disk->SetReady(true);
	disk->SetLocked(false);
	EXPECT_CALL(*controller, Status);
	EXPECT_CALL(*disk, FlushCache);
	disk->Dispatch(scsi_command::eCmdStartStop);
	EXPECT_EQ(status::good, controller->GetStatus());

	disk->SetReady(false);
	EXPECT_CALL(*disk, FlushCache).Times(0);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdStartStop); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::load_or_eject_failed))));

	disk->SetReady(true);
	disk->SetLocked(true);
	EXPECT_CALL(*disk, FlushCache).Times(0);
	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdStartStop); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::illegal_request),
			Property(&scsi_exception::get_asc, asc::load_or_eject_failed))));

	// Start/Unload
	controller->SetCmdByte(4, 0x01);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdStartStop);
	EXPECT_EQ(status::good, controller->GetStatus());
	EXPECT_FALSE(disk->IsStopped());

	// Start/Load
	controller->SetCmdByte(4, 0x03);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdStartStop);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, PreventAllowMediumRemoval)
{
	auto [controller, disk] = CreateDisk();
	// Required by the bullseye clang++ compiler
	auto d = disk;

	EXPECT_THAT([&] { d->Dispatch(scsi_command::eCmdPreventAllowMediumRemoval); }, Throws<scsi_exception>(AllOf(
			Property(&scsi_exception::get_sense_key, sense_key::not_ready),
			Property(&scsi_exception::get_asc, asc::medium_not_present))))
		<< "REMOVAL must fail because drive is not ready";

	disk->SetReady(true);

	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdPreventAllowMediumRemoval);
	EXPECT_EQ(status::good, controller->GetStatus());
	EXPECT_FALSE(disk->IsLocked());

	controller->SetCmdByte(4, 1);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdPreventAllowMediumRemoval);
	EXPECT_EQ(status::good, controller->GetStatus());
	EXPECT_TRUE(disk->IsLocked());
}

TEST(DiskTest, Eject)
{
	MockDisk disk;

	disk.SetReady(false);
	disk.SetRemovable(false);
	disk.SetLocked(false);
	EXPECT_CALL(disk, FlushCache).Times(0);
	EXPECT_FALSE(disk.Eject(false));

	disk.SetRemovable(true);
	EXPECT_CALL(disk, FlushCache).Times(0);
	EXPECT_FALSE(disk.Eject(false));

	disk.SetReady(true);
	disk.SetLocked(true);
	EXPECT_CALL(disk, FlushCache).Times(0);
	EXPECT_FALSE(disk.Eject(false));

	disk.SetReady(true);
	disk.SetLocked(false);
	EXPECT_CALL(disk, FlushCache);
	EXPECT_TRUE(disk.Eject(false));

	disk.SetReady(true);
	EXPECT_CALL(disk, FlushCache);
	EXPECT_TRUE(disk.Eject(true));
}

void DiskTest_ValidateFormatPage(AbstractController& controller, int offset)
{
	const auto& buf = controller.GetBuffer();
	EXPECT_EQ(0x08, buf[offset + 3]) << "Wrong number of trackes in one zone";
	EXPECT_EQ(25, GetInt16(buf, offset + 10)) << "Wrong number of sectors per track";
	EXPECT_EQ(1024, GetInt16(buf, offset + 12)) << "Wrong number of bytes per sector";
	EXPECT_EQ(1, GetInt16(buf, offset + 14)) << "Wrong interleave";
	EXPECT_EQ(11, GetInt16(buf, offset + 16)) << "Wrong track skew factor";
	EXPECT_EQ(20, GetInt16(buf, offset + 18)) << "Wrong cylinder skew factor";
	EXPECT_FALSE(buf[offset + 20] & 0x20) << "Wrong removable flag";
	EXPECT_TRUE(buf[offset + 20] & 0x40) << "Wrong hard-sectored flag";
}

void DiskTest_ValidateDrivePage(AbstractController& controller, int offset)
{
	const auto& buf = controller.GetBuffer();
	EXPECT_EQ(0x17, buf[offset + 2]);
	EXPECT_EQ(0x4d3b, GetInt16(buf, offset + 3));
	EXPECT_EQ(8, buf[offset + 5]) << "Wrong number of heads";
	EXPECT_EQ(7200, GetInt16(buf, offset + 20)) << "Wrong medium rotation rate";
}

void DiskTest_ValidateCachePage(AbstractController& controller, int offset)
{
	const auto& buf = controller.GetBuffer();
	EXPECT_EQ(0xffff, GetInt16(buf, offset + 4)) << "Wrong pre-fetch transfer length";
	EXPECT_EQ(0xffff, GetInt16(buf, offset + 8)) << "Wrong maximum pre-fetch";
	EXPECT_EQ(0xffff, GetInt16(buf, offset + 10)) << "Wrong maximum pre-fetch ceiling";
}

TEST(DiskTest, ModeSense6)
{
	auto [controller, disk] = CreateDisk();

	// Drive must be ready on order to return all data
	disk->SetReady(true);

	controller->SetCmdByte(2, 0x3f);
	// ALLOCATION LENGTH
	controller->SetCmdByte(4, 255);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	EXPECT_EQ(0x08, controller->GetBuffer()[3]) << "Wrong block descriptor length";

	// No block descriptor
	controller->SetCmdByte(1, 0x08);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	EXPECT_EQ(0x00, controller->GetBuffer()[2]) << "Wrong device-specific parameter";

	disk->SetReadOnly(false);
	disk->SetProtectable(true);
	disk->SetProtected(true);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	const auto& buf = controller->GetBuffer();
	EXPECT_EQ(0x80, buf[2]) << "Wrong device-specific parameter";

	// Return block descriptor
	controller->SetCmdByte(1, 0x00);

	// Format page
	controller->SetCmdByte(2, 3);
	disk->SetSectorSizeInBytes(1024);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	DiskTest_ValidateFormatPage(*controller, 12);

	// Rigid disk drive page
	controller->SetCmdByte(2, 4);
	disk->SetBlockCount(0x12345678);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	DiskTest_ValidateDrivePage(*controller, 12);

	// Cache page
	controller->SetCmdByte(2, 8);
	disk->Dispatch(scsi_command::eCmdModeSense6);
	DiskTest_ValidateCachePage(*controller, 12);
}

TEST(DiskTest, ModeSense10)
{
	auto [controller, disk] = CreateDisk();

	// Drive must be ready on order to return all data
	disk->SetReady(true);

	controller->SetCmdByte(2, 0x3f);
	// ALLOCATION LENGTH
	controller->SetCmdByte(8, 255);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	EXPECT_EQ(0x08, controller->GetBuffer()[7]) << "Wrong block descriptor length";

	// No block descriptor
	controller->SetCmdByte(1, 0x08);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	auto& buf = controller->GetBuffer();
	EXPECT_EQ(0x00, controller->GetBuffer()[3]) << "Wrong device-specific parameter";

	disk->SetReadOnly(false);
	disk->SetProtectable(true);
	disk->SetProtected(true);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	buf = controller->GetBuffer();
	EXPECT_EQ(0x80, buf[3]) << "Wrong device-specific parameter";

	// Return short block descriptor
	controller->SetCmdByte(1, 0x00);
	disk->SetBlockCount(0x1234);
	disk->SetSectorSizeInBytes(1024);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	buf = controller->GetBuffer();
	EXPECT_EQ(0x00, buf[4]) << "Wrong LONGLBA field";
	EXPECT_EQ(0x08, buf[7]) << "Wrong block descriptor length";
	EXPECT_EQ(0x00, GetInt16(buf, 8));
	EXPECT_EQ(0x1234, GetInt16(buf, 10));
	EXPECT_EQ(0x00, GetInt16(buf, 12));
	EXPECT_EQ(1024, GetInt16(buf, 14));

	// Return long block descriptor
	controller->SetCmdByte(1, 0x10);
	disk->SetBlockCount((uint64_t)0xffffffff + 1);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	buf = controller->GetBuffer();
	EXPECT_EQ(0x01, buf[4]) << "Wrong LONGLBA field";
	EXPECT_EQ(0x10, buf[7]) << "Wrong block descriptor length";
	EXPECT_EQ(0x00, GetInt16(buf, 8));
	EXPECT_EQ(0x01, GetInt16(buf, 10));
	EXPECT_EQ(0x00, GetInt16(buf, 12));
	EXPECT_EQ(0x00, GetInt16(buf, 14));
	EXPECT_EQ(0x00, GetInt16(buf, 20));
	EXPECT_EQ(1024, GetInt16(buf, 22));

	controller->SetCmdByte(1, 0x00);

	// Format page
	controller->SetCmdByte(2, 3);
	disk->SetSectorSizeInBytes(1024);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	DiskTest_ValidateFormatPage(*controller, 16);

	// Rigid disk drive page
	controller->SetCmdByte(2, 4);
	disk->SetBlockCount(0x12345678);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	DiskTest_ValidateDrivePage(*controller, 16);

	// Cache page
	controller->SetCmdByte(2, 8);
	disk->Dispatch(scsi_command::eCmdModeSense10);
	DiskTest_ValidateCachePage(*controller, 16);
}

TEST(DiskTest, SynchronizeCache)
{
	auto [controller, disk] = CreateDisk();

	EXPECT_CALL(*disk, FlushCache);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdSynchronizeCache10);
	EXPECT_EQ(status::good, controller->GetStatus());

	EXPECT_CALL(*disk, FlushCache);
	EXPECT_CALL(*controller, Status);
	disk->Dispatch(scsi_command::eCmdSynchronizeCache16);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, ReadDefectData)
{
	auto [controller, disk] = CreateDisk();

	EXPECT_CALL(*controller, DataIn);
	disk->Dispatch(scsi_command::eCmdReadDefectData10);
	EXPECT_EQ(status::good, controller->GetStatus());
}

TEST(DiskTest, SectorSize)
{
	MockDisk disk;

	unordered_set<uint32_t> sizes = { 512, 1024 };
	disk.SetSectorSizes(sizes);
	EXPECT_TRUE(disk.IsSectorSizeConfigurable());

	sizes.clear();
	disk.SetSectorSizes(sizes);
	EXPECT_FALSE(disk.IsSectorSizeConfigurable());

	disk.SetSectorSizeShiftCount(9);
	EXPECT_EQ(9, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(512, disk.GetSectorSizeInBytes());
	disk.SetSectorSizeInBytes(512);
	EXPECT_EQ(9, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(512, disk.GetSectorSizeInBytes());

	disk.SetSectorSizeShiftCount(10);
	EXPECT_EQ(10, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(1024, disk.GetSectorSizeInBytes());
	disk.SetSectorSizeInBytes(1024);
	EXPECT_EQ(10, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(1024, disk.GetSectorSizeInBytes());

	disk.SetSectorSizeShiftCount(11);
	EXPECT_EQ(11, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(2048, disk.GetSectorSizeInBytes());
	disk.SetSectorSizeInBytes(2048);
	EXPECT_EQ(11, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(2048, disk.GetSectorSizeInBytes());

	disk.SetSectorSizeShiftCount(12);
	EXPECT_EQ(12, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(4096, disk.GetSectorSizeInBytes());
	disk.SetSectorSizeInBytes(4096);
	EXPECT_EQ(12, disk.GetSectorSizeShiftCount());
	EXPECT_EQ(4096, disk.GetSectorSizeInBytes());

	EXPECT_THROW(disk.SetSectorSizeInBytes(1234), io_exception);
}

TEST(DiskTest, ConfiguredSectorSize)
{
	DeviceFactory device_factory;
	MockSCSIHD disk(0, {}, false);

	EXPECT_TRUE(disk.SetConfiguredSectorSize(device_factory, 512));
	EXPECT_EQ(512, disk.GetConfiguredSectorSize());

	EXPECT_FALSE(disk.SetConfiguredSectorSize(device_factory, 1234));
	EXPECT_EQ(512, disk.GetConfiguredSectorSize());
}

TEST(DiskTest, BlockCount)
{
	MockDisk disk;

	disk.SetBlockCount(0x1234567887654321);
	EXPECT_EQ(0x1234567887654321, disk.GetBlockCount());
}