package com.estate.expert.admin.service;

import com.estate.expert.admin.domain.standard.hierachy.Hierarchy;
import com.estate.expert.admin.domain.standard.hierachy.HierarchyData;
import com.estate.expert.admin.utils.tree.SimpleTreeRoot;
import com.estate.expert.admin.utils.tree.TreeNode;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.transaction.annotation.Transactional;

/**
 * @author ligf
 * @Description
 * @date 2023-09-05
 **/
@Slf4j
@SpringBootTest
@ExtendWith(SpringExtension.class)
@Transactional
public class
DepartmentServiceTest {

    @Autowired
    private HierarchyService departmentService;

    private Hierarchy hierarchy;

    @BeforeEach
    public void setUp() {
        TreeNode<HierarchyData> root = new TreeNode<HierarchyData>("总公司");
        root.setName("总公司");

        var c0 = root.addChild("子公司0");
        var c00 = c0.addChild("子公司0-0");
        var c01 = c0.addChild("子公司0-1");
        var c1 = root.addChild("子公司1");
        var c2 = root.addChild("子公司2");
        var c20 = c2.addChild("子公司2-1");

        hierarchy = new Hierarchy();
        hierarchy.setContent(new SimpleTreeRoot<>(root));

        departmentService.add(null, hierarchy);
    }

    @Test
    public void testRoot() {
        TreeNode<HierarchyData> root = departmentService.root(hierarchy.getId()).root();
        Assertions.assertEquals("总公司", root.getName());
        Assertions.assertEquals("/", root.getPath());
        Assertions.assertEquals(3, root.getChildren().size());
    }

    @Test
    public void testGet() {
        var department = departmentService.node(hierarchy.getId(),"/0", true);
        Assertions.assertEquals("子公司0", department.getName());
    }

    @Test
    public void testEdit() {
        departmentService.edit(hierarchy.getId(),"/0/0", "子公司0-0-X",null);
        var department = departmentService.node(hierarchy.getId(),"/0", true);
        Assertions.assertEquals("子公司0-0-X", department.getName());
    }

    @Test
    public void testAddChild() {
        departmentService.addChild(hierarchy.getId(),"/0", "子公司0-2",null);
        var department = departmentService.node(hierarchy.getId(),"/0/2", true);
        Assertions.assertEquals("子公司0-2", department.getName());
    }

    @Test
    public void testDeleteNode() {
        departmentService.deleteNode( hierarchy.getId(),"/0/0");
        var department = departmentService.node(hierarchy.getId(), "/0/0", true);
        Assertions.assertNull(department);
    }

    @Test
    public void testGetLeafNodes() {
        var root = departmentService.root(hierarchy.getId()).root();
        var leafNodes = root.getLeafNodes();
        Assertions.assertEquals(4, leafNodes.size());
    }

}