spec/mwo/mech_spec.rb in mwo-0.1.0 vs spec/mwo/mech_spec.rb in mwo-0.1.1

@@ -1,14 +1,46 @@
 require 'spec_helper'
 
 describe MWO::Mech do
   describe '.all', vcr: {cassette_name: 'all_mechs'} do
     let (:subject) { described_class.all }
-    let (:hbk4g) { an_object_having_attributes(name: 'hbk-4g', max_armor: 338, mech_id: 1, total_tons: 50) }
+    let (:hbk4g) { an_object_having_attributes(name: 'hbk-4g', max_armor: 338, mech_id: 1, total_tons: 50, weight_class: 'Medium') }
+    let (:com3a) { an_object_having_attributes(name: 'com-3a', max_armor: 178, mech_id: 6, total_tons: 25, weight_class: 'Light') }
     it "returns a collection of mechs collection" do
       expect(subject).to_not be_empty
       expect(subject).to include(an_instance_of(described_class))
-      expect(subject).to include(hbk4g)
+      expect(subject).to include(hbk4g, com3a)
+    end
+  end
+
+  describe 'Weight class scoping', vcr: {cassette_name: 'all_mechs'} do
+    let(:light_mech) {an_object_having_attributes(weight_class: 'Light')}
+    let(:medium_mech) {an_object_having_attributes(weight_class: 'Medium')}
+    let(:heavy_mech) {an_object_having_attributes(weight_class: 'Heavy')}
+    let(:assault_mech) {an_object_having_attributes(weight_class: 'Assault')}
+
+    describe '.light' do
+      subject {described_class.lights}
+      it { is_expected.to include(light_mech) }
+      it { is_expected.to_not include(medium_mech, heavy_mech, assault_mech) }
+    end
+
+    describe '.mediums' do
+      subject {described_class.mediums}
+      it { is_expected.to include(medium_mech) }
+      it { is_expected.to_not include(light_mech, heavy_mech, assault_mech) }
+    end
+
+    describe '.heavies' do
+      subject {described_class.heavies}
+      it { is_expected.to include(heavy_mech) }
+      it { is_expected.to_not include(light_mech, medium_mech, assault_mech) }
+    end
+
+    describe '.assaults' do
+      subject {described_class.assaults}
+      it { is_expected.to include(assault_mech) }
+      it { is_expected.to_not include(light_mech, medium_mech, heavy_mech) }
     end
   end
 
   describe '.dictionary', vcr: {cassette_name: 'mech_dictionary'} do
     let (:subject) { described_class.dictionary }