from django.test import TestCase

from wagtail.blocks.migrations.operations import (
    AlterBlockValueOperation,
    ListChildrenToStructBlockOperation,
    RemoveStreamChildrenOperation,
    RemoveStructChildrenOperation,
    RenameStreamChildrenOperation,
    RenameStructChildrenOperation,
    StreamChildrenToListBlockOperation,
    StreamChildrenToStreamBlockOperation,
    StreamChildrenToStructBlockOperation,
)
from wagtail.blocks.migrations.utils import apply_changes_to_raw_data
from wagtail.test.streamfield_migrations import factories, models


class FieldChildBlockTest(TestCase):
    """Tests involving changes to top level blocks"""

    def setUp(self):
        raw_data = factories.SampleModelFactory(
            content__0__char1__value="Char Block 1",
            content__1__char2__value="Char Block 2",
            content__2__char1__value="Char Block 1",
            content__3__char2__value="Char Block 2",
        ).content.raw_data
        self.raw_data = raw_data

    def test_rename(self):
        """Rename `char1` blocks to `renamed1`

        Check whether all `char1` blocks have been renamed correctly.
        Check whether ids and values for renamed blocks are intact.
        Check whether other block types are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=RenameStreamChildrenOperation(
                old_name="char1", new_name="renamed1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(altered_raw_data[0]["type"], "renamed1")
        self.assertEqual(altered_raw_data[2]["type"], "renamed1")

        self.assertEqual(altered_raw_data[0]["id"], self.raw_data[0]["id"])
        self.assertEqual(altered_raw_data[2]["id"], self.raw_data[2]["id"])
        self.assertEqual(altered_raw_data[0]["value"], self.raw_data[0]["value"])
        self.assertEqual(altered_raw_data[2]["value"], self.raw_data[2]["value"])

        self.assertEqual(altered_raw_data[1], self.raw_data[1])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

    def test_remove(self):
        """Remove all `char1` blocks

        Check whether all `char1` blocks have been removed and whether other blocks are intact.
        """
        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=RemoveStreamChildrenOperation(name="char1"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 2)
        self.assertEqual(altered_raw_data[0], self.raw_data[1])
        self.assertEqual(altered_raw_data[1], self.raw_data[3])

    def test_combine_to_listblock(self):
        """Combine all `char1` blocks into a new ListBlock named `list1`

        Check whether no `char1` blocks are present among the stream children and whether other
        blocks are intact.
        Check whether a new `list1` block has been added to the stream children and whether it has
        child blocks corresponding to the previous `char1` blocks.
        Check whether the ids and values from the `char1` blocks are intact in the list children.
        """
        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToListBlockOperation(
                block_name="char1", list_block_name="list1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 3)
        self.assertEqual(altered_raw_data[0], self.raw_data[1])
        self.assertEqual(altered_raw_data[1], self.raw_data[3])

        self.assertEqual(altered_raw_data[2]["type"], "list1")
        self.assertEqual(len(altered_raw_data[2]["value"]), 2)
        self.assertEqual(altered_raw_data[2]["value"][0]["type"], "item")
        self.assertEqual(altered_raw_data[2]["value"][1]["type"], "item")

        self.assertEqual(altered_raw_data[2]["value"][0]["id"], self.raw_data[0]["id"])
        self.assertEqual(altered_raw_data[2]["value"][1]["id"], self.raw_data[2]["id"])
        self.assertEqual(
            altered_raw_data[2]["value"][0]["value"], self.raw_data[0]["value"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"][1]["value"], self.raw_data[2]["value"]
        )

    def test_combine_to_listblock_no_existing_children(self):
        """Combine all `simplestruct` blocks into a new ListBlock named `list1`
        We have no `simplestruct` blocks in our existing data, so there should be no list1 blocks
        created and the data should be intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToListBlockOperation(
                block_name="simplestruct", list_block_name="list1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 4)
        self.assertEqual(altered_raw_data[0], self.raw_data[0])
        self.assertEqual(altered_raw_data[1], self.raw_data[1])
        self.assertEqual(altered_raw_data[2], self.raw_data[2])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

    def test_combine_single_type_to_streamblock(self):
        """Combine all `char1` blocks as children of a new StreamBlock named `stream1`

        Check whether no `char1` blocks are present among the (top) stream children and whether
        other blocks are intact.
        Check whether a new `stream1` block has been added to the (top) stream children.
        Check whether the new `stream1` block has the `char1` blocks as children and whether they
        are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToStreamBlockOperation(
                block_names=["char1"], stream_block_name="stream1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 3)
        self.assertEqual(altered_raw_data[0], self.raw_data[1])
        self.assertEqual(altered_raw_data[1], self.raw_data[3])

        self.assertEqual(altered_raw_data[2]["type"], "stream1")

        self.assertEqual(len(altered_raw_data[2]["value"]), 2)
        self.assertEqual(altered_raw_data[2]["value"][0], self.raw_data[0])
        self.assertEqual(altered_raw_data[2]["value"][1], self.raw_data[2])

    def test_combine_multiple_types_to_streamblock(self):
        """Combine all `char1` and `char2` blocks as children of a new StreamBlock named `stream1`

        Check whether no `char1` or `char2` blocks are present among the (top) stream children.
        Check whether a new `stream1` block has been added to the (top) stream children.
        Check whether the new `stream1` block has the `char1` and `char2` blocks as children and
        that they are intact.

        Note:
            We only have `char1` and `char2` blocks in our existing data.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToStreamBlockOperation(
                block_names=["char1", "char2"], stream_block_name="stream1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 1)
        self.assertEqual(altered_raw_data[0]["type"], "stream1")

        self.assertEqual(len(altered_raw_data[0]["value"]), 4)
        self.assertEqual(altered_raw_data[0]["value"][0], self.raw_data[0])
        self.assertEqual(altered_raw_data[0]["value"][1], self.raw_data[1])
        self.assertEqual(altered_raw_data[0]["value"][2], self.raw_data[2])
        self.assertEqual(altered_raw_data[0]["value"][3], self.raw_data[3])

    def test_combine_to_streamblock_no_existing_children(self):
        """Combine all `simplestruct` blocks as children of a new StreamBlock named `stream1`
        We have no `simplestruct` blocks in our existing data, so there should be no stream1 blocks
        created and the data should be intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToStreamBlockOperation(
                block_names=["simplestruct"], stream_block_name="stream1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data), 4)
        self.assertEqual(altered_raw_data[0], self.raw_data[0])
        self.assertEqual(altered_raw_data[1], self.raw_data[1])
        self.assertEqual(altered_raw_data[2], self.raw_data[2])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

    def test_to_structblock(self):
        """Move each `char1` block inside a new StructBlock named `struct1`

        Check whether each `char1` block has been replaced with a `struct1` block in the stream
        children.
        Check whether other blocks are intact.
        Check whether each `struct1` block has a `char1` child and whether it has the value of the
        previous `char1` block.

        Note:
            Block ids are not preserved here.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="",
            operation=StreamChildrenToStructBlockOperation("char1", "struct1"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(altered_raw_data[0]["type"], "struct1")
        self.assertEqual(altered_raw_data[2]["type"], "struct1")

        self.assertEqual(altered_raw_data[1], self.raw_data[1])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

        self.assertIn("char1", altered_raw_data[0]["value"])
        self.assertIn("char1", altered_raw_data[2]["value"])
        self.assertEqual(
            altered_raw_data[0]["value"]["char1"], self.raw_data[0]["value"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"]["char1"], self.raw_data[2]["value"]
        )

    def test_alter_value(self):
        """Change the value of each `char1` block to `foo`

        Check whether the value of each `char1` block has changed to `foo`.
        Check whether the values of other blocks are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="char1",
            operation=AlterBlockValueOperation(new_value="foo"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(altered_raw_data[0]["value"], "foo")
        self.assertEqual(altered_raw_data[1]["value"], self.raw_data[1]["value"])
        self.assertEqual(altered_raw_data[2]["value"], "foo")
        self.assertEqual(altered_raw_data[3]["value"], self.raw_data[3]["value"])


class FieldStructChildBlockTest(TestCase):
    """Tests involving changes to direct children of a StructBlock

    We use `simplestruct` blocks as the StructBlocks here.
    """

    def setUp(self):
        raw_data = factories.SampleModelFactory(
            content__0__char1__value="Char Block 1",
            content__1="simplestruct",
            content__2="simplestruct",
            content__3__char2__value="Char Block 2",
        ).content.raw_data
        self.raw_data = raw_data

    def test_blocks_and_data_not_operated_on_intact(self):
        """Test whether other blocks and data not passed to an operation are intact.

        We are checking whether the parts of the data which are not passed to an operation are
        intact. Since the recursion process depends just on the block path and block structure,
        this check is independent of the operation used. We will use a rename operation for now.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestruct",
            operation=RenameStructChildrenOperation(
                old_name="char1", new_name="renamed1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(altered_raw_data[0], self.raw_data[0])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

        self.assertEqual(altered_raw_data[1]["id"], self.raw_data[1]["id"])
        self.assertEqual(altered_raw_data[2]["id"], self.raw_data[2]["id"])
        self.assertEqual(altered_raw_data[1]["type"], self.raw_data[1]["type"])
        self.assertEqual(altered_raw_data[2]["type"], self.raw_data[2]["type"])

    def test_rename(self):
        """Rename `simplestruct.char1` blocks to `renamed1`

        Check whether all `simplestruct.char1` blocks have been renamed correctly.
        Check whether values for renamed blocks are intact.
        Check whether other children of `simplestruct` are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestruct",
            operation=RenameStructChildrenOperation(
                old_name="char1", new_name="renamed1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data[1]["value"]), 2)
        self.assertEqual(len(altered_raw_data[2]["value"]), 2)
        self.assertNotIn("char1", altered_raw_data[1]["value"])
        self.assertNotIn("char1", altered_raw_data[2]["value"])
        self.assertIn("renamed1", altered_raw_data[1]["value"])
        self.assertIn("renamed1", altered_raw_data[2]["value"])

        self.assertEqual(
            altered_raw_data[1]["value"]["renamed1"], self.raw_data[1]["value"]["char1"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"]["renamed1"], self.raw_data[2]["value"]["char1"]
        )

        self.assertIn("char2", altered_raw_data[1]["value"])
        self.assertIn("char2", altered_raw_data[2]["value"])
        self.assertEqual(
            altered_raw_data[1]["value"]["char2"], self.raw_data[1]["value"]["char2"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"]["char2"], self.raw_data[2]["value"]["char2"]
        )

    def test_remove(self):
        """Remove `simplestruct.char1` blocks

        Check whether all `simplestruct.char1` blocks have been removed.
        Check whether other children of `simplestruct` are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestruct",
            operation=RemoveStructChildrenOperation(name="char1"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data[1]["value"]), 1)
        self.assertEqual(len(altered_raw_data[2]["value"]), 1)
        self.assertNotIn("char1", altered_raw_data[1]["value"])
        self.assertNotIn("char1", altered_raw_data[2]["value"])

        self.assertIn("char2", altered_raw_data[1]["value"])
        self.assertIn("char2", altered_raw_data[2]["value"])
        self.assertEqual(
            altered_raw_data[1]["value"]["char2"], self.raw_data[1]["value"]["char2"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"]["char2"], self.raw_data[2]["value"]["char2"]
        )


class FieldStreamChildBlockTest(TestCase):
    """Tests involving changes to direct children of a StreamBlock

    We use `simplestream` blocks as the StreamBlocks here.
    """

    def setUp(self):
        raw_data = factories.SampleModelFactory(
            content__0__char1__value="Char Block 1",
            content__1="simplestream",
            content__1__simplestream__0__char1__value="Char Block 1",
            content__1__simplestream__1__char2__value="Char Block 2",
            content__1__simplestream__2__char1__value="Char Block 1",
            content__2="simplestream",
            content__2__simplestream__0__char1__value="Char Block 1",
            content__3__char2__value="Char Block 2",
        ).content.raw_data
        self.raw_data = raw_data

    def test_blocks_and_data_not_operated_on_intact(self):
        """Test whether other blocks and data not passed to an operation are intact.

        We are checking whether the parts of the data which are not passed to an operation are
        intact. Since the recursion process depends just on the block path and block structure,
        this check is independent of the operation used. We will use a rename operation for now.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestream",
            operation=RenameStreamChildrenOperation(
                old_name="char1", new_name="renamed1"
            ),
            streamfield=models.SampleModel.content,
        )
        self.assertEqual(altered_raw_data[0], self.raw_data[0])
        self.assertEqual(altered_raw_data[3], self.raw_data[3])

        self.assertEqual(altered_raw_data[1]["id"], self.raw_data[1]["id"])
        self.assertEqual(altered_raw_data[2]["id"], self.raw_data[2]["id"])
        self.assertEqual(altered_raw_data[1]["type"], self.raw_data[1]["type"])
        self.assertEqual(altered_raw_data[2]["type"], self.raw_data[2]["type"])

    def test_rename(self):
        """Rename `simplestream.char1` blocks to `renamed1`

        Check whether all `simplestream.char1` blocks have been renamed correctly.
        Check whether values and ids for renamed blocks are intact.
        Check whether other children of `simplestream` are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestream",
            operation=RenameStreamChildrenOperation(
                old_name="char1", new_name="renamed1"
            ),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(altered_raw_data[1]["value"][0]["type"], "renamed1")
        self.assertEqual(altered_raw_data[1]["value"][2]["type"], "renamed1")
        self.assertEqual(altered_raw_data[2]["value"][0]["type"], "renamed1")

        self.assertEqual(
            altered_raw_data[1]["value"][0]["id"], self.raw_data[1]["value"][0]["id"]
        )
        self.assertEqual(
            altered_raw_data[1]["value"][2]["id"], self.raw_data[1]["value"][2]["id"]
        )
        self.assertEqual(
            altered_raw_data[2]["value"][0]["id"], self.raw_data[2]["value"][0]["id"]
        )
        self.assertEqual(
            altered_raw_data[1]["value"][0]["value"],
            self.raw_data[1]["value"][0]["value"],
        )
        self.assertEqual(
            altered_raw_data[1]["value"][2]["value"],
            self.raw_data[1]["value"][2]["value"],
        )
        self.assertEqual(
            altered_raw_data[2]["value"][0]["value"],
            self.raw_data[2]["value"][0]["value"],
        )

        self.assertEqual(altered_raw_data[1]["value"][1], self.raw_data[1]["value"][1])

    def test_remove(self):
        """Remove `simplestream.char1` blocks

        Check whether all `simplestream.char1` blocks have been removed.
        Check whether other children of `simplestream` are intact.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplestream",
            operation=RemoveStreamChildrenOperation(name="char1"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(len(altered_raw_data[1]["value"]), 1)
        self.assertEqual(len(altered_raw_data[2]["value"]), 0)

        self.assertEqual(altered_raw_data[1]["value"][0], self.raw_data[1]["value"][1])


class FieldListChildBlockTest(TestCase):
    """Tests involving changes to direct children of a ListBlock

    We use `simplelist` blocks as the ListBlocks here.
    """

    def setUp(self):
        raw_data = factories.SampleModelFactory(
            content__0__char1__value="Char Block 1",
            content__1__simplelist__0="Foo 1",
            content__1__simplelist__1="Foo 2",
            content__2__simplelist__0="Foo 3",
        ).content.raw_data
        self.raw_data = raw_data

    def test_to_structblock(self):
        """Turn each list child into a StructBlock and move value inside as a child named `text`

        Check whether each list child has been converted to a StructBlock with a child named `text`
        in it.
        Check whether the previous value of each list child is now the value that `text` takes.

        Note:
            Block ids are not preserved here.
        """

        altered_raw_data = apply_changes_to_raw_data(
            raw_data=self.raw_data,
            block_path_str="simplelist",
            operation=ListChildrenToStructBlockOperation(block_name="text"),
            streamfield=models.SampleModel.content,
        )

        self.assertEqual(type(altered_raw_data[1]["value"][0]["value"]), dict)
        self.assertEqual(type(altered_raw_data[1]["value"][1]["value"]), dict)
        self.assertEqual(type(altered_raw_data[2]["value"][0]["value"]), dict)

        self.assertIn("text", altered_raw_data[1]["value"][0]["value"])
        self.assertIn("text", altered_raw_data[1]["value"][1]["value"])
        self.assertIn("text", altered_raw_data[2]["value"][0]["value"])

        self.assertEqual(
            altered_raw_data[1]["value"][0]["value"]["text"],
            self.raw_data[1]["value"][0]["value"],
        )
        self.assertEqual(
            altered_raw_data[1]["value"][1]["value"]["text"],
            self.raw_data[1]["value"][1]["value"],
        )
        self.assertEqual(
            altered_raw_data[2]["value"][0]["value"]["text"],
            self.raw_data[2]["value"][0]["value"],
        )