1# frozen_string_literal: true
2
3require "simplecov"
4SimpleCov.start do
5 add_filter "/test/"
6 enable_coverage :branch
7end
8
9require "em_promise"
10require "fiber"
11require "minitest/autorun"
12require "rantly/minitest_extensions"
13require "sentry-ruby"
14require "webmock/minitest"
15begin
16 require "pry-rescue/minitest"
17 require "pry-reload"
18
19 module Minitest
20 class Test
21 alias old_capture_exceptions capture_exceptions
22 def capture_exceptions
23 old_capture_exceptions do
24 yield
25 rescue Minitest::Skip => e
26 failures << e
27 end
28 end
29 end
30 end
31rescue LoadError
32 # Just helpers for dev, no big deal if missing
33 nil
34end
35
36require "backend_sgx"
37
38$VERBOSE = nil
39Sentry.init
40
41def customer(customer_id="test", plan_name: nil, **kwargs)
42 jid = Blather::JID.new("#{customer_id}@example.net")
43 if plan_name
44 expires_at = kwargs.delete(:expires_at) || Time.now
45 plan = CustomerPlan.new(
46 customer_id,
47 plan: Plan.for(plan_name),
48 expires_at: expires_at
49 )
50 Customer.new(customer_id, jid, plan: plan, **kwargs)
51 else
52 Customer.new(customer_id, jid, **kwargs)
53 end
54end
55
56CONFIG = {
57 sgx: "sgx",
58 component: {
59 jid: "component"
60 },
61 creds: {
62 account: "test_bw_account",
63 username: "test_bw_user",
64 password: "test_bw_password"
65 },
66 catapult: {
67 user: "catapult_user",
68 token: "catapult_token",
69 secret: "catapult_secret",
70 domain: "catapult_domain",
71 sip_host: "host.bwapp.io.example.com",
72 application_id: "catapult_app"
73 },
74 activation_amount: 1,
75 plans: [
76 {
77 name: "test_usd",
78 currency: :USD,
79 monthly_price: 10000
80 },
81 {
82 name: "test_bad_currency",
83 currency: :BAD
84 },
85 {
86 name: "test_cad",
87 currency: :CAD,
88 monthly_price: 10000
89 }
90 ],
91 braintree: {
92 merchant_accounts: {
93 USD: "merchant_usd"
94 }
95 },
96 credit_card_url: ->(*) { "http://creditcard.example.com" },
97 electrum_notify_url: ->(*) { "http://notify.example.com" },
98 upstream_domain: "example.net"
99}.freeze
100
101def panic(e)
102 raise e
103end
104
105LOG = Class.new {
106 def child(*)
107 Minitest::Mock.new
108 end
109}.new.freeze
110
111BLATHER = Class.new {
112 def <<(*); end
113}.new.freeze
114
115class Matching
116 def initialize(&block)
117 @block = block
118 end
119
120 def ===(other)
121 @block.call(other)
122 end
123end
124
125class PromiseMock < Minitest::Mock
126 def then(succ=nil, _=nil)
127 if succ
128 succ.call(self)
129 else
130 yield self
131 end
132 end
133end
134
135class FakeRedis
136 def initialize(values={})
137 @values = values
138 end
139
140 def set(key, value)
141 @values[key] = value
142 EMPromise.resolve("OK")
143 end
144
145 def setex(key, value, _expiry)
146 set(key, value)
147 end
148
149 def get(key)
150 EMPromise.resolve(@values[key])
151 end
152
153 def exists(*keys)
154 EMPromise.resolve(
155 @values.select { |k, _| keys.include? k }.size
156 )
157 end
158
159 def lindex(key, index)
160 get(key).then { |v| v&.fetch(index) }
161 end
162end
163
164class FakeDB
165 def initialize(items)
166 @items = items
167 end
168
169 def query_defer(_, args)
170 EMPromise.resolve(@items.fetch(args, []))
171 end
172end
173
174module EventMachine
175 class << self
176 # Patch EM.add_timer to be instant in tests
177 alias old_add_timer add_timer
178 def add_timer(*args, &block)
179 args[0] = 0
180 old_add_timer(*args, &block)
181 end
182 end
183end
184
185module Minitest
186 class Test
187 def self.property(m, &block)
188 define_method("test_#{m}") do
189 property_of(&block).check { |args| send(m, *args) }
190 end
191 end
192
193 def self.em(m)
194 alias_method "raw_#{m}", m
195 define_method(m) do
196 EM.run do
197 Fiber.new {
198 begin
199 send("raw_#{m}")
200 ensure
201 EM.stop
202 end
203 }.resume
204 end
205 end
206 end
207 end
208end