brotherton-erpnext/erpnext/tests/ui_test_helpers.py

import frappe
from frappe.utils import getdate

@frappe.whitelist()
def create_employee_records():
	create_company()
	create_missing_designation()

	emp1 = create_employee('Test Employee 1', 'CEO')
	emp2 = create_employee('Test Employee 2', 'CTO')
	emp3 = create_employee('Test Employee 3', 'Head of Marketing and Sales', emp1)
	emp4 = create_employee('Test Employee 4', 'Project Manager', emp2)
	emp5 = create_employee('Test Employee 5', 'Engineer', emp2)
	emp6 = create_employee('Test Employee 6', 'Analyst', emp3)
	emp7 = create_employee('Test Employee 7', 'Software Developer', emp4)

	employees = [emp1, emp2, emp3, emp4, emp5, emp6, emp7]
	return employees

@frappe.whitelist()
def get_employee_records():
	return frappe.db.get_list('Employee', filters={
		'company': 'Test Org Chart'
	}, pluck='name', order_by='name')

def create_company():
	company = frappe.db.exists('Company', 'Test Org Chart')
	if not company:
		company = frappe.get_doc({
			'doctype': 'Company',
			'company_name': 'Test Org Chart',
			'country': 'India',
			'default_currency': 'INR'
		}).insert().name

	return company

def create_employee(first_name, designation, reports_to=None):
	employee = frappe.db.exists('Employee', {'first_name': first_name, 'designation': designation})
	if not employee:
		employee = frappe.get_doc({
			'doctype': 'Employee',
			'first_name': first_name,
			'company': 'Test Org Chart',
			'gender': 'Female',
			'date_of_birth': getdate('08-12-1998'),
			'date_of_joining': getdate('01-01-2021'),
			'designation': designation,
			'reports_to': reports_to
		}).insert().name

	return employee

def create_missing_designation():
	if not frappe.db.exists('Designation', 'CTO'):
		frappe.get_doc({
			'doctype': 'Designation',
			'designation_name': 'CTO'
		}).insert()