Class: StripeMock::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/stripe_mock/client.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(port) ⇒ Client

Returns a new instance of Client.



5
6
7
8
9
10
11
12
13
14
 
# File 'lib/stripe_mock/client.rb', line 5

def initialize(port)
  @port = port

  DRb.start_service
  @pipe = DRbObject.new_with_uri "druby://localhost:#{port}"

  # Ensure client can connect to server
  timeout_wrap(5) { @pipe.ping }
  @state = 'ready'
end

Instance Attribute Details

#port ⇒ Object (readonly)

Returns the value of attribute port.



3
4
5
 
# File 'lib/stripe_mock/client.rb', line 3

def port
  @port
end

#state ⇒ Object (readonly)

Returns the value of attribute state.



3
4
5
 
# File 'lib/stripe_mock/client.rb', line 3

def state
  @state
end

Instance Method Details

#cleanup ⇒ Object 



81
82
83
84
85
 
# File 'lib/stripe_mock/client.rb', line 81

def cleanup
  return if @state == 'closed'
  set_server_debug(false)
  @state = 'closed'
end

#clear_server_data ⇒ Object 



72
73
74
 
# File 'lib/stripe_mock/client.rb', line 72

def clear_server_data
  timeout_wrap { @pipe.clear_data }
end

#close! ⇒ Object 



76
77
78
79
 
# File 'lib/stripe_mock/client.rb', line 76

def close!
  self.cleanup
  StripeMock.stop_client(:clear_server_data => false)
end

#destroy_resource(type, id) ⇒ Object 



68
69
70
 
# File 'lib/stripe_mock/client.rb', line 68

def destroy_resource(type, id)
  timeout_wrap { @pipe.destroy_resource(type, id) }
end

#error_queue ⇒ Object 



36
37
38
 
# File 'lib/stripe_mock/client.rb', line 36

def error_queue
  timeout_wrap { @pipe.error_queue }
end

#generate_bank_token(recipient_params) ⇒ Object 



56
57
58
 
# File 'lib/stripe_mock/client.rb', line 56

def generate_bank_token(recipient_params)
  timeout_wrap { @pipe.generate_bank_token(recipient_params) }
end

#generate_card_token(card_params) ⇒ Object 



60
61
62
 
# File 'lib/stripe_mock/client.rb', line 60

def generate_card_token(card_params)
  timeout_wrap { @pipe.generate_card_token(card_params) }
end

#generate_webhook_event(event_data) ⇒ Object 



64
65
66
 
# File 'lib/stripe_mock/client.rb', line 64

def generate_webhook_event(event_data)
  timeout_wrap { Stripe::Util.symbolize_names @pipe.generate_webhook_event(event_data) }
end

#get_server_data(key) ⇒ Object 



27
28
29
30
31
32
33
34
 
# File 'lib/stripe_mock/client.rb', line 27

def get_server_data(key)
  timeout_wrap {
    # Massage the data make this behave the same as the local StripeMock.start
    result = {}
    @pipe.get_data(key).each {|k,v| result[k] = Stripe::Util.symbolize_names(v) }
    result
  }
end

#mock_request(method, url, api_key, params = {}, headers = {}) ⇒ Object 



16
17
18
19
20
21
22
23
24
25
 
# File 'lib/stripe_mock/client.rb', line 16

def mock_request(method, url, api_key, params={}, headers={})
  timeout_wrap do
    @pipe.mock_request(method, url, api_key, params, headers).tap {|result|
      response, api_key = result
      if response.is_a?(Hash) && response[:error_raised] == 'invalid_request'
        raise Stripe::InvalidRequestError.new(*response[:error_params])
      end
    }
  end
end

#server_debug? ⇒ Boolean

Returns:

  • (Boolean) 


44
45
46
 
# File 'lib/stripe_mock/client.rb', line 44

def server_debug?
  timeout_wrap { @pipe.debug? }
end

#server_global_id_prefix ⇒ Object 



52
53
54
 
# File 'lib/stripe_mock/client.rb', line 52

def server_global_id_prefix
  timeout_wrap { @pipe.global_id_prefix }
end

#set_server_debug(toggle) ⇒ Object 



40
41
42
 
# File 'lib/stripe_mock/client.rb', line 40

def set_server_debug(toggle)
  timeout_wrap { @pipe.set_debug(toggle) }
end

#set_server_global_id_prefix(value) ⇒ Object 



48
49
50
 
# File 'lib/stripe_mock/client.rb', line 48

def set_server_global_id_prefix(value)
  timeout_wrap { @pipe.set_global_id_prefix(value) }
end

#timeout_wrap(tries = 1) ⇒ Object 



87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
 
# File 'lib/stripe_mock/client.rb', line 87

def timeout_wrap(tries=1)
  original_tries = tries
  begin
    raise ClosedClientConnectionError if @state == 'closed'
    yield
  rescue ClosedClientConnectionError
    raise
  rescue Errno::ECONNREFUSED, DRb::DRbConnError => e
    tries -= 1
    if tries > 0
      if tries == original_tries - 1
        print "Waiting for StripeMock Server.."
      else
        print '.'
      end
      sleep 1
      retry
    else
      raise StripeMock::ServerTimeoutError.new(e)
    end
  end
end