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