###############################################################################
#
# Tests for XlsxWriter.
#
# SPDX-License-Identifier: BSD-2-Clause
#
# Copyright (c), 2013-2025, John McNamara, jmcnamara@cpan.org
#

import unittest
from io import StringIO

from xlsxwriter.worksheet import Worksheet

from ..helperfunctions import _xml_to_list


class TestAssembleWorksheet(unittest.TestCase):
    """
    Test assembling a complete Worksheet file.

    """

    def test_assemble_xml_file(self):
        """Test writing a worksheet with conditional formatting."""
        self.maxDiff = None

        fh = StringIO()
        worksheet = Worksheet()
        worksheet._set_filehandle(fh)
        worksheet.select()

        worksheet.write("A1", 10)
        worksheet.write("A2", 20)
        worksheet.write("A3", 30)
        worksheet.write("A4", 40)

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "above",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "below",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "equal or above",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "equal or below",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "1 std dev above",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "1 std dev below",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "2 std dev above",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "2 std dev below",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "3 std dev above",
            },
        )

        worksheet.conditional_format(
            "A1:A4",
            {
                "type": "average",
                "format": None,
                "criteria": "3 std dev below",
            },
        )

        worksheet._assemble_xml_file()

        exp = _xml_to_list(
            """
                <?xml version="1.0" encoding="UTF-8" standalone="yes"?>
                <worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships">
                  <dimension ref="A1:A4"/>
                  <sheetViews>
                    <sheetView tabSelected="1" workbookViewId="0"/>
                  </sheetViews>
                  <sheetFormatPr defaultRowHeight="15"/>
                  <sheetData>
                    <row r="1" spans="1:1">
                      <c r="A1">
                        <v>10</v>
                      </c>
                    </row>
                    <row r="2" spans="1:1">
                      <c r="A2">
                        <v>20</v>
                      </c>
                    </row>
                    <row r="3" spans="1:1">
                      <c r="A3">
                        <v>30</v>
                      </c>
                    </row>
                    <row r="4" spans="1:1">
                      <c r="A4">
                        <v>40</v>
                      </c>
                    </row>
                  </sheetData>
                  <conditionalFormatting sqref="A1:A4">
                    <cfRule type="aboveAverage" priority="1"/>
                    <cfRule type="aboveAverage" priority="2" aboveAverage="0"/>
                    <cfRule type="aboveAverage" priority="3" equalAverage="1"/>
                    <cfRule type="aboveAverage" priority="4" aboveAverage="0" equalAverage="1"/>
                    <cfRule type="aboveAverage" priority="5" stdDev="1"/>
                    <cfRule type="aboveAverage" priority="6" aboveAverage="0" stdDev="1"/>
                    <cfRule type="aboveAverage" priority="7" stdDev="2"/>
                    <cfRule type="aboveAverage" priority="8" aboveAverage="0" stdDev="2"/>
                    <cfRule type="aboveAverage" priority="9" stdDev="3"/>
                    <cfRule type="aboveAverage" priority="10" aboveAverage="0" stdDev="3"/>
                  </conditionalFormatting>
                  <pageMargins left="0.7" right="0.7" top="0.75" bottom="0.75" header="0.3" footer="0.3"/>
                </worksheet>
                """
        )

        got = _xml_to_list(fh.getvalue())

        self.assertEqual(exp, got)
