jumpcheng · August 29, 2015 14:21 · May 12, 2015
diff --git a/cohort_spec.rb b/cohort_spec.rb
@@ -0,0 +1,38 @@
+require_relative '../app'
+
+describe Cohort do
+  before :each do
+    Cohort.transaction
+  end
+
+  after :each do
+    Cohort.rollback
+  end
+
+  describe "#initialize" do
+    it "should take value from input attributes" do
+      cohort = Cohort.new("name" => "FierySkippers!")
+      expect(cohort[:name]).to eq "FierySkippers!"
+    end  
+
+    it "should not take value from input attributes which not in the list" do
+      expect{
+        cohort = Cohort.new("made_up_attribute" => "Hello Mum!")
+      }.to raise_error(Database::InvalidAttributeError)
+    end
+  end
+
+  describe "::all" do
+
+    it "should return an array of Cohort objects" do
+       expect(Cohort.all[0]).to be_a Cohort 
+    end
+
+    it "should return an array of count(*) length" do
+      count = Database::Model.execute("SELECT count(*) AS CNT FROM cohorts")[0]['CNT']
+
+       expect(Cohort.all.length).to eq count
+    end
+
+  end
+end