Ruby Unit test

class SimpleNumber
 
  def initialize(num)
    raise unless num.is_a?(Numeric)
    @x = num
  end
        
  def add(y)
    @x + y
  end
 
  def multiply(y)
    @x * y
  end
 
end

require 'simple_number'
require 'test/unit'

class TestSimpleNumber < Test::Unit::TestCase
    def test_simple
    assert_equal(4, SimpleNumber.new(2).add(2) )
    assert_equal(6, SimpleNumber.new(2).multiply(3) )
  end
end

# ./tc_simple_number.rb 
Loaded suite ./tc_simple_number
Started
.
Finished in 0.000246 seconds.

1 tests, 2 assertions, 0 failures, 0 errors

require 'simple_number'
require 'test/unit'

class TestSimpleNumber < Test::Unit::TestCase
  # 간단한 성공 테스트
  def test_simple
    assert_equal(4, SimpleNumber.new(2).add(2) )
    assert_equal(4, SimpleNumber.new(2).multiply(2) )
  end
  # 타입체크 테스트
  def test_typecheck
    assert_raise( RuntimeError ) { SimpleNumber.new('a') }
  end
  # 실패 테스트
  def test_failure
    assert_equal(3, SimpleNumber.new(2).add(2), "Adding doesn't work" )
  end
end

Loaded suite ./tc_simple_number
Started
F..
Finished in 0.003596 seconds.

  1) Failure:
test_failure(TestSimpleNumber) [./tc_simple_number.rb:17]:
Adding doesn't work.
<3> expected but was
<4>.

3 tests, 4 assertions, 1 failures, 0 errors

# gem install mocha

require "test/unit"
require "mocha"

require 'order'
require 'test/unit'
require 'rubygems'
require 'mocha'

class OrderTest < Test::Unit::TestCase
    def test_delay_cost
        goods_infos = stub()
        # get_delay_cost를 호출할 경우 4(%)를 반환한다.
        goods_infos.stubs(:get_delay_cost).returns(4)
        # get_cost를 호출할 경우 1500000을 반환한다. 
        goods_infos.stubs(:get_cost).returns(1500000)
        myorder = Order.new(goods_infos)

        assert_equal(myorder.goods_cost(1440,50),75000000)
        assert_equal (myorder.delay_cost(2), 6000000)
    end
end

# gem install rspec

require 'order'

describe 'Order App' do
    def mock_goods
        @mock_obj = mock(GoodsInfo)
        @mock_obj.stub(:get_cost) {2000000}
        @mock_obj.stub(:get_delay_cost) {4}
        return @mock_obj
    end 
    it "Order Database test" do
        order = Order.new(mock_goods)
        puts order.goods_cost 1440,50
        order.goods_cost(1440, 50).should eq(100000000)
    end
end 

# rspec tc_order.spec 

# cat myapp.rb
require 'rubygems'
require 'sinatra'
    
module Backend
    class GoodsInfo
        def get_cost (code_num)
            return 100000
        end
        def get_delay_cost
            return 5
        end
    end
end     
        
helpers Backend
    
before do
    @goods = Backend::GoodsInfo.new
end

# 상품번호 1004 의 1 컨테이너당 가격을 가져온다.   
# num은 컨테이너 갯수
# 사용 : GET /goods/cost/1004?num=50
get '/goods/cost/:code_num' do
    cost = @goods.get_cost(params[:code_num])
    goods_cost= cost * Integer(params[:num]) 
    "#{goods_cost}"
end 

require 'myapp'
require 'rack/test'

describe 'Goods Order' do
    include Rack::Test::Methods
    def app
        Sinatra::Application
    end     
    def mock_object
        @goods_mock = mock("Backend::GoodsInfo")
        @goods_mock.stub(:get_cost) {1000000}
        Backend::GoodsInfo.stub(:new).with(any_args()).and_return(@goods_mock)
    end
    it " : Order database test" do
        mock_object
        get '/goods/cost/1004', :num=>40
        last_response.body.should eq("40000000")
    end 
end

require 'myapp'
require 'rack/test'

describe 'Goods Order' do
    include Rack::Test::Methods
    def app
        Sinatra::Application
    end
    def mock_object h
        if h[:cost] == nil then h[:cost] = 1000000 end
        @goods_mock = mock("Backend::GoodsInfo")
        @goods_mock.stub(:get_cost) {h[:cost]}
        Backend::GoodsInfo.stub(:new).with(any_args()).and_return(@goods_mock)
    end
    it " : Order database test, cost is 1000000" do
        mock_object :cost=>1000000
        get '/goods/cost/1004', :num=>40
        last_response.body.should eq("40000000")
    end
    it " : Order database test, cost is 500000" do
        mock_object :cost=>500000
        get '/goods/cost/1004', :num=>40
        last_response.body.should eq("20000000")
    end
end