#include "GameDependencies.h"
#include <gtest/gtest.h>

#include "BitStream.h"
#include "PetComponent.h"
#include "Entity.h"
#include "eReplicaComponentType.h"
#include "ePetAbilityType.h"
#include "eStateChangeType.h"

class PetTest : public GameDependenciesTest {
protected:
	Entity* baseEntity;
	PetComponent* petComponent;
	CBITSTREAM

	void SetUp() override {
		SetUpDependencies();

		// Set up entity and pet component
		baseEntity = new Entity(15, GameDependenciesTest::info);
		petComponent = baseEntity->AddComponent<PetComponent>(1);

		// Initialize some values to be not default

	}

	void TearDown() override {
		delete baseEntity;
		TearDownDependencies();
	}
};

TEST_F(PetTest, PlacementNewAddComponentTest) {
	using enum PetFlag;

	// Test adding component
	ASSERT_NE(petComponent, nullptr);
	baseEntity->AddComponent<PetComponent>(1);
	ASSERT_NE(baseEntity->GetComponent<PetComponent>(), nullptr);

	// Test getting initial status
	ASSERT_EQ(petComponent->GetParent()->GetObjectID(), 15);
	ASSERT_TRUE(petComponent->HasFlag(NONE));
	ASSERT_EQ(petComponent->GetPetAiState(), PetAiState::spawn);
	ASSERT_EQ(petComponent->GetAbility(), ePetAbilityType::Invalid);
}

/**
 * Test bitset pet flags
*/
TEST_F(PetTest, PetComponentFlagTest) {
	using enum PetFlag;

	// Test setting and reading single flags, exclusively
	petComponent->SetOnlyFlag(NONE);
	ASSERT_TRUE(petComponent->HasOnlyFlag(NONE));
	petComponent->SetOnlyFlag(TAMEABLE);
	ASSERT_TRUE(petComponent->HasOnlyFlag(TAMEABLE));
	ASSERT_FALSE(petComponent->HasOnlyFlag(SPAWNING));

	// Test setting and reading multiple flags, exclusively
	petComponent->SetOnlyFlag(NOT_WAITING, SPAWNING);
	ASSERT_FALSE(petComponent->HasFlag(TAMEABLE));
	ASSERT_TRUE(petComponent->HasFlag(NOT_WAITING));
	ASSERT_TRUE(petComponent->HasFlag(SPAWNING));
	ASSERT_TRUE(petComponent->HasFlag(NOT_WAITING, SPAWNING));
	ASSERT_FALSE(petComponent->HasOnlyFlag(NOT_WAITING));
	ASSERT_FALSE(petComponent->HasOnlyFlag(SPAWNING));
	ASSERT_TRUE(petComponent->HasOnlyFlag(NOT_WAITING, SPAWNING));

	// Test flags are being properly reset for next batch of tests
	petComponent->SetOnlyFlag(NONE);
	ASSERT_TRUE(petComponent->HasFlag(NONE));
	ASSERT_TRUE(petComponent->HasOnlyFlag(NONE));
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING));
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING, SPAWNING, TAMEABLE));

	// Test setting and reading single flags, non-exclusively
	petComponent->SetFlag(NOT_WAITING);
	ASSERT_TRUE(petComponent->HasFlag(NOT_WAITING));
	ASSERT_FALSE(petComponent->HasFlag(SPAWNING));

	// Test setting and reading multiple flags, non-exclusively
	petComponent->SetFlag(TAMEABLE, BEING_TAMED);
	ASSERT_TRUE(petComponent->HasFlag(TAMEABLE, BEING_TAMED));
	ASSERT_TRUE(petComponent->HasFlag(NOT_WAITING));
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING, SPAWNING));
	ASSERT_TRUE(petComponent->HasFlag(NOT_WAITING, TAMEABLE, BEING_TAMED));
	ASSERT_FALSE(petComponent->HasFlag(SPAWNING));
	ASSERT_FALSE(petComponent->HasFlag(SPAWNING, NOT_WAITING, TAMEABLE, BEING_TAMED));

	// Test unsetting and reading multiple flags, non-exclusively
	petComponent->UnsetFlag(NOT_WAITING, SPAWNING);
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING, SPAWNING));
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING));
	ASSERT_TRUE(petComponent->HasFlag(TAMEABLE, BEING_TAMED));
	ASSERT_FALSE(petComponent->HasFlag(NOT_WAITING, TAMEABLE, BEING_TAMED));
	ASSERT_FALSE(petComponent->HasFlag(SPAWNING));
	ASSERT_FALSE(petComponent->HasFlag(SPAWNING, NOT_WAITING, TAMEABLE, BEING_TAMED));
}

TEST_F(PetTest, PetAiState) {
	const auto initialState = petComponent->GetPetAiState();
	ASSERT_EQ(initialState, PetAiState::spawn);

	petComponent->SetPetAiState(PetAiState::follow);
	ASSERT_EQ(PetAiState::follow, petComponent->GetPetAiState());

	petComponent->SetPetAiState(PetAiState::idle);
	ASSERT_EQ(PetAiState::idle, petComponent->GetPetAiState());

	petComponent->SetPetAiState(PetAiState::interact);
	ASSERT_EQ(PetAiState::interact, petComponent->GetPetAiState());

	petComponent->SetPetAiState(PetAiState::despawn);
	ASSERT_EQ(PetAiState::despawn, petComponent->GetPetAiState());
}

// Test the pet use logic
TEST_F(PetTest, PetUse) {
	ASSERT_FALSE(petComponent->IsReadyToInteract());

	petComponent->SetIsReadyToInteract(true);
	ASSERT_TRUE(petComponent->IsReadyToInteract());

	// Test bouncer logic
	ASSERT_FALSE(petComponent->IsHandlingInteraction());
	petComponent->SetAbility(ePetAbilityType::JumpOnObject);
	ASSERT_EQ(petComponent->GetAbility(), ePetAbilityType::JumpOnObject);
	petComponent->SetInteractType(PetInteractType::bouncer);
	petComponent->OnUse(baseEntity);

	// need to add a destroyable component to the test entity and test the imagination drain

}