"""
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/>.
"""

from tests.base_test import ArchesTestCase
from django.test.utils import captured_stdout
from arches.app.models.models import TileModel
from arches.app.utils.betterJSONSerializer import JSONDeserializer
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 --settings="tests.test_settings"


class mappedArchesJSONImportTests(ArchesTestCase):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        path = "tests/fixtures/data/json/cardinality_test_data/target.json"
        with open(path, "r") as f:
            archesfile = JSONDeserializer().deserialize(f)
        ResourceGraphImporter(archesfile["graph"])

    def test_single_1(self):
        og_tile_count = TileModel.objects.count()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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()
        with captured_stdout():
            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)