spec/unit/cassanity/column_family_spec.rb in cassanity-0.4.0 vs spec/unit/cassanity/column_family_spec.rb in cassanity-0.5.0

@@ -1,10 +1,10 @@
 require 'helper'
 require 'cassanity/column_family'
 
 describe Cassanity::ColumnFamily do
-  let(:column_family_name) { 'analytics' }
+  let(:column_family_name) { :analytics }
   let(:keyspace_name) { 'foo' }
 
   let(:keyspace) {
     double('Keyspace', {
       executor: executor,
@@ -79,10 +79,19 @@
         expect { described_class.new(args) }.to raise_error(KeyError)
       end
     end
   end
 
+  context "with string name" do
+    it "converts name to symbol" do
+      column_family = described_class.new(required_arguments.merge({
+        name: 'foo',
+      }))
+      column_family.name.should be(:foo)
+    end
+  end
+
   describe "#schema" do
     it "returns schema if set" do
       described_class.new(required_arguments.merge({
         schema: schema,
       })).schema.should eq(schema)
@@ -98,22 +107,30 @@
   shared_examples_for "column family existence" do |method_name|
     it "returns true if column family name included in column families" do
       executor.should_receive(:call).with({
         command: :column_families,
         arguments: {keyspace_name: keyspace.name},
+        transformer_arguments: {keyspace: keyspace},
       }).and_return([
-        {'columnfamily' => column_family_name},
+        Cassanity::ColumnFamily.new({
+          name: column_family_name,
+          keyspace: keyspace,
+        })
       ])
 
       subject.send(method_name).should be_true
     end
 
     it "returns false if column family name not included in column families" do
       executor.should_receive(:call).with({
         command: :column_families,
         arguments: {keyspace_name: keyspace.name},
+        transformer_arguments: {keyspace: keyspace},
       }).and_return([
-        {'columnfamily' => 'boo'},
+        Cassanity::ColumnFamily.new({
+          name: 'boo',
+          keyspace: keyspace,
+        })
       ])
 
       subject.send(method_name).should be_false
     end
   end