schema response: support method access to command information

ranguba · Oct 25, 2017 · a1a179c · a1a179c
1 parent 51a3fe0
commit a1a179c
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 0 deletions.
diff --git a/lib/groonga/client/response/schema.rb b/lib/groonga/client/response/schema.rb
@@ -147,6 +147,11 @@ class ValueType < ::Hash
           include Hashie::Extensions::MethodAccess
         end
 
+        class Command < ::Hash
+          include Hashie::Extensions::MergeInitializer
+          include Hashie::Extensions::MethodAccess
+        end
+
         class Index < ::Hash
           include Hashie::Extensions::MethodAccess
 
@@ -204,6 +209,8 @@ def []=(key, value)
             when :value_type
               value = ValueType.new(value) unless value.nil?
               super(key, value)
+            when :command
+              super(key, Command.new(value))
             else
               super
             end
@@ -243,6 +250,8 @@ def []=(key, value)
               super(key, coerce_columns(value))
             when :indexes
               super(key, coerce_indexes(value))
+            when :command
+              super(key, Command.new(value))
             else
               super
             end

diff --git a/test/response/test-schema.rb b/test/response/test-schema.rb
@@ -263,6 +263,30 @@ def test_indexes
           end
         end
       end
+
+      def test_command
+        body = {
+          "tables" => {
+            "Users" => {
+              "command" => {
+                "name" => "table_create",
+                "arguments" => {
+                  "name" => "Users",
+                  "flags" => "TABLE_HASH_KEY",
+                  "key_type" => "ShortText",
+                },
+              }
+            }
+          }
+        }
+        response = create_response(body)
+        assert_equal({
+                       "name" => "Users",
+                       "flags" => "TABLE_HASH_KEY",
+                       "key_type" => "ShortText",
+                     },
+                     response.tables["Users"].command.arguments)
+      end
     end
 
     class TestColumn < self
@@ -410,6 +434,36 @@ def test_indexes
           end
         end
       end
+
+      def test_command
+        body = {
+          "tables" => {
+            "Users" => {
+              "columns" => {
+                "name" => {
+                  "command" => {
+                    "name" => "column_create",
+                    "arguments" => {
+                      "table" => "Users",
+                      "name" => "name",
+                      "flags" => "COLUMN_SCALAR",
+                      "type" => "ShortText",
+                    },
+                  }
+                }
+              }
+            }
+          }
+        }
+        response = create_response(body)
+        assert_equal({
+                       "table" => "Users",
+                       "name" => "name",
+                       "flags" => "COLUMN_SCALAR",
+                       "type" => "ShortText",
+                     },
+                     response.tables["Users"].columns["name"].command.arguments)
+      end
     end
 
     class TestIndex < self