import { CACHE_MANAGER } from "@nestjs/cache-manager";
import { Test, TestingModule } from "@nestjs/testing";
import { DatabaseService } from "./database/database.service";
import { HealthController } from "./health.controller";

describe("HealthController", () => {
	let controller: HealthController;

	const mockDb = {
		execute: jest.fn().mockResolvedValue([]),
	};

	const mockCacheManager = {
		set: jest.fn().mockResolvedValue(undefined),
	};

	beforeEach(async () => {
		const module: TestingModule = await Test.createTestingModule({
			controllers: [HealthController],
			providers: [
				{
					provide: DatabaseService,
					useValue: {
						db: mockDb,
					},
				},
				{
					provide: CACHE_MANAGER,
					useValue: mockCacheManager,
				},
			],
		}).compile();

		controller = module.get<HealthController>(HealthController);
	});

	it("should return ok if database and redis are connected", async () => {
		mockDb.execute.mockResolvedValue([]);
		mockCacheManager.set.mockResolvedValue(undefined);
		const result = await controller.check();
		expect(result.status).toBe("ok");
		expect(result.database).toBe("connected");
		expect(result.redis).toBe("connected");
	});

	it("should return error if database is disconnected", async () => {
		mockDb.execute.mockRejectedValue(new Error("DB Error"));
		mockCacheManager.set.mockResolvedValue(undefined);
		const result = await controller.check();
		expect(result.status).toBe("error");
		expect(result.database).toBe("disconnected");
		expect(result.databaseError).toBe("DB Error");
		expect(result.redis).toBe("connected");
	});

	it("should return error if redis is disconnected", async () => {
		mockDb.execute.mockResolvedValue([]);
		mockCacheManager.set.mockRejectedValue(new Error("Redis Error"));
		const result = await controller.check();
		expect(result.status).toBe("error");
		expect(result.database).toBe("connected");
		expect(result.redis).toBe("disconnected");
		expect(result.redisError).toBe("Redis Error");
	});
});