Add column_family_property to mock class.

cassandra-rb · Apr 23, 2011 · f51ef30 · f51ef30
1 parent e13ea36
commit f51ef30
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/cassandra/mock.rb b/lib/cassandra/mock.rb
@@ -283,7 +283,11 @@ def to_compare_with_type(column_name, column_family, standard=true)
     end
 
     def column_family_type(column_family)
-      schema[column_family.to_s]['Type']
+      column_family_property(column_family, 'Type')
+    end
+
+    def column_family_property(column_family, property)
+      schema[column_family.to_s][property]
     end
 
     def cf(column_family)