class TestSQLite3(unittest.TestCase):
engine: Engine
def setUp(self):
self.engine = sqlalchemy.create_engine("sqlite:///:memory:")
self.store = SQLAlchemyInstallationStore(client_id="111.222",
engine=self.engine)
self.store.metadata.create_all(self.engine)
def tearDown(self):
self.store.metadata.drop_all(self.engine)
self.engine.dispose()
def test_save_and_find(self):
installation = Installation(app_id="A111",
enterprise_id="E111",
team_id="T111",
user_id="U111",
bot_id="B111",
bot_token="xoxb-111",
bot_scopes=["chat:write"],
bot_user_id="U222")
self.store.save(installation)
bot = self.store.find_bot(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(bot)
bot = self.store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
bot = self.store.find_bot(enterprise_id=None, team_id="T111")
self.assertIsNone(bot)
class TestSQLite3(unittest.TestCase):
engine: Engine
def setUp(self):
self.engine = sqlalchemy.create_engine("sqlite:///:memory:")
self.store = SQLAlchemyInstallationStore(client_id="111.222",
engine=self.engine)
self.store.metadata.create_all(self.engine)
def tearDown(self):
self.store.metadata.drop_all(self.engine)
self.engine.dispose()
def test_save_and_find(self):
installation = Installation(
app_id="A111",
enterprise_id="E111",
team_id="T111",
user_id="U111",
bot_id="B111",
bot_token="xoxb-111",
bot_scopes=["chat:write"],
bot_user_id="U222",
)
self.store.save(installation)
store = self.store
# find bots
bot = store.find_bot(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(bot)
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
bot = store.find_bot(enterprise_id=None, team_id="T111")
self.assertIsNone(bot)
# delete bots
store.delete_bot(enterprise_id="E111", team_id="T222")
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
# find installations
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="E111", team_id="T222")
self.assertIsNone(i)
i = store.find_installation(enterprise_id=None, team_id="T111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U222")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T222",
user_id="U111")
self.assertIsNone(i)
# delete installations
store.delete_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNone(i)
# delete all
store.save(installation)
store.delete_all(enterprise_id="E111", team_id="T111")
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNone(i)
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
def test_org_installation(self):
installation = Installation(
app_id="AO111",
enterprise_id="EO111",
user_id="UO111",
bot_id="BO111",
bot_token="xoxb-O111",
bot_scopes=["chat:write"],
bot_user_id="UO222",
is_enterprise_install=True,
)
self.store.save(installation)
store = self.store
# find bots
bot = store.find_bot(enterprise_id="EO111", team_id=None)
self.assertIsNotNone(bot)
bot = store.find_bot(enterprise_id="EO111",
team_id="TO222",
is_enterprise_install=True)
self.assertIsNotNone(bot)
bot = store.find_bot(enterprise_id="EO111", team_id="TO222")
self.assertIsNone(bot)
bot = store.find_bot(enterprise_id=None, team_id="TO111")
self.assertIsNone(bot)
# delete bots
store.delete_bot(enterprise_id="EO111", team_id="TO222")
bot = store.find_bot(enterprise_id="EO111", team_id=None)
self.assertIsNotNone(bot)
store.delete_bot(enterprise_id="EO111", team_id=None)
bot = store.find_bot(enterprise_id="EO111", team_id=None)
self.assertIsNone(bot)
# find installations
i = store.find_installation(enterprise_id="EO111", team_id=None)
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="EO111",
team_id="T111",
is_enterprise_install=True)
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="EO111", team_id="T222")
self.assertIsNone(i)
i = store.find_installation(enterprise_id=None, team_id="T111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="EO111",
team_id=None,
user_id="UO111")
self.assertIsNotNone(i)
i = store.find_installation(
enterprise_id="E111",
team_id="T111",
is_enterprise_install=True,
user_id="U222",
)
self.assertIsNone(i)
i = store.find_installation(enterprise_id=None,
team_id="T222",
user_id="U111")
self.assertIsNone(i)
# delete installations
store.delete_installation(enterprise_id="E111", team_id=None)
i = store.find_installation(enterprise_id="E111", team_id=None)
self.assertIsNone(i)
# delete all
store.save(installation)
store.delete_all(enterprise_id="E111", team_id=None)
i = store.find_installation(enterprise_id="E111", team_id=None)
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id=None,
user_id="U111")
self.assertIsNone(i)
bot = store.find_bot(enterprise_id=None, team_id="T222")
self.assertIsNone(bot)
def test_save_and_find_token_rotation(self):
store = self.store
installation = Installation(
app_id="A111",
enterprise_id="E111",
team_id="T111",
user_id="U111",
bot_id="B111",
bot_token="xoxe.xoxp-1-initial",
bot_scopes=["chat:write"],
bot_user_id="U222",
bot_refresh_token="xoxe-1-initial",
bot_token_expires_in=43200,
)
store.save(installation)
bot = store.find_bot(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(bot)
self.assertEqual(bot.bot_refresh_token, "xoxe-1-initial")
# Update the existing data
refreshed_installation = Installation(
app_id="A111",
enterprise_id="E111",
team_id="T111",
user_id="U111",
bot_id="B111",
bot_token="xoxe.xoxp-1-refreshed",
bot_scopes=["chat:write"],
bot_user_id="U222",
bot_refresh_token="xoxe-1-refreshed",
bot_token_expires_in=43200,
)
store.save(refreshed_installation)
# find bots
bot = store.find_bot(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(bot)
self.assertEqual(bot.bot_refresh_token, "xoxe-1-refreshed")
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
bot = store.find_bot(enterprise_id=None, team_id="T111")
self.assertIsNone(bot)
# delete bots
store.delete_bot(enterprise_id="E111", team_id="T222")
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)
# find installations
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="E111", team_id="T222")
self.assertIsNone(i)
i = store.find_installation(enterprise_id=None, team_id="T111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNotNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U222")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T222",
user_id="U111")
self.assertIsNone(i)
# delete installations
store.delete_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNone(i)
# delete all
store.save(installation)
store.delete_all(enterprise_id="E111", team_id="T111")
i = store.find_installation(enterprise_id="E111", team_id="T111")
self.assertIsNone(i)
i = store.find_installation(enterprise_id="E111",
team_id="T111",
user_id="U111")
self.assertIsNone(i)
bot = store.find_bot(enterprise_id="E111", team_id="T222")
self.assertIsNone(bot)