#!/usr/bin/env vpython3

# Copyright 2024 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""File for testing count.py."""

import unittest
from count import Count


class CountTest(unittest.TestCase):
    """Test count.py."""

    def test_no_record(self) -> None:
        count = Count("a")
        self.assertEqual(count.dump().name, "a")
        self.assertEqual(count.dump().value, 0)

    def test_one_record(self) -> None:
        count = Count("b")
        count.record()
        self.assertEqual(count.dump().name, "b")
        self.assertEqual(count.dump().value, 1)

    def test_more_records(self) -> None:
        count = Count("c")
        count.record()
        count.record()
        self.assertEqual(count.dump().name, "c")
        self.assertEqual(count.dump().value, 2)


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