"""
ARCHES - a program developed to inventory and manage immovable cultural heritage.
Copyright (C) 2013 J. Paul Getty Trust and World Monuments Fund

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.

You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""

import os
from operator import itemgetter
from tests import test_settings
from tests.base_test import ArchesTestCase
from django.core import management
from arches.app.models.models import TileModel, ResourceInstance
from arches.app.utils.betterJSONSerializer import JSONSerializer, JSONDeserializer
from arches.app.search.search_engine_factory import SearchEngineFactory
from arches.app.utils.data_management.resource_graphs.importer import import_graph as ResourceGraphImporter
from arches.app.utils.data_management.resources.importer import BusinessDataImporter as ArchesFileImporter


# these tests can be run from the command line via
# python manage.py test tests/models/mapped_archesjson_import_tests.py --pattern="*.py" --settings="tests.test_settings"


class mappedArchesJSONImportTests(ArchesTestCase):
    @classmethod
    def setUpClass(cls):
        pass

    def setUp(self):
        ResourceInstance.objects.all().delete()
        with open(os.path.join("tests/fixtures/data/json/cardinality_test_data/target.json"), "rU") as f:
            archesfile = JSONDeserializer().deserialize(f)
        ResourceGraphImporter(archesfile["graph"])

    @classmethod
    def tearDownClass(cls):
        pass

    def test_single_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/single-1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 1)

    def test_single_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/single-n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 3)

    def test_single_n_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json",
            "tests/fixtures/data/json/cardinality_test_data/single-n_1.mapping",
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 1)

    def test_single_n_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json",
            "tests/fixtures/data/json/cardinality_test_data/single-n_n.mapping",
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 3)

    def test_1_1_1_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_1_1_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_1_1_n_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_1_n_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_1_1_1_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_1_1_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_1_1_n_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_1_n_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_n_1_n_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_1_n_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 6)

    def test_n_1_n_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_1_n_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 6)

    def test_1_n_1_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_n_1_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 5)

    def test_1_n_n_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_n_n_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 5)

    def test_n_n_n_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_n_n_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 12)

    def test_n_n_1_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_n_1_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 4)

    def test_n_1_1_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_1_1_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_n_1_1_n(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_1_1_n.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_n_n_1_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_1_1_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_1_n_1_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_n_1_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_1_n_n_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/1_n_n_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_n_n_1_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_n_1_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 2)

    def test_n_n_n_1(self):
        og_tile_count = TileModel.objects.count()
        ArchesFileImporter(
            "tests/fixtures/data/json/cardinality_test_data/source.json", "tests/fixtures/data/json/cardinality_test_data/n_n_n_1.mapping"
        ).import_business_data()
        new_tile_count = TileModel.objects.count()
        tile_difference = new_tile_count - og_tile_count
        self.assertEqual(tile_difference, 6)