OLD | NEW |
1 // Copyright 2009 The RE2 Authors. All Rights Reserved. | 1 // Copyright 2009 The RE2 Authors. All Rights Reserved. |
2 // Use of this source code is governed by a BSD-style | 2 // Use of this source code is governed by a BSD-style |
3 // license that can be found in the LICENSE file. | 3 // license that can be found in the LICENSE file. |
4 | 4 |
5 #include <stdio.h> | 5 #include <stdio.h> |
6 #ifndef _WIN32 | 6 #ifndef WIN32 |
7 #include <sys/resource.h> | 7 #include <sys/resource.h> |
8 #endif | 8 #endif |
9 #include "util/test.h" | 9 #include "util/test.h" |
10 | 10 |
11 DEFINE_string(test_tmpdir, "/var/tmp", "temp directory"); | 11 DEFINE_string(test_tmpdir, "/var/tmp", "temp directory"); |
12 | 12 |
13 struct Test { | 13 struct Test { |
14 void (*fn)(void); | 14 void (*fn)(void); |
15 const char *name; | 15 const char *name; |
16 }; | 16 }; |
17 | 17 |
18 static Test tests[10000]; | 18 static Test tests[10000]; |
19 static int ntests; | 19 static int ntests; |
20 | 20 |
21 void RegisterTest(void (*fn)(void), const char *name) { | 21 void RegisterTest(void (*fn)(void), const char *name) { |
22 tests[ntests].fn = fn; | 22 tests[ntests].fn = fn; |
23 tests[ntests++].name = name; | 23 tests[ntests++].name = name; |
24 } | 24 } |
25 | 25 |
| 26 namespace re2 { |
| 27 int64 VirtualProcessSize() { |
| 28 #ifndef WIN32 |
| 29 struct rusage ru; |
| 30 getrusage(RUSAGE_SELF, &ru); |
| 31 return (int64)ru.ru_maxrss*1024; |
| 32 #else |
| 33 return 0; |
| 34 #endif |
| 35 } |
| 36 } // namespace re2 |
| 37 |
26 int main(int argc, char **argv) { | 38 int main(int argc, char **argv) { |
27 for (int i = 0; i < ntests; i++) { | 39 for (int i = 0; i < ntests; i++) { |
28 printf("%s\n", tests[i].name); | 40 printf("%s\n", tests[i].name); |
29 tests[i].fn(); | 41 tests[i].fn(); |
30 } | 42 } |
31 printf("PASS\n"); | 43 printf("PASS\n"); |
32 return 0; | 44 return 0; |
33 } | 45 } |
OLD | NEW |