#! /usr/bin/env vpython3
# Copyright 2020 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Tests for ini.py."""


import os
import sys
import textwrap
import unittest

sys.path.append(
    os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', '..')))
from pylib.local.emulator import ini


class IniTest(unittest.TestCase):
  def testLoadsBasic(self):
    ini_str = textwrap.dedent("""\
        foo.bar = 1
        foo.baz= example
        bar.bad =/path/to/thing

        [section_1]
        foo.bar = 1
        foo.baz= example

        [section_2]
        foo.baz= example
        bar.bad =/path/to/thing

        [section_1]
        bar.bad =/path/to/thing
        """)
    expected = {
        'foo.bar': '1',
        'foo.baz': 'example',
        'bar.bad': '/path/to/thing',
        'section_1': {
            'foo.bar': '1',
            'foo.baz': 'example',
            'bar.bad': '/path/to/thing',
        },
        'section_2': {
            'foo.baz': 'example',
            'bar.bad': '/path/to/thing',
        },
    }
    self.assertEqual(expected, ini.loads(ini_str))

  def testLoadsDuplicatedKeysStrictFailure(self):
    ini_str = textwrap.dedent("""\
        foo.bar = 1
        foo.baz = example
        bar.bad = /path/to/thing
        foo.bar = duplicate
        """)
    with self.assertRaises(ValueError):
      ini.loads(ini_str, strict=True)

  def testLoadsDuplicatedKeysInSectionStrictFailure(self):
    ini_str = textwrap.dedent("""\
        [section_1]
        foo.bar = 1
        foo.baz = example
        bar.bad = /path/to/thing
        foo.bar = duplicate
        """)
    with self.assertRaises(ValueError):
      ini.loads(ini_str, strict=True)

  def testLoadsPermissive(self):
    ini_str = textwrap.dedent("""\
        foo.bar = 1
        foo.baz = example
        bar.bad = /path/to/thing
        foo.bar = duplicate

        [section_1]
        foo.bar = 1
        foo.baz = example
        bar.bad = /path/to/thing
        foo.bar = duplicate
        """)
    expected = {
        'foo.bar': 'duplicate',
        'foo.baz': 'example',
        'bar.bad': '/path/to/thing',
        'section_1': {
            'foo.bar': 'duplicate',
            'foo.baz': 'example',
            'bar.bad': '/path/to/thing',
        },
    }
    self.assertEqual(expected, ini.loads(ini_str, strict=False))

  def testDumpsBasic(self):
    ini_contents = {
        'foo.bar': '1',
        'foo.baz': 'example',
        'bar.bad': '/path/to/thing',
        'section_2': {
            'foo.baz': 'example',
            'bar.bad': '/path/to/thing',
        },
        'section_1': {
            'foo.bar': '1',
            'foo.baz': 'example',
        },
    }
    # ini.dumps is expected to dump to string alphabetically
    # by key and section name.
    expected = textwrap.dedent("""\
        bar.bad = /path/to/thing
        foo.bar = 1
        foo.baz = example

        [section_1]
        foo.bar = 1
        foo.baz = example

        [section_2]
        bar.bad = /path/to/thing
        foo.baz = example
        """)
    self.assertEqual(expected, ini.dumps(ini_contents))

  def testDumpsSections(self):
    ini_contents = {
        'section_2': {
            'foo.baz': 'example',
            'bar.bad': '/path/to/thing',
        },
        'section_1': {
            'foo.bar': '1',
            'foo.baz': 'example',
        },
    }
    # ini.dumps is expected to dump to string alphabetically
    # by key first, and then by section and the associated keys
    expected = textwrap.dedent("""\
        [section_1]
        foo.bar = 1
        foo.baz = example

        [section_2]
        bar.bad = /path/to/thing
        foo.baz = example
        """)
    self.assertEqual(expected, ini.dumps(ini_contents))


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