package helper;

import base.DemoVo;
import com.yanzuoguang.util.base.CollectionString;
import com.yanzuoguang.util.base.ColorUtils;
import com.yanzuoguang.util.helper.StringHelper;
import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class TestCollectionString {

    @Test
    public void testCh() {
        List<DemoVo> mapList = new ArrayList<>(Arrays.asList(
                new DemoVo("1", "颜佐光"),
                new DemoVo("10000", "成吉思汗"),
                new DemoVo("10001", "毛泽东")
        ));
        System.out.println(CollectionString.getCollectionString("人名:", mapList));
        System.out.println(CollectionString.getCollectionString(StringHelper.EMPTY, mapList));
    }

    @Test
    public void testCh1() {
        List<DemoVo> mapList = new ArrayList<>(Arrays.asList(
                new DemoVo("1", "颜佐光"),
                new DemoVo("10000", "成吉思汗"),
                new DemoVo("10001", "毛泽东")
        ));
        System.out.println(CollectionString.getCollectionString("人名:", mapList, new CollectionString.Handle<DemoVo>() {
            @Override
            public void start(StringBuilder sb, DemoVo row) {
                if (StringHelper.compare(row.getId(), "1")) {
                    sb.append(ColorUtils.Color.RED);
                }
            }

            @Override
            public void end(StringBuilder sb, DemoVo row) {
                if (StringHelper.compare(row.getId(), "1")) {
                    sb.append(ColorUtils.Color.RESET);
                }
            }

            @Override
            public String columnName(StringBuilder sb, String name) {
                return name;
            }

            @Override
            public String columnValue(StringBuilder sb, DemoVo row, String name, String value) {
                if (StringHelper.compare(value, "毛泽东")) {
                    return ColorUtils.colorize(value, ColorUtils.Color.GREEN);
                } else {
                    return value;
                }
            }
        }));
    }


    @Test
    public void testEn() {
        List<DemoVo> mapList = new ArrayList<>(Arrays.asList(
                new DemoVo("1", "yan zuo guang"),
                new DemoVo("10000", "chen jie si han"),
                new DemoVo("10001", "mao ze dong")
        ));
        System.out.println(CollectionString.getCollectionString(StringHelper.EMPTY, mapList));

        System.out.println(CollectionString.getCollectionString("人名:", mapList));
    }
}