#!/usr/bin/env python3
########################################################################
#
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
#
########################################################################

import unittest
import os
import os.path
import mmap

from orcus import ods, FormulaTokenType

import file_load_common as common


class TestCase(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        # base directory for ods test files.
        basedir = os.path.join(os.path.dirname(__file__), "..", "ods")
        cls.basedir = os.path.normpath(basedir)

    def test_import(self):
        test_dirs = ("raw-values-1", "formula-1", "formula-2")
        for test_dir in test_dirs:
            test_dir = os.path.join(self.basedir, test_dir)
            common.run_test_dir(self, test_dir, common.DocLoader(ods))

    def test_formula_tokens(self):
        filepath = os.path.join(self.basedir, "formula-1", "input.ods")
        with open(filepath, "rb") as f:
            doc = ods.read(f, recalc=False)

        self.assertEqual(len(doc.sheets), 1)

        # The 'Formula' sheet contains 4 formula cells in A1:A4.
        sheet = doc.sheets[0]
        self.assertEqual(sheet.name, "Formula")
        rows = [row for row in sheet.get_rows()]
        self.assertEqual(len(rows), 4)

        expected = ("1*2", "12/3", "AVERAGE($A1:A$2)", "SUM($A$1:$A$3)")
        for row, expected_formula in zip(sheet.get_rows(), expected):
            c = row[0]
            self.assertEqual(c.formula, expected_formula)

        expected = (
            (("1", FormulaTokenType.VALUE), ("*", FormulaTokenType.OPERATOR), ("2", FormulaTokenType.VALUE)),
            (("12", FormulaTokenType.VALUE), ("/", FormulaTokenType.OPERATOR), ("3", FormulaTokenType.VALUE)),
            (("AVERAGE", FormulaTokenType.FUNCTION), ("(", FormulaTokenType.OPERATOR), ("$A1:A$2", FormulaTokenType.REFERENCE), (")", FormulaTokenType.OPERATOR)),
            (("SUM", FormulaTokenType.FUNCTION), ("(", FormulaTokenType.OPERATOR), ("$A$1:$A$3", FormulaTokenType.REFERENCE), (")", FormulaTokenType.OPERATOR)),
        )

        for row, expected_formula_tokens in zip(sheet.get_rows(), expected):
            c = row[0]
            iter = c.get_formula_tokens()
            for token, expected_token in zip(iter, expected_formula_tokens):
                self.assertEqual(str(token), expected_token[0])
                self.assertEqual(token.type, expected_token[1])


if __name__ == '__main__':
    unittest.main()