浏览代码

Made some fixes to the forgotten test_driver...

Kirk Trombley 6 年之前
父节点
当前提交
5431be33d7
共有 1 个文件被更改,包括 15 次插入4 次删除
  1. 15 4
      src/test_driver.py

+ 15 - 4
src/test_driver.py

@@ -4,18 +4,29 @@ import os
 
 from rollbot import Rollbot
 from command_system import RollbotMessage
-from config import DB_FILE
+from config import BOTS_LOOKUP, DB_FILE, get_config
 
 try:
     os.remove(DB_FILE)
 except FileNotFoundError:
     pass
-rollbot = Rollbot(logging.getLogger(__name__))
+
+import db
+
+rollbot = Rollbot(
+    plugins=get_config("plugins"),
+    aliases=get_config("aliases"),
+    responses=get_config("responses"),
+    lookup=BOTS_LOOKUP,
+    sleep_time=float(get_config("sleep_time")),
+    callback=None,
+    session_factory=db.session_scope
+)
+db.init_db()
 
 
 def test_drive(msg, from_admin=True):
-    rmsg = RollbotMessage("mock", None, None, "test_group", None, msg)
-    rmsg.from_admin = from_admin
+    rmsg = RollbotMessage("mock", None, None, "test_group", None, msg, from_admin)
     r =rollbot.run_command(rmsg)
     print(msg, ":", r.txt, ",", r.failure)