diff --git a/coffer/units.py b/coffer/units.py index c27fcdc..00c1f44 100644 --- a/coffer/units.py +++ b/coffer/units.py @@ -52,6 +52,7 @@ ACCOUNT_COFFER = "company:coffer" ACCOUNT_NEW_INTERNET = "company:new-internet-co" ACCOUNT_LEORA = "agent:leora" ACCOUNT_SIGIL = "company:sigil" +ACCOUNT_ORBIT = "company:orbit" # Orbit: open-source CRM for agents; founder agent:orbit def agent_account(agent_id: str) -> str: diff --git a/tests/test_companies.py b/tests/test_companies.py new file mode 100644 index 0000000..c5d23a7 --- /dev/null +++ b/tests/test_companies.py @@ -0,0 +1,19 @@ +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()