浏览代码

Fix issues in tests

add headers Server and Date
check queries for empty string and not int
Jesterovskiy 9 年之前
父节点
当前提交
6129e51dbd
共有 1 个文件被更改,包括 6 次插入0 次删除
  1. 6 0
      frameworks/Crystal/kemal/server-postgres.cr

+ 6 - 0
frameworks/Crystal/kemal/server-postgres.cr

@@ -38,6 +38,7 @@ end
 
 private def sanitizedQueryCount(request)
   queries = request.params["queries"] as String
+  return 1 if queries.empty? || queries.to_i?.nil?
   if queries.to_i > 500
     queries = 500
   elsif queries.to_i < 1
@@ -46,6 +47,11 @@ private def sanitizedQueryCount(request)
   queries.to_i
 end
 
+before_all do |env|
+  env.response.headers["Server"] = "Kemal"
+  env.response.headers["Date"] = Time.now.to_s
+end
+
 #
 # Basic Tests
 #