import unittest

import frappe
from frappe.utils import add_days, getdate

from erpnext.hr.doctype.employee.test_employee import make_employee
from erpnext.hr.doctype.exit_interview.test_exit_interview import create_exit_interview
from erpnext.hr.doctype.full_and_final_statement.test_full_and_final_statement import (
	create_full_and_final_statement,
)
from erpnext.hr.report.employee_exits.employee_exits import execute


class TestEmployeeExits(unittest.TestCase):
	@classmethod
	def setUpClass(cls):
		create_company()
		frappe.db.sql("delete from `tabEmployee` where company='Test Company'")
		frappe.db.sql("delete from `tabFull and Final Statement` where company='Test Company'")
		frappe.db.sql("delete from `tabExit Interview` where company='Test Company'")

		cls.create_records()

	@classmethod
	def tearDownClass(cls):
		frappe.db.rollback()

	@classmethod
	def create_records(cls):
		cls.emp1 = make_employee(
			"employeeexit1@example.com",
			company="Test Company",
			date_of_joining=getdate("01-10-2021"),
			relieving_date=add_days(getdate(), 14),
			designation="Accountant",
		)
		cls.emp2 = make_employee(
			"employeeexit2@example.com",
			company="Test Company",
			date_of_joining=getdate("01-12-2021"),
			relieving_date=add_days(getdate(), 15),
			designation="Accountant",
		)

		cls.emp3 = make_employee(
			"employeeexit3@example.com",
			company="Test Company",
			date_of_joining=getdate("02-12-2021"),
			relieving_date=add_days(getdate(), 29),
			designation="Engineer",
		)
		cls.emp4 = make_employee(
			"employeeexit4@example.com",
			company="Test Company",
			date_of_joining=getdate("01-12-2021"),
			relieving_date=add_days(getdate(), 30),
			designation="Engineer",
		)

		# exit interview for 3 employees only
		cls.interview1 = create_exit_interview(cls.emp1)
		cls.interview2 = create_exit_interview(cls.emp2)
		cls.interview3 = create_exit_interview(cls.emp3)

		# create fnf for some records
		cls.fnf1 = create_full_and_final_statement(cls.emp1)
		cls.fnf2 = create_full_and_final_statement(cls.emp2)

		# link questionnaire for a few records
		# setting employee doctype as reference instead of creating a questionnaire
		# since this is just for a test
		frappe.db.set_value(
			"Exit Interview",
			cls.interview1.name,
			{"ref_doctype": "Employee", "reference_document_name": cls.emp1},
		)

		frappe.db.set_value(
			"Exit Interview",
			cls.interview2.name,
			{"ref_doctype": "Employee", "reference_document_name": cls.emp2},
		)

		frappe.db.set_value(
			"Exit Interview",
			cls.interview3.name,
			{"ref_doctype": "Employee", "reference_document_name": cls.emp3},
		)

	def test_employee_exits_summary(self):
		filters = {
			"company": "Test Company",
			"from_date": getdate(),
			"to_date": add_days(getdate(), 15),
			"designation": "Accountant",
		}

		report = execute(filters)

		employee1 = frappe.get_doc("Employee", self.emp1)
		employee2 = frappe.get_doc("Employee", self.emp2)
		expected_data = [
			{
				"employee": employee1.name,
				"employee_name": employee1.employee_name,
				"date_of_joining": employee1.date_of_joining,
				"relieving_date": employee1.relieving_date,
				"department": employee1.department,
				"designation": employee1.designation,
				"reports_to": None,
				"exit_interview": self.interview1.name,
				"interview_status": self.interview1.status,
				"employee_status": "",
				"questionnaire": employee1.name,
				"full_and_final_statement": self.fnf1.name,
			},
			{
				"employee": employee2.name,
				"employee_name": employee2.employee_name,
				"date_of_joining": employee2.date_of_joining,
				"relieving_date": employee2.relieving_date,
				"department": employee2.department,
				"designation": employee2.designation,
				"reports_to": None,
				"exit_interview": self.interview2.name,
				"interview_status": self.interview2.status,
				"employee_status": "",
				"questionnaire": employee2.name,
				"full_and_final_statement": self.fnf2.name,
			},
		]

		self.assertEqual(expected_data, report[1])  # rows

	def test_pending_exit_interviews_summary(self):
		filters = {
			"company": "Test Company",
			"from_date": getdate(),
			"to_date": add_days(getdate(), 30),
			"exit_interview_pending": 1,
		}

		report = execute(filters)

		employee4 = frappe.get_doc("Employee", self.emp4)
		expected_data = [
			{
				"employee": employee4.name,
				"employee_name": employee4.employee_name,
				"date_of_joining": employee4.date_of_joining,
				"relieving_date": employee4.relieving_date,
				"department": employee4.department,
				"designation": employee4.designation,
				"reports_to": None,
				"exit_interview": None,
				"interview_status": None,
				"employee_status": None,
				"questionnaire": None,
				"full_and_final_statement": None,
			}
		]

		self.assertEqual(expected_data, report[1])  # rows

	def test_pending_exit_questionnaire_summary(self):
		filters = {
			"company": "Test Company",
			"from_date": getdate(),
			"to_date": add_days(getdate(), 30),
			"questionnaire_pending": 1,
		}

		report = execute(filters)

		employee4 = frappe.get_doc("Employee", self.emp4)
		expected_data = [
			{
				"employee": employee4.name,
				"employee_name": employee4.employee_name,
				"date_of_joining": employee4.date_of_joining,
				"relieving_date": employee4.relieving_date,
				"department": employee4.department,
				"designation": employee4.designation,
				"reports_to": None,
				"exit_interview": None,
				"interview_status": None,
				"employee_status": None,
				"questionnaire": None,
				"full_and_final_statement": None,
			}
		]

		self.assertEqual(expected_data, report[1])  # rows

	def test_pending_fnf_summary(self):
		filters = {"company": "Test Company", "fnf_pending": 1}

		report = execute(filters)

		employee3 = frappe.get_doc("Employee", self.emp3)
		employee4 = frappe.get_doc("Employee", self.emp4)
		expected_data = [
			{
				"employee": employee3.name,
				"employee_name": employee3.employee_name,
				"date_of_joining": employee3.date_of_joining,
				"relieving_date": employee3.relieving_date,
				"department": employee3.department,
				"designation": employee3.designation,
				"reports_to": None,
				"exit_interview": self.interview3.name,
				"interview_status": self.interview3.status,
				"employee_status": "",
				"questionnaire": employee3.name,
				"full_and_final_statement": None,
			},
			{
				"employee": employee4.name,
				"employee_name": employee4.employee_name,
				"date_of_joining": employee4.date_of_joining,
				"relieving_date": employee4.relieving_date,
				"department": employee4.department,
				"designation": employee4.designation,
				"reports_to": None,
				"exit_interview": None,
				"interview_status": None,
				"employee_status": None,
				"questionnaire": None,
				"full_and_final_statement": None,
			},
		]

		self.assertEqual(expected_data, report[1])  # rows


def create_company():
	if not frappe.db.exists("Company", "Test Company"):
		frappe.get_doc(
			{
				"doctype": "Company",
				"company_name": "Test Company",
				"default_currency": "INR",
				"country": "India",
			}
		).insert()