From 1becbd9fcceb5b9576a78833a82f984afe6c3f4b Mon Sep 17 00:00:00 2001
From: Ilya Lysenkov <lysenkov.ilya@gmail.com>
Date: Thu, 31 Jan 2013 23:17:07 +0400
Subject: [PATCH] Added a test for invalid key sizes in LSH tables

---
 modules/flann/test/test_lshtable_badarg.cpp | 91 +++++++++++++++++++++
 modules/flann/test/test_main.cpp            |  3 +
 modules/flann/test/test_precomp.cpp         |  1 +
 modules/flann/test/test_precomp.hpp         | 17 ++++
 4 files changed, 112 insertions(+)
 create mode 100644 modules/flann/test/test_lshtable_badarg.cpp
 create mode 100644 modules/flann/test/test_main.cpp
 create mode 100644 modules/flann/test/test_precomp.cpp
 create mode 100644 modules/flann/test/test_precomp.hpp

diff --git a/modules/flann/test/test_lshtable_badarg.cpp b/modules/flann/test/test_lshtable_badarg.cpp
new file mode 100644
index 0000000000..9e42335c2c
--- /dev/null
+++ b/modules/flann/test/test_lshtable_badarg.cpp
@@ -0,0 +1,91 @@
+/*M///////////////////////////////////////////////////////////////////////////////////////
+//
+//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
+//
+//  By downloading, copying, installing or using the software you agree to this license.
+//  If you do not agree to this license, do not download, install,
+//  copy or use the software.
+//
+//
+//                        Intel License Agreement
+//                For Open Source Computer Vision Library
+//
+// Copyright (C) 2000, Intel Corporation, all rights reserved.
+// Third party copyrights are property of their respective owners.
+//
+// Redistribution and use in source and binary forms, with or without modification,
+// are permitted provided that the following conditions are met:
+//
+//   * Redistribution's of source code must retain the above copyright notice,
+//     this list of conditions and the following disclaimer.
+//
+//   * Redistribution's in binary form must reproduce the above copyright notice,
+//     this list of conditions and the following disclaimer in the documentation
+//     and/or other materials provided with the distribution.
+//
+//   * The name of Intel Corporation may not be used to endorse or promote products
+//     derived from this software without specific prior written permission.
+//
+// This software is provided by the copyright holders and contributors "as is" and
+// any express or implied warranties, including, but not limited to, the implied
+// warranties of merchantability and fitness for a particular purpose are disclaimed.
+// In no event shall the Intel Corporation or contributors be liable for any direct,
+// indirect, incidental, special, exemplary, or consequential damages
+// (including, but not limited to, procurement of substitute goods or services;
+// loss of use, data, or profits; or business interruption) however caused
+// and on any theory of liability, whether in contract, strict liability,
+// or tort (including negligence or otherwise) arising in any way out of
+// the use of this software, even if advised of the possibility of such damage.
+//
+//M*/
+
+#include "test_precomp.hpp"
+
+using namespace cv;
+
+class CV_LshTableBadArgTest : public cvtest::BadArgTest
+{
+protected:
+    void run(int);
+    void run_func(void) {};
+
+    struct Caller
+    {
+        int table_number, key_size, multi_probe_level;
+        Mat features;
+
+        void operator()() const
+        {
+            flann::LshIndexParams indexParams(table_number, key_size, multi_probe_level);
+            flann::Index lsh(features, indexParams);
+        }
+    };
+};
+
+void CV_LshTableBadArgTest::run( int /* start_from */ )
+{
+    RNG &rng = ts->get_rng();
+
+    Caller caller;
+    Size featuresSize = cvtest::randomSize(rng, 10.0);
+    caller.features = cvtest::randomMat(rng, featuresSize, CV_8UC1, 0, 255, false);
+    caller.table_number = 12;
+    caller.multi_probe_level = 2;
+
+    int errors = 0;
+    caller.key_size = 0;
+    errors += run_test_case(CV_StsBadArg, "key_size is zero", caller);
+
+    caller.key_size = static_cast<int>(sizeof(size_t) * CHAR_BIT);
+    errors += run_test_case(CV_StsBadArg, "key_size is too big", caller);
+
+    caller.key_size += cvtest::randInt(rng) % 100;
+    errors += run_test_case(CV_StsBadArg, "key_size is too big", caller);
+
+    if (errors != 0)
+        ts->set_failed_test_info(cvtest::TS::FAIL_MISMATCH);
+    else
+        ts->set_failed_test_info(cvtest::TS::OK);
+}
+
+TEST(Flann_LshTable, badarg) { CV_LshTableBadArgTest test; test.safe_run(); }
diff --git a/modules/flann/test/test_main.cpp b/modules/flann/test/test_main.cpp
new file mode 100644
index 0000000000..6b24993447
--- /dev/null
+++ b/modules/flann/test/test_main.cpp
@@ -0,0 +1,3 @@
+#include "test_precomp.hpp"
+
+CV_TEST_MAIN("cv")
diff --git a/modules/flann/test/test_precomp.cpp b/modules/flann/test/test_precomp.cpp
new file mode 100644
index 0000000000..5956e13e3e
--- /dev/null
+++ b/modules/flann/test/test_precomp.cpp
@@ -0,0 +1 @@
+#include "test_precomp.hpp"
diff --git a/modules/flann/test/test_precomp.hpp b/modules/flann/test/test_precomp.hpp
new file mode 100644
index 0000000000..809a97e702
--- /dev/null
+++ b/modules/flann/test/test_precomp.hpp
@@ -0,0 +1,17 @@
+#ifdef __GNUC__
+#  pragma GCC diagnostic ignored "-Wmissing-declarations"
+#  if defined __clang__ || defined __APPLE__
+#    pragma GCC diagnostic ignored "-Wmissing-prototypes"
+#    pragma GCC diagnostic ignored "-Wextra"
+#  endif
+#endif
+
+#ifndef __OPENCV_TEST_PRECOMP_HPP__
+#define __OPENCV_TEST_PRECOMP_HPP__
+
+#include "opencv2/ts/ts.hpp"
+#include "opencv2/core/core.hpp"
+#include "opencv2/flann/flann.hpp"
+#include <iostream>
+
+#endif