From 18404f70e7b69a549e9e5d806875967f0ebe70b1 Mon Sep 17 00:00:00 2001
From: Ryan Barrett <git@ryanb.org>
Date: Sun, 5 Feb 2023 22:22:22 -0800
Subject: [PATCH] speed up tests by only using 1024 bit keys

fixes #398
---
 models.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/models.py b/models.py
index 898a876..ba351ce 100644
--- a/models.py
+++ b/models.py
@@ -13,6 +13,7 @@ from Crypto.Util import number
 from flask import request
 from google.cloud import ndb
 from granary import as2, microformats2
+from oauth_dropins.webutil.appengine_info import DEBUG
 from oauth_dropins.webutil.models import StringIdModel
 from oauth_dropins.webutil import util
 from oauth_dropins.webutil.util import json_dumps, json_loads
@@ -24,7 +25,8 @@ WWW_DOMAINS = frozenset((
     'www.jvt.me',
 ))
 PROTOCOLS = ('activitypub', 'bluesky', 'ostatus', 'webmention', 'ui')
-KEY_BITS = 2048
+# 2048 bits makes tests slow, so use 1024 for them
+KEY_BITS = 1024 if DEBUG else 2048
 
 logger = logging.getLogger(__name__)