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"
37require "tel_selections"
38
39$VERBOSE = nil
40Sentry.init
41
42def customer(customer_id="test", plan_name: nil, **kwargs)
43 jid = kwargs.delete(:jid) || Blather::JID.new("#{customer_id}@example.net")
44 if plan_name
45 expires_at = kwargs.delete(:expires_at) || Time.now
46 plan = CustomerPlan.new(
47 customer_id,
48 plan: Plan.for(plan_name),
49 expires_at: expires_at
50 )
51 Customer.new(customer_id, jid, plan: plan, **kwargs)
52 else
53 Customer.new(customer_id, jid, **kwargs)
54 end
55end
56
57CONFIG = {
58 sgx: "sgx",
59 component: {
60 jid: "component"
61 },
62 creds: {
63 account: "test_bw_account",
64 username: "test_bw_user",
65 password: "test_bw_password"
66 },
67 catapult: {
68 user: "catapult_user",
69 token: "catapult_token",
70 secret: "catapult_secret",
71 domain: "catapult_domain",
72 sip_host: "host.bwapp.io.example.com",
73 application_id: "catapult_app"
74 },
75 activation_amount: 1,
76 plans: [
77 {
78 name: "test_usd",
79 currency: :USD,
80 monthly_price: 10000
81 },
82 {
83 name: "test_bad_currency",
84 currency: :BAD
85 },
86 {
87 name: "test_cad",
88 currency: :CAD,
89 monthly_price: 10000
90 }
91 ],
92 braintree: {
93 merchant_accounts: {
94 USD: "merchant_usd"
95 }
96 },
97 sip: {
98 realm: "sip.example.com",
99 app: "sipappid"
100 },
101 credit_card_url: ->(*) { "http://creditcard.example.com" },
102 electrum_notify_url: ->(*) { "http://notify.example.com" },
103 upstream_domain: "example.net",
104 approved_domains: {
105 "approved.example.com": nil,
106 "refer.example.com": "refer_to"
107 }
108}.freeze
109
110def panic(e)
111 raise e
112end
113
114LOG = Class.new {
115 def child(*)
116 Minitest::Mock.new
117 end
118}.new.freeze
119
120BLATHER = Class.new {
121 def <<(*); end
122}.new.freeze
123
124class Matching
125 def initialize(&block)
126 @block = block
127 end
128
129 def ===(other)
130 @block.call(other)
131 end
132end
133
134class PromiseMock < Minitest::Mock
135 def then(succ=nil, _=nil)
136 if succ
137 succ.call(self)
138 else
139 yield self
140 end
141 end
142end
143
144class FakeTelSelections
145 def initialize
146 @selections = {}
147 end
148
149 def set(jid, tel)
150 @selections[jid] = EMPromise.resolve(TelSelections::HaveTel.new(tel))
151 end
152
153 def delete(jid)
154 @selections.delete(jid)
155 EMPromise.resolve("OK")
156 end
157
158 def [](jid)
159 @selections.fetch(jid) do
160 TelSelections::ChooseTel.new
161 end
162 end
163end
164
165class FakeRedis
166 def initialize(values={})
167 @values = values
168 end
169
170 def set(key, value)
171 @values[key] = value
172 EMPromise.resolve("OK")
173 end
174
175 def setex(key, _expiry, value)
176 set(key, value)
177 end
178
179 def mget(*keys)
180 EMPromise.all(keys.map(&method(:get)))
181 end
182
183 def get(key)
184 EMPromise.resolve(@values[key])
185 end
186
187 def getbit(key, bit)
188 get(key).then { |v| v.to_i.to_s(2)[bit].to_i }
189 end
190
191 def exists(*keys)
192 EMPromise.resolve(
193 @values.select { |k, _| keys.include? k }.size
194 )
195 end
196
197 def lindex(key, index)
198 get(key).then { |v| v&.fetch(index) }
199 end
200end
201
202class FakeDB
203 def initialize(items)
204 @items = items
205 end
206
207 def query_defer(_, args)
208 EMPromise.resolve(@items.fetch(args, []))
209 end
210end
211
212module EventMachine
213 class << self
214 # Patch EM.add_timer to be instant in tests
215 alias old_add_timer add_timer
216 def add_timer(*args, &block)
217 args[0] = 0
218 old_add_timer(*args, &block)
219 end
220 end
221end
222
223module Minitest
224 class Test
225 def self.property(m, &block)
226 define_method("test_#{m}") do
227 property_of(&block).check { |args| send(m, *args) }
228 end
229 end
230
231 def self.em(m)
232 alias_method "raw_#{m}", m
233 define_method(m) do
234 EM.run do
235 Fiber.new {
236 begin
237 send("raw_#{m}")
238 ensure
239 EM.stop
240 end
241 }.resume
242 end
243 end
244 end
245 end
246end