Unit test cases added

Author: Ravikirana-Microsoft

src/tests/backend/kernal_tools/test_hr_tools.py

@@ -0,0 +1,105 @@
+"""Test cases for ProcurementTools class."""
+import pytest
+from src.backend.kernel_tools.hr_tools import HrTools
+
+
+@pytest.mark.asyncio
+async def test_schedule_orientation_session():
+    """Test schedule_orientation_session method."""
+    result = await HrTools.schedule_orientation_session("John Doe", "2025-05-20")
+    assert "Orientation Session Scheduled" in result
+    assert "**Employee Name:** John Doe" in result
+    assert "**Date:** 2025-05-20" in result
+
+
+@pytest.mark.asyncio
+async def test_assign_mentor():
+    """Test assign_mentor method."""
+    result = await HrTools.assign_mentor("Jane Doe")
+    assert "Mentor Assigned" in result
+    assert "**Employee Name:** Jane Doe" in result
+
+
+@pytest.mark.asyncio
+async def test_register_for_benefits():
+    """Test register_for_benefits method."""
+    result = await HrTools.register_for_benefits("John Doe")
+    assert "Benefits Registration" in result
+    assert "**Employee Name:** John Doe" in result
+
+
+@pytest.mark.asyncio
+async def test_enroll_in_training_program():
+    """Test enroll_in_training_program method."""
+    result = await HrTools.enroll_in_training_program("John Doe", "Leadership Training")
+    assert "Training Program Enrollment" in result
+    assert "**Employee Name:** John Doe" in result
+    assert "**Program Name:** Leadership Training" in result
+
+
+@pytest.mark.asyncio
+async def test_provide_employee_handbook():
+    """Test provide_employee_handbook method."""
+    result = await HrTools.provide_employee_handbook("Jane Doe")
+    assert "Employee Handbook Provided" in result
+    assert "**Employee Name:** Jane Doe" in result
+
+
+@pytest.mark.asyncio
+async def test_update_employee_record():
+    """Test update_employee_record method."""
+    result = await HrTools.update_employee_record("John Doe", "Address", "123 Main St")
+    assert "Employee Record Updated" in result
+    assert "**Field Updated:** Address" in result
+    assert "**New Value:** 123 Main St" in result
+
+
+@pytest.mark.asyncio
+async def test_request_id_card():
+    """Test request_id_card method."""
+    result = await HrTools.request_id_card("John Doe")
+    assert "ID Card Request" in result
+    assert "**Employee Name:** John Doe" in result
+
+
+@pytest.mark.asyncio
+async def test_set_up_payroll():
+    """Test set_up_payroll method."""
+    result = await HrTools.set_up_payroll("Jane Doe")
+    assert "Payroll Setup" in result
+    assert "**Employee Name:** Jane Doe" in result
+
+
+@pytest.mark.asyncio
+async def test_add_emergency_contact():
+    """Test add_emergency_contact method."""
+    result = await HrTools.add_emergency_contact("John Doe", "Jane Smith", "555-1234")
+    assert "Emergency Contact Added" in result
+    assert "**Contact Name:** Jane Smith" in result
+    assert "**Contact Phone:** 555-1234" in result
+
+
+@pytest.mark.asyncio
+async def test_process_leave_request():
+    """Test process_leave_request method."""
+    result = await HrTools.process_leave_request("John Doe", "Vacation", "2025-06-01", "2025-06-10")
+    assert "Leave Request Processed" in result
+    assert "**Leave Type:** Vacation" in result
+    assert "**Start Date:** 2025-06-01" in result
+    assert "**End Date:** 2025-06-10" in result
+
+
+def test_get_all_kernel_functions():
+    """Test get_all_kernel_functions method."""
+    kernel_functions = HrTools.get_all_kernel_functions()
+    assert "schedule_orientation_session" in kernel_functions
+    assert "assign_mentor" in kernel_functions
+    assert callable(kernel_functions["schedule_orientation_session"])
+
+
+def test_generate_tools_json_doc():
+    """Test generate_tools_json_doc method."""
+    tools_json = HrTools.generate_tools_json_doc()
+    assert "schedule_orientation_session" in tools_json
+    assert "assign_mentor" in tools_json
+    assert "arguments" in tools_json

src/tests/backend/kernal_tools/test_procurement_tools.py

@@ -0,0 +1,140 @@
+"""Test cases for ProcurementTools class."""
+import pytest
+from src.backend.kernel_tools.procurement_tools import ProcurementTools
+
+
+@pytest.mark.asyncio
+async def test_order_hardware():
+    """Test order_hardware method."""
+    result = await ProcurementTools.order_hardware("Laptop", 5)
+    assert "Hardware Order Placed" in result
+    assert "**Item:** Laptop" in result
+    assert "**Quantity:** 5" in result
+
+
+@pytest.mark.asyncio
+async def test_order_software_license():
+    """Test order_software_license method."""
+    result = await ProcurementTools.order_software_license("Microsoft Office", "Enterprise", 10)
+    assert "Software License Ordered" in result
+    assert "**Software:** Microsoft Office" in result
+    assert "**License Type:** Enterprise" in result
+    assert "**Quantity:** 10" in result
+
+
+@pytest.mark.asyncio
+async def test_check_inventory():
+    """Test check_inventory method."""
+    result = await ProcurementTools.check_inventory("Laptop")
+    assert "Inventory Status" in result
+    assert "**Item:** Laptop" in result
+    assert "**Status:** In Stock" in result
+
+
+@pytest.mark.asyncio
+async def test_process_purchase_order():
+    """Test process_purchase_order method."""
+    result = await ProcurementTools.process_purchase_order("PO12345")
+    assert "Purchase Order Processed" in result
+    assert "**PO Number:** PO12345" in result
+
+
+@pytest.mark.asyncio
+async def test_initiate_contract_negotiation():
+    """Test initiate_contract_negotiation method."""
+    result = await ProcurementTools.initiate_contract_negotiation("Vendor A", "Contract Details")
+    assert "Contract Negotiation Initiated" in result
+    assert "**Vendor:** Vendor A" in result
+    assert "**Contract Details:** Contract Details" in result
+
+
+@pytest.mark.asyncio
+async def test_approve_invoice():
+    """Test approve_invoice method."""
+    result = await ProcurementTools.approve_invoice("INV12345")
+    assert "Invoice Approved" in result
+    assert "**Invoice Number:** INV12345" in result
+
+
+@pytest.mark.asyncio
+async def test_track_order():
+    """Test track_order method."""
+    result = await ProcurementTools.track_order("ORD12345")
+    assert "Order Tracking" in result
+    assert "**Order Number:** ORD12345" in result
+    assert "**Status:** In Transit" in result
+
+
+@pytest.mark.asyncio
+async def test_manage_vendor_relationship():
+    """Test manage_vendor_relationship method."""
+    result = await ProcurementTools.manage_vendor_relationship("Vendor A", "updated")
+    assert "Vendor Relationship Update" in result
+    assert "**Vendor:** Vendor A" in result
+    assert "**Action:** updated" in result
+
+
+@pytest.mark.asyncio
+async def test_update_procurement_policy():
+    """Test update_procurement_policy method."""
+    result = await ProcurementTools.update_procurement_policy("Policy A", "New Policy Content")
+    assert "Procurement Policy Updated" in result
+    assert "**Policy:** Policy A" in result
+
+
+def test_get_all_kernel_functions():
+    """Test get_all_kernel_functions method."""
+    kernel_functions = ProcurementTools.get_all_kernel_functions()
+    assert "order_hardware" in kernel_functions
+    assert "order_software_license" in kernel_functions
+    assert callable(kernel_functions["order_hardware"])
+
+
+def test_generate_tools_json_doc():
+    """Test generate_tools_json_doc method."""
+    tools_json = ProcurementTools.generate_tools_json_doc()
+    assert "order_hardware" in tools_json
+    assert "order_software_license" in tools_json
+    assert "arguments" in tools_json
+
+
+@pytest.mark.asyncio
+async def test_order_hardware_invalid_quantity():
+    """Test order_hardware with invalid quantity."""
+    result = await ProcurementTools.order_hardware("Laptop", 0)
+    assert "Hardware Order Placed" in result
+    assert "**Quantity:** 0" in result
+
+
+@pytest.mark.asyncio
+async def test_order_software_license_invalid_quantity():
+    """Test order_software_license with invalid quantity."""
+    result = await ProcurementTools.order_software_license("Microsoft Office", "Enterprise", -1)
+    assert "Software License Ordered" in result
+    assert "**Quantity:** -1" in result
+
+
+# @pytest.mark.asyncio
+# async def test_check_inventory_item_not_found():
+#     """Test check_inventory with an item not found."""
+#     result = await ProcurementTools.check_inventory("NonExistentItem")
+#     assert "Inventory Status" in result
+#     assert "**Item:** NonExistentItem" in result
+#     assert "**Status:** Not Found" in result
+
+
+@pytest.mark.asyncio
+async def test_process_purchase_order_invalid_po():
+    """Test process_purchase_order with an invalid PO number."""
+    result = await ProcurementTools.process_purchase_order("")
+    assert "Purchase Order Processed" in result
+    assert "**PO Number:** " in result
+
+
+@pytest.mark.asyncio
+async def test_initiate_contract_negotiation_empty_details():
+    """Test initiate_contract_negotiation with empty contract details."""
+    result = await ProcurementTools.initiate_contract_negotiation("Vendor A", "")
+    assert "Contract Negotiation Initiated" in result
+    assert "**Vendor:** Vendor A" in result
+    assert "**Contract Details:** " in result