import unittest

from coffer.units import (
    ACCOUNT_COFFER, ACCOUNT_ORBIT, ACCOUNT_SIGIL, agent_account, company_account,
)


class CompanyAccountsTest(unittest.TestCase):
    def test_orbit_company_book_is_separate_from_founder(self):
        self.assertEqual(ACCOUNT_ORBIT, "company:orbit")
        self.assertEqual(company_account("orbit"), ACCOUNT_ORBIT)
        self.assertNotEqual(ACCOUNT_ORBIT, agent_account("orbit"))

    def test_company_ids_unique(self):
        self.assertEqual(len({ACCOUNT_COFFER, ACCOUNT_SIGIL, ACCOUNT_ORBIT}), 3)


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