Search sample data in more folders for perftest

Tencent · Apr 16, 2015 · 556d154 · 556d154
1 parent 22021d6
commit 556d154
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/test/perftest/perftest.h b/test/perftest/perftest.h
@@ -71,9 +71,20 @@ class PerfTest : public ::testing::Test {
     PerfTest() : filename_(), json_(), length_(), whitespace_(), whitespace_length_() {}
 
     virtual void SetUp() {
-        FILE *fp = fopen(filename_ = "data/sample.json", "rb");
-        if (!fp) 
-            fp = fopen(filename_ = "../../bin/data/sample.json", "rb");
+
+        const char *paths[] = {
+            "data/sample.json",
+            "bin/data/sample.json",
+            "../bin/data/sample.json",
+            "../../bin/data/sample.json",
+            "../../../bin/data/sample.json"
+        };
+        FILE *fp = 0;
+        for (size_t i = 0; i < sizeof(paths) / sizeof(paths[0]); i++) {
+            fp = fopen(paths[i], "rb");
+            if (fp)
+                break;
+        }
         ASSERT_TRUE(fp != 0);
 
         fseek(fp, 0, SEEK_END);