spec/models/naf/affinity_classification_spec.rb in naf-1.1.4 vs spec/models/naf/affinity_classification_spec.rb in naf-2.0.0

@@ -50,7 +50,27 @@
       it "return the application affinity classification" do
         ::Naf::AffinityClassification.application.should == application_affinity_classification
       end
     end
 
+    describe "#weight" do
+      let!(:weight_affinity_classification) { FactoryGirl.create(:affinity_classification,
+                                                                 affinity_classification_name: 'weight',
+                                                                 id: 4) }
+
+      it "return the weight affinity classification" do
+        ::Naf::AffinityClassification.weight.should == weight_affinity_classification
+      end
+    end
+
+    describe "#machine" do
+      let!(:machine_affinity_classification) { FactoryGirl.create(:affinity_classification,
+                                                                  affinity_classification_name: 'machine',
+                                                                  id: 5) }
+
+      it "return the machine affinity classification" do
+        ::Naf::AffinityClassification.machine.should == machine_affinity_classification
+      end
+    end
+
   end
 end