# Copyright 2023 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Tests for actions."""

import os
import unittest

import actions
import cf_parse

TEST_DATA_DIR = 'test_data/'

class TestActions(unittest.TestCase):
    """Tests for actions."""
    def test_remove_contacts(self):
        delete_me = 'removable@google.com'
        action = actions.remove_contacts([delete_me])
        test_file = os.path.join(TEST_DATA_DIR, 'control.actions')
        cf = cf_parse.ControlFile(test_file)

        self.assertTrue('contacts' in cf.metadata)
        self.assertTrue(delete_me in cf.metadata['contacts'])
        self.assertTrue(action(cf))
        self.assertTrue('contacts' in cf.metadata)
        self.assertFalse(delete_me in cf.metadata['contacts'])

        self.assertFalse(action(cf))

    def test_append_contacts(self):
        append_1 = "appendable1@google.com"
        append_2 = "appendable2@google.com"
        action = actions.append_contacts([append_1, append_2])
        test_file = os.path.join(TEST_DATA_DIR, 'control.actions')
        cf = cf_parse.ControlFile(test_file)
        starting_len = len(cf.metadata['contacts'])

        self.assertTrue(action(cf))
        self.assertTrue('contacts' in cf.metadata)
        self.assertTrue(append_1 in cf.metadata['contacts'])
        self.assertTrue(append_2 in cf.metadata['contacts'])
        self.assertEqual(cf.metadata['contacts'].index(append_1), starting_len)
        self.assertEqual(cf.metadata['contacts'].index(append_2),
                         starting_len + 1)
        self.assertEqual(len(cf.metadata['contacts']), starting_len + 2)

        self.assertTrue(action(cf))
        self.assertEqual(len(cf.metadata['contacts']), starting_len + 2)

    def test_prepend_contacts(self):
        prepend = "prependable@google.com"
        action = actions.prepend_contacts([prepend])
        test_file = os.path.join(TEST_DATA_DIR, 'control.actions')
        cf = cf_parse.ControlFile(test_file)
        starting_len = len(cf.metadata['contacts'])

        self.assertTrue(action(cf))
        self.assertTrue('contacts' in cf.metadata)
        self.assertTrue(prepend in cf.metadata['contacts'])
        self.assertEqual(cf.metadata['contacts'].index(prepend), 0)
        self.assertEqual(len(cf.metadata['contacts']), starting_len + 1)

        self.assertTrue(action(cf))
        self.assertEqual(len(cf.metadata['contacts']), starting_len + 1)


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