2005-12-08 Martin Hunt <hunt@redhat.com>

	* maps/map.test: Add size test.
	* pmaps/pmap.test: Add size test.

1 files changed, 72 insertions, 0 deletions

diff --git a/runtime/tests/pmaps/size.c b/runtime/tests/pmaps/size.c
new file mode 100644
index 00000000..be8b4590
--- /dev/null
+++ b/runtime/tests/pmaps/size.c
@@ -0,0 +1,72 @@
+#include "runtime.h"
+
+/* test of _stp_pmap_size() */
+
+/* It's not clear this would ever be used in the systemtap language. 
+   It would be useful as an array of counters. */
+
+#define VALUE_TYPE INT64
+#define KEY1_TYPE INT64
+#include "pmap-gen.c"
+
+#include "map.c"
+
+#define check(map,num)                          \
+  {                                             \
+    int size = _stp_pmap_size(map);              \
+    if (size != num)							\
+      printf("ERROR at line %d: expected size %d and got %d instead.\n", __LINE__, num, size); \
+  }
+
+int main ()
+{
+  PMAP map = _stp_pmap_new_ii(8);
+  int64_t x;
+
+  check(map,0);
+
+  /* put some data in. _processor_number is a global hack that allows */
+  /* us to set the current emulated cpu number for our userspace tests. */
+  /* Note that we set values based on the cpu number just to show that */
+  /* different values are stored in each cpu */
+  for (_processor_number = 0; _processor_number < NR_CPUS; _processor_number++) {
+    _stp_pmap_add_ii(map, 1, _processor_number);
+    _stp_pmap_add_ii(map, 2, 10 *_processor_number + 1);
+    _stp_pmap_add_ii(map, 3, _processor_number * _processor_number);
+    _stp_pmap_add_ii(map, 4, 1);
+  }  
+  _processor_number = 0;
+
+  check(map,4*NR_CPUS-2);
+
+  _stp_pmap_add_ii(map, 1, 1);
+  _stp_pmap_add_ii(map, 3, 1);
+  check(map,4*NR_CPUS);  
+
+  _stp_pmap_add_ii(map, 5, 100);
+  check(map,4*NR_CPUS+1);
+
+  _processor_number = 1;
+  _stp_pmap_add_ii(map, 5, 100);
+  check(map,4*NR_CPUS+2);
+
+  _stp_pmap_set_ii(map, 5, 0);
+  check(map,4*NR_CPUS+1);
+
+  _processor_number = 0;
+  _stp_pmap_set_ii(map, 5, 0);
+  check(map,4*NR_CPUS);
+
+  for (_processor_number = 0; _processor_number < NR_CPUS; _processor_number++) {
+    _stp_pmap_set_ii(map, 1, 0);
+    _stp_pmap_set_ii(map, 2, 0);
+    _stp_pmap_set_ii(map, 3, 0);
+    _stp_pmap_set_ii(map, 4, 0);
+  }  
+  _processor_number = 0;
+  check(map,0);
+
+  _stp_pmap_del (map);
+  return 0;
+}
+