relaxed assertion on not found random addresses

agladysh · Oct 20, 2012 · 8ed5cbc · 8ed5cbc
1 parent 24d2b60
commit 8ed5cbc
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/test.lua b/test/test.lua
@@ -222,7 +222,10 @@ for i = 1, #profiles do
       if i % 1e4 == 0 then
         print("#", i, "of", num_queries)
       end
-      assert(geodb:query_by_ipnum(cases[i], p.field))
+      local result, err = geodb:query_by_ipnum(cases[i], p.field)
+      if not result and err ~= "not found" then
+        error(err)
+      end
     end
 
     print(
@@ -253,7 +256,10 @@ for i = 1, #profiles do
       if i % 1e4 == 0 then
         print("#", i, "of", num_queries)
       end
-      assert(geodb:query_by_name(cases[i], p.field))
+      local result, err = geodb:query_by_name(cases[i], p.field)
+      if not result and err ~= "not found" then
+        error(err)
+      end
     end
 
     print(