2016-12-07 65 views
0

我有兩個方法沒有類,它們都有相同的代碼,但返回對象是不同的。Chefspec嘲諷問題

def get_load_balancer(network_client, resource_group_name, load_balancer_name) 
    promise = network_client.load_balancers.get(resource_group_name, load_balancer_name) 
    load_balancer = promise.value! 
    return load_balancer.body 
end 

def get_public_ip(network_client, resource_group_name, public_ip_name) 
    promise = network_client.public_ip_addresses.get(resource_group_name, public_ip_name) 
    public_ip = promise.value! 
    return public_ip.body 
end 

我想嘲笑它像

@network_client = NetworkResourceProviderClient.new(token) 

@promise_lb = Concurrent::Promise.new 
allow(@network_client).to receive_message_chain(:load_balancers, :get).and_return(@promise_lb) 
allow(@promise_lb).to receive(:value!).and_return(response_lb) 

@promise_pip = Concurrent::Promise.new 
allow(@network_client).to receive_message_chain(:public_ip_addresses, :get).and_return(@promise_pip) 
allow(@promise_pip).to receive(:value!).and_return(response_pip) 

問題是,傳遞對象,用嘲諷不工作。請幫我解決這個問題。

回答

0

你爲什麼用Concurrent::Promise? ChefSpec是RSpec庫,您可以使用rspec-mocks系統。

before do 
    network_client = instance_double(NetworkResourceProviderClient) 
    lbs = double() 
    allow(lbs).to receive(:get).with('foo', 'bar').and_return(double(value!: 'baz')) 
    allow(network_client).to receive(:load_balancers).and_return(lbs) 
end