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)
def __init__(
self,
client_id: str,
database_url: str,
logger: Logger = logging.getLogger(__name__),
):
self.client_id = client_id
self.database_url = database_url
self._logger = logger
self.metadata = MetaData()
self.installations = SQLAlchemyInstallationStore.build_installations_table(
metadata=self.metadata,
table_name=SQLAlchemyInstallationStore.
default_installations_table_name,
)
self.bots = SQLAlchemyInstallationStore.build_bots_table(
metadata=self.metadata,
table_name=SQLAlchemyInstallationStore.default_bots_table_name,
)
from strings import projects_starter, rutorrent, ideas_block, sample_platform
logging.basicConfig(level=logging.INFO)
logging.getLogger("sqlalchemy.engine").setLevel(logging.INFO)
logger = logging.getLogger(__name__)
database_url = os.environ["DATABASE_URL"]
client_id, client_secret, signing_secret = (
os.environ["SLACK_CLIENT_ID"],
os.environ["SLACK_CLIENT_SECRET"],
os.environ["SLACK_SIGNING_SECRET"],
)
engine: Engine = sqlalchemy.create_engine(database_url)
installation_store = SQLAlchemyInstallationStore(
client_id=client_id,
engine=engine,
logger=logger,
)
oauth_state_store = SQLAlchemyOAuthStateStore(
expiration_seconds=120,
engine=engine,
logger=logger,
)
try:
engine.execute("select count(*) from slack_bots")
except Exception as e:
installation_store.metadata.create_all(engine)
oauth_state_store.metadata.create_all(engine)
# update ngrok/production links in slash commands and enable events page
# Initializes your app with your bot token and signing secret
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)
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)