Fix read_attribute, see #396 (#397)

* Fix read_attribute, see #396 * Fix typo Co-authored-by: George Dietrich <george@dietrich.app>
amberframework · May 5, 2020 · 1daaa35 · 1daaa35
1 parent 536a9b4
commit 1daaa35
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/spec/granite/columns/read_attribute_spec.cr b/spec/granite/columns/read_attribute_spec.cr
@@ -0,0 +1,10 @@
+require "../../spec_helper"
+
+describe "read_attribute" do
+  # Only PG supports array types
+  {% if env("CURRENT_ADAPTER") == "pg" %}
+    it "able to read arrays" do
+      ArrayModel.new.read_attribute("i32_array").should be_nil
+    end
+  {% end %}
+end
diff --git a/src/granite/columns.cr b/src/granite/columns.cr
@@ -145,7 +145,7 @@ module Granite::Columns
     self
   end
 
-  def read_attribute(attribute_name : Symbol | String) : DB::Any
+  def read_attribute(attribute_name : Symbol | String) : Type
     {% begin %}
       case attribute_name.to_s
       {% for column in @type.instance_vars.select { |ivar| ivar.annotation(Granite::Column) } %}