package com.estate.expert.admin.service;

import com.estate.expert.admin.service.eval.EvalFuzzyService;
import com.estate.expert.admin.service.eval.EvalScoreService;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.junit.jupiter.SpringExtension;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;

@Slf4j
@SpringBootTest
@ExtendWith(SpringExtension.class)
@Transactional
@Rollback(value = false)
class EvalServiceTest {

    @Resource
    private EvalScoreService evalScoreService;

    @Resource
    private EvalFuzzyService evalFuzzyService;

    @Resource
    private CheckTaskService checkTaskService;

    @Resource
    private ObjectMapper objectMapper;

    @Test
    void scoreEval() throws IOException {
        var checkTask = checkTaskService.get(1269);
//        var checkTask = checkTaskService.get(241, true);
        var res = evalScoreService.eval(checkTask);

        log.info("Check Task {} {}", checkTask.getPlan().getName(), checkTask.getProject().getName());

        // save to tmp
        var str = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(res.getSystemRes());
        Files.write(Path.of("/Volumes/Works/tmp/eval_score.json"), str.getBytes());

        var resMap = res.getDeviceResMap();
        for (var entry : resMap.entrySet()) {
            var device = entry.getKey();
            var deviceRes = entry.getValue();
            var deviceStr = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(deviceRes);
            Files.write(Path.of("/Volumes/Works/tmp/eval_score_" + device + ".json"), deviceStr.getBytes());
        }
    }

    @Test
    void fuzzyEval() throws IOException {
        var checkTask = checkTaskService.get(1347, true);
//        var checkTask = checkTaskService.get(527, true);
        var res = evalFuzzyService.eval(checkTask);

        // save to tmp
        var str = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(res.getSystemRes());
        Files.write(Path.of("/Volumes/Works/tmp/eval_fuzzy.json"), str.getBytes());

        var resMap = res.getDeviceResMap();
        for (var entry : resMap.entrySet()) {
            var device = entry.getKey();
            var deviceRes = entry.getValue();
            var deviceStr = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(deviceRes);
            Files.write(Path.of("/Volumes/Works/tmp/eval_fuzzy_" + device + ".json"), deviceStr.getBytes());
        }
    }

}