import org.cesnet.*;

import org.cesnet.Module;

import static org.cesnet.LYD_FORMAT.LYD_XML;

import static org.cesnet.LYS_INFORMAT.LYS_IN_YANG;

import static org.cesnet.LYS_INFORMAT.LYS_IN_YIN;

import static org.cesnet.yangConstants.LYD_OPT_CONFIG;

import static org.cesnet.yangConstants.LYD_OPT_STRICT;

import static org.cesnet.yangConstants.LYD_VAL_OK;

import org.junit.Test;

import static org.junit.Assert.*;

public class LibyangTest {

    static {

        System.loadLibrary("yangJava");

    }

    @Test

    public void test_ly_ctx_new(){

        String yang_folder1 = Constants.TESTS_DIR + "/data/files";

        String yang_folder2 = Constants.TESTS_DIR + "/data:" + Constants.TESTS_DIR + "/data/files";

        try {

            Context ctx = new Context(yang_folder1);

            assertNotNull(ctx);

            vector_String list = ctx.get_searchdirs();

            assertEquals(1, list.size());

            ctx = new Context(yang_folder2);

            assertNotNull(ctx);

            list = ctx.get_searchdirs();

            assertEquals(2, list.size());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_new_invalid(){

        String yang_folder = "INVALID_PATH";

        try {

            Context ctx = new Context(yang_folder);

            throw new Exception("exception not thrown");

        } catch(Exception e) {

            assertTrue(e.getMessage().contains("No Context"));

        }

    }

    @Test

    public void test_ly_ctx_get_searchdirs() {

        String yang_folder = Constants.TESTS_DIR + "/data/files";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            vector_String list = ctx.get_searchdirs();

            assertEquals(1, list.size());

            assertEquals(yang_folder, list.get(0));

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void  test_ly_ctx_set_searchdir() {

        String yang_folder = Constants.TESTS_DIR + "/data/files";

        String new_yang_folder = Constants.TESTS_DIR + "/schema/yin";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            vector_String list = ctx.get_searchdirs();

            assertEquals(1, list.size());

            assertEquals(yang_folder, list.get(0));

            ctx.set_searchdir(new_yang_folder);

            list = ctx.get_searchdirs();

            assertEquals(2, list.size());

            assertEquals(new_yang_folder, list.get(1));

            ctx.unset_searchdirs(0);

            list = ctx.get_searchdirs();

            assertEquals(1, list.size());

            assertEquals(new_yang_folder, list.get(0));

        } catch(Exception e ) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_set_searchdir_invalid() {

        String yang_folder = Constants.TESTS_DIR + "/data/files";

        String new_yang_folder = Constants.TESTS_DIR + "INVALID_PATH";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            ctx.set_searchdir(new_yang_folder);

            throw new Exception("exception not thrown");

        } catch(Exception e) {

            assertTrue(e.getMessage().contains(new_yang_folder));

        }

    }

    @Test

    public void test_ly_ctx_info(){

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Data_Node info = ctx.info();

            assertNotNull(info);

            assertEquals(LYD_VAL_OK, info.validity());

        } catch(Exception e ) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_load_module_invalid() {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.load_module("invalid", null);

            throw new Exception("exception not thrown");

        } catch(Exception e ) {

            assertTrue(e.getMessage().contains("invalid"));

        }

    }

    @Test

    public void test_ly_ctx_load_get_module()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String name1 = "a";

        String name2 = "b";

        String revision = "2016-03-01";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.get_module("invalid");

            assertNull(module);

            module = ctx.get_module(name1);

            assertNull(module);

            module = ctx.load_module(name1);

            assertNotNull(module);

            assertEquals(name1, module.name());

            module = ctx.load_module(name2, revision);

            assertNotNull(module);

            assertEquals(name2, module.name());

            assertEquals(revision, module.rev().date());

            module = ctx.get_module(name2, "INVALID_REVISION");

            assertNull(module);

            module = ctx.get_module(name1);

            assertNotNull(module);

            assertEquals(name1, module.name());

            module = ctx.get_module(name2, revision);

            assertNotNull(module);

            assertEquals(name2, module.name());

            assertEquals(revision, module.rev().date());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_get_module_older()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String name = "b";

        String revision = "2016-03-01";

        String revision_older = "2015-01-01";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.load_module("c");

            assertNotNull(module);

            assertEquals("c", module.name());

            module = ctx.load_module(name, revision);

            assertNotNull(module);

            assertEquals(name, module.name());

            assertEquals(revision, module.rev().date());

            Module module_older = ctx.get_module_older(module);

            assertNotNull(module_older);

            assertEquals(name, module_older.name());

            assertEquals(revision_older, module_older.rev().date());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_get_module_by_ns()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String module_name = "a";

        String ns = "urn:a";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.load_module(module_name);

            assertNotNull(module);

            assertEquals(module_name, module.name());

            module = ctx.get_module_by_ns(ns);

            assertNotNull(module);

            assertEquals(module_name, module.name());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_clean()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String module_name = "a";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.load_module(module_name);

            assertNotNull(module);

            assertEquals(module_name, module.name());

            module = ctx.get_module(module_name);

            assertNotNull(module);

            assertEquals(module_name, module.name());

            ctx.clean();

            module = ctx.get_module(module_name);

            assertNull(module);

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_parse_module_path()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String yin_file = Constants.TESTS_DIR + "/api/files/a.yin";

        String yang_file = Constants.TESTS_DIR + "/api/files/b.yang";

        String module_name1 = "a";

        String module_name2 = "b";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.parse_module_path(yin_file, LYS_IN_YIN);

            assertNotNull(module);

            assertEquals(module_name1, module.name());

            module = ctx.parse_module_path(yang_file, LYS_IN_YANG);

            assertNotNull(module);

            assertEquals(module_name2, module.name());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_parse_module_path_invalid()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            Module module = ctx.parse_module_path("INVALID_YANG_FILE", LYS_IN_YANG);

            throw new Exception("exception not thrown");

        } catch(Exception e) {

            assertTrue(e.getMessage().contains("INVALID_YANG_FILE"));

        }

    }

    @Test

    public void test_ly_ctx_get_submodule()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String yin_file = Constants.TESTS_DIR + "/api/files/a.yin";

        String module_name = "a";

        String sub_name = "asub";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            ctx.parse_module_path(yin_file, LYS_IN_YIN);

            Submodule submodule = ctx.get_submodule(module_name, null, sub_name, null);

            assertNotNull(submodule);

            assertEquals(sub_name, submodule.name());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_ctx_get_submodule2()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String yin_file = Constants.TESTS_DIR + "/api/files/a.yin";

        String config_file = Constants.TESTS_DIR + "/api/files/a.xml";

        String sub_name = "asub";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            ctx.parse_module_path(yin_file, LYS_IN_YIN);

            Data_Node root = ctx.parse_data_path(config_file, LYD_XML, LYD_OPT_CONFIG | LYD_OPT_STRICT);

            assertNotNull(root);

            assertNotNull(root.schema().module());

            Submodule submodule = ctx.get_submodule2(root.schema().module(), sub_name);

            assertNotNull(submodule);

            assertEquals(sub_name, submodule.name());

        } catch(Exception e) {

            fail(e.getMessage());

        return;

    }

    }

    @Test

    public void test_ly_ctx_find_path()

    {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String yin_file = Constants.TESTS_DIR + "/api/files/a.yin";

        String yang_file = Constants.TESTS_DIR + "/api/files/b.yang";

        String schema_path1 = "/b:x/b:bubba";

        String schema_path2 = "/a:x/a:bubba";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            ctx.parse_module_path(yang_file, LYS_IN_YANG);

            Set set = ctx.find_path(schema_path1);

            assertNotNull(set);

            ctx.parse_module_path(yin_file, LYS_IN_YIN);

            set = ctx.find_path(schema_path2);

            assertNotNull(set);

            new Set();

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    @Test

    public void test_ly_set() {

        String yang_folder = Constants.TESTS_DIR + "/api/files";

        String yin_file = Constants.TESTS_DIR + "/api/files/a.yin";

        String config_file = Constants.TESTS_DIR + "/api/files/a.xml";

        try {

            Context ctx = new Context(yang_folder);

            assertNotNull(ctx);

            ctx.parse_module_path(yin_file, LYS_IN_YIN);

            Data_Node root = ctx.parse_data_path(config_file, LYD_XML, LYD_OPT_CONFIG | LYD_OPT_STRICT);

            assertNotNull(root);

            Set set = new Set();

            assertNotNull(set);

            assertEquals(0, set.number());

            set.add(root.child().schema());

            assertEquals(1, set.number());

            set.add(root.schema());

            assertEquals(2, set.number());

            set.rm(root.schema());

            assertEquals(1, set.number());

            set.add(root.schema());

            assertEquals(2, set.number());

            set.rm_index(1);

            assertEquals(1, set.number());

            set.clean();

            assertEquals(0, set.number());

        } catch(Exception e) {

            fail(e.getMessage());

        }

    }

    public static void main(String[] args) {

        new LibyangTest();

    }

}