import unittest

import frappe

from erpnext.buying.doctype.purchase_order.test_purchase_order import create_purchase_order
from erpnext.buying.doctype.supplier.test_supplier import create_supplier
from erpnext.tests.utils import ERPNextTestSuite


class TestWebsite(ERPNextTestSuite):
	def test_permission_for_custom_doctype(self):
		create_user("Supplier 1", "supplier1@gmail.com")
		create_user("Supplier 2", "supplier2@gmail.com")

		supplier1 = create_supplier(supplier_name="Supplier1")
		supplier2 = create_supplier(supplier_name="Supplier2")
		supplier1.append("portal_users", {"user": "supplier1@gmail.com"})
		supplier1.save()
		supplier2.append("portal_users", {"user": "supplier2@gmail.com"})
		supplier2.save()

		po1 = create_purchase_order(supplier="Supplier1")
		po2 = create_purchase_order(supplier="Supplier2")

		create_webform()
		create_order_assignment(supplier="Supplier1", po=po1.name)
		create_order_assignment(supplier="Supplier2", po=po2.name)

		# checking if data consist of all order assignment of Supplier1 and Supplier2
		self.assertTrue("Supplier1" and "Supplier2" in [data.supplier for data in get_data()])

		with self.set_user("supplier1@gmail.com"):
			# checking if data only consist of order assignment of Supplier1
			self.assertTrue("Supplier1" in [data.supplier for data in get_data()])
			self.assertFalse([data.supplier for data in get_data() if data.supplier != "Supplier1"])

		with self.set_user("supplier2@gmail.com"):
			# checking if data only consist of order assignment of Supplier2
			self.assertTrue("Supplier2" in [data.supplier for data in get_data()])
			self.assertFalse([data.supplier for data in get_data() if data.supplier != "Supplier2"])


def get_data():
	webform_list_contexts = frappe.get_hooks("webform_list_context")
	if webform_list_contexts:
		context = frappe._dict(frappe.get_attr(webform_list_contexts[0])("Buying") or {})
	kwargs = dict(doctype="Order Assignment", order_by="creation desc")
	return context.get_list(**kwargs)


def create_user(name, email):
	frappe.get_doc(
		{
			"doctype": "User",
			"send_welcome_email": 0,
			"user_type": "Website User",
			"first_name": name,
			"email": email,
			"roles": [{"doctype": "Has Role", "role": "Supplier"}],
		}
	).insert(ignore_if_duplicate=True)


def create_webform():
	frappe.get_doc(
		{
			"doctype": "Web Form",
			"module": "Buying",
			"title": "SO Schedule",
			"route": "so-schedule",
			"doc_type": "Order Assignment",
			"web_form_fields": [
				{
					"doctype": "Web Form Field",
					"fieldname": "po",
					"fieldtype": "Link",
					"options": "Purchase Order",
					"label": "PO",
				},
				{
					"doctype": "Web Form Field",
					"fieldname": "supplier",
					"fieldtype": "Data",
					"label": "Supplier",
				},
			],
		}
	).insert(ignore_if_duplicate=True)


def create_order_assignment(supplier, po):
	frappe.get_doc(
		{
			"doctype": "Order Assignment",
			"po": po,
			"supplier": supplier,
		}
	).insert(ignore_if_duplicate=True)