spec/lib/order_spec.rb in trebbianno-ruby-api-0.0.3 vs spec/lib/order_spec.rb in trebbianno-ruby-api-0.0.4

@@ -1,35 +1,51 @@
 require 'spec_helper'
 
 describe Trebbianno::Order do
 
-  before do
-    @client       = Trebbianno::Client.new("the_username", "the_password")
-    @order_client = Trebbianno::Order.new(@client)
-    @xml          = Builder::XmlMarkup.new
-  end
+  let(:client)       { Trebbianno::Client.new("the_username", "the_password") }
+  let(:order_client) { Trebbianno::Order.new(client) }
+  let(:xml)          { Builder::XmlMarkup.new }
 
   describe '#build_order_request' do
     it 'should' do
-      response = @order_client.build_order_request(order_hash)
-      @client.type.should == "new_order_entry"
-      request_body = xml_order_request_string(order_hash)
-      response.should == xml_string("new_order_entry", request_body)
+      response = order_client.build_order_request(order_hash)
+      expect(response).to eq read_xml(:test_build_order_request)
     end
   end
 
   describe 'private#build_address' do
+    before do
+      xml = ::Builder::XmlMarkup.new :indent => 3
+      xml.instruct! :xml, :version=>"1.0", :encoding=>"UTF-8"
+    end
     it 'should build the proper address xml' do
-      type    = "CUSTOMER"
-      request = @order_client.send(:build_address, @xml, type, order_hash[:shipping_address])
-      request.should == xml_address_string(type, order_hash[:shipping_address])
+      request = order_client.send(:build_address, xml, order_hash[:shipping_address])
+      expect(request).to eq read_xml(:test_build_address)
     end
   end
 
   describe 'private#build_line_items' do
     it 'should build the proper line items xml' do
-      request = @order_client.send(:build_line_items, @xml, order_hash)
-      request.should == xml_line_items_string(order_hash)
+      request = order_client.send(:build_line_items, xml, order_hash)
+      expect(request).to eq test_line_items_string(order_hash)
     end
   end
 
-end
\ No newline at end of file
+  describe 'private#freight_charge' do
+    it 'should claculate freight charges for order' do
+      charge = order_client.send(:freight_charge, order_hash)
+      expect(charge).to eq 30
+    end
+  end
+
+  describe 'private#line_item_discount' do
+    let(:discount) {[20, 20]}
+    it 'should calculate discount price for line item' do
+      order_hash[:line_items].each_with_index  do |item, index|
+        discount_amount = order_client.send(:line_item_discount, order_hash)
+        expect(discount_amount).to eq  discount[index]
+      end
+    end
+  end
+
+end