package eu.dnetlib.dhp.oa.dedup.dedup;

import com.google.common.collect.Lists;
import com.google.common.hash.HashFunction;
import com.google.common.hash.Hashing;
import eu.dnetlib.dhp.oa.dedup.SparkCreateSimRels;
import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpException;
import eu.dnetlib.pace.config.DedupConfig;
import org.apache.commons.io.IOUtils;
import org.dom4j.DocumentException;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.io.IOException;
import java.util.List;

import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
public class SparkDedupTest {

    @Mock
    SparkCreateSimRels sparkCreateSimRels;

    public List<DedupConfig> prepareConfigurations() throws IOException {

        return Lists.newArrayList(
                DedupConfig.load(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dhp/dedup/conf/org.curr.conf.json"))),
                DedupConfig.load(IOUtils.toString(getClass().getResourceAsStream("/eu/dnetlib/dhp/dedup/conf/org.curr.conf.json")))
        );
    }

    @BeforeEach
    public void setUp() throws IOException, ISLookUpException, DocumentException {

        when(sparkCreateSimRels.getConfigurations(anyString(), anyString())).thenReturn(prepareConfigurations());

    }

    @Test
    public void createSimRelsTest() throws Exception {
        SparkCreateSimRels.main(new String[]{
                "-mt", "local[*]",
                "-i", "/Users/miconis/dumps",
                "-asi", "dedup-similarity-result-levenstein",
                "-la", "lookupurl",
                "-w", "workingPath"
        });
    }

//    @Disabled("must be parametrized to run locally")
//    public void createCCTest() throws Exception {
//
//        SparkCreateConnectedComponent.main(new String[]{
//                "-mt", "local[*]",
//                "-s", "/Users/miconis/dumps",
//                "-e", entity,
//                "-c", ArgumentApplicationParser.compressArgument(configuration),
//                "-t", "/tmp/dedup",
//        });
//    }
//
//    @Disabled("must be parametrized to run locally")
//    public void dedupRecordTest() throws Exception {
//        SparkCreateDedupRecord.main(new String[]{
//                "-mt", "local[*]",
//                "-s", "/Users/miconis/dumps",
//                "-e", entity,
//                "-c", ArgumentApplicationParser.compressArgument(configuration),
//                "-d", "/tmp/dedup",
//        });
//    }
//
//    @Disabled("must be parametrized to run locally")
//    public void printConfiguration() throws Exception {
//        System.out.println(ArgumentApplicationParser.compressArgument(configuration));
//    }

    @Disabled("must be parametrized to run locally")
    public void testHashCode() {
        final String s1 = "20|grid________::6031f94bef015a37783268ec1e75f17f";
        final String s2 = "20|nsf_________::b12be9edf414df8ee66b4c52a2d8da46";

        final HashFunction hashFunction = Hashing.murmur3_128();

        System.out.println(s1.hashCode());
        System.out.println(hashFunction.hashString(s1).asLong());
        System.out.println(s2.hashCode());
        System.out.println(hashFunction.hashString(s2).asLong());
    }
}