Add grn_obj_is_tokenizer_proc()

groonga · Oct 19, 2015 · a5e8ba7 · a5e8ba7
1 parent 1a6d14f
commit a5e8ba7
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/include/groonga/obj.h b/include/groonga/obj.h
@@ -27,6 +27,7 @@ GRN_API grn_bool grn_obj_is_builtin(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_table(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_type(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_proc(grn_ctx *ctx, grn_obj *obj);
+GRN_API grn_bool grn_obj_is_tokenizer_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_function_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_selector_proc(grn_ctx *ctx, grn_obj *obj);
 GRN_API grn_bool grn_obj_is_selector_only_proc(grn_ctx *ctx, grn_obj *obj);

diff --git a/lib/obj.c b/lib/obj.c
@@ -76,6 +76,19 @@ grn_obj_is_proc(grn_ctx *ctx, grn_obj *obj)
   return obj->header.type == GRN_PROC;
 }
 
+grn_bool
+grn_obj_is_tokenizer_proc(grn_ctx *ctx, grn_obj *obj)
+{
+  grn_proc *proc;
+
+  if (!grn_obj_is_proc(ctx, obj)) {
+    return GRN_FALSE;
+  }
+
+  proc = (grn_proc *)obj;
+  return proc->type == GRN_PROC_TOKENIZER;
+}
+
 grn_bool
 grn_obj_is_function_proc(grn_ctx *ctx, grn_obj *obj)
 {

diff --git a/test/unit/core/test-object.c b/test/unit/core/test-object.c
@@ -33,6 +33,8 @@ void data_is_type(void);
 void test_is_type(gconstpointer data);
 void data_is_proc(void);
 void test_is_proc(gconstpointer data);
+void data_is_tokenizer_proc(void);
+void test_is_tokenizer_proc(gconstpointer data);
 void data_is_function_proc(void);
 void test_is_function_proc(gconstpointer data);
 void data_is_selector_proc(void);
@@ -231,6 +233,38 @@ test_is_proc(gconstpointer data)
   }
 }
 
+void
+data_is_tokenizer_proc(void)
+{
+#define ADD_DATUM(expected, name)                                       \
+  gcut_add_datum((expected ?                                            \
+                  "tokenizer-proc - " name :                            \
+                  "not tokenizer-proc - " name),                        \
+                 "expected", G_TYPE_BOOLEAN, expected,                  \
+                 "name", G_TYPE_STRING, name,                           \
+                 NULL)
+
+  ADD_DATUM(TRUE, "TokenBigram");
+  ADD_DATUM(FALSE, "status");
+
+#undef ADD_DATUM
+}
+
+void
+test_is_tokenizer_proc(gconstpointer data)
+{
+  const gchar *name;
+  grn_obj *object;
+
+  name = gcut_data_get_string(data, "name");
+  object = grn_ctx_get(context, name, strlen(name));
+  if (gcut_data_get_string(data, "expected")) {
+    cut_assert_true(grn_obj_is_tokenizer_proc(context, object));
+  } else {
+    cut_assert_false(grn_obj_is_tokenizer_proc(context, object));
+  }
+}
+
 void
 data_is_function_proc(void)
 {