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 messages: :unlimited,
76 minutes: { included: 120, price: 87 }
77 },
78 {
79 name: "test_bad_currency",
80 currency: :BAD
81 },
82 {
83 name: "test_cad",
84 currency: :CAD,
85 monthly_price: 10000
86 }
87 ],
88 braintree: {
89 merchant_accounts: {
90 USD: "merchant_usd"
91 }
92 },
93 sip: {
94 realm: "sip.example.com",
95 app: "sipappid"
96 },
97 credit_card_url: ->(*) { "http://creditcard.example.com" },
98 electrum_notify_url: ->(*) { "http://notify.example.com" },
99 upstream_domain: "example.net",
100 approved_domains: {
101 "approved.example.com": nil,
102 "refer.example.com": "refer_to"
103 }
104}.freeze
105
106def panic(e)
107 raise e
108end
109
110LOG = Class.new {
111 def child(*)
112 Minitest::Mock.new
113 end
114}.new.freeze
115
116BLATHER = Class.new {
117 def <<(*); end
118}.new.freeze
119
120class Matching
121 def initialize(&block)
122 @block = block
123 end
124
125 def ===(other)
126 @block.call(other)
127 end
128end
129
130class PromiseMock < Minitest::Mock
131 def then(succ=nil, _=nil)
132 if succ
133 succ.call(self)
134 else
135 yield self
136 end
137 end
138end
139
140class FakeTelSelections
141 def initialize
142 @selections = {}
143 end
144
145 def set(jid, tel)
146 @selections[jid] = EMPromise.resolve(TelSelections::HaveTel.new(tel))
147 end
148
149 def delete(jid)
150 @selections.delete(jid)
151 EMPromise.resolve("OK")
152 end
153
154 def [](jid)
155 @selections.fetch(jid) do
156 TelSelections::ChooseTel.new
157 end
158 end
159end
160
161class FakeRedis
162 def initialize(values={})
163 @values = values
164 end
165
166 def set(key, value)
167 @values[key] = value
168 EMPromise.resolve("OK")
169 end
170
171 def setex(key, _expiry, value)
172 set(key, value)
173 end
174
175 def mget(*keys)
176 EMPromise.all(keys.map(&method(:get)))
177 end
178
179 def get(key)
180 EMPromise.resolve(@values[key])
181 end
182
183 def getbit(key, bit)
184 get(key).then { |v| v.to_i.to_s(2)[bit].to_i }
185 end
186
187 def exists(*keys)
188 EMPromise.resolve(
189 @values.select { |k, _| keys.include? k }.size
190 )
191 end
192
193 def lindex(key, index)
194 get(key).then { |v| v&.fetch(index) }
195 end
196end
197
198class FakeDB
199 def initialize(items={})
200 @items = items
201 end
202
203 def query_defer(_, args)
204 EMPromise.resolve(@items.fetch(args, []))
205 end
206end
207
208class FakeIBRRepo
209 def initialize(registrations={})
210 @registrations = registrations
211 end
212
213 def registered?(jid, from:)
214 @registrations.dig(jid.to_s, from.to_s) || false
215 end
216end
217
218module EventMachine
219 class << self
220 # Patch EM.add_timer to be instant in tests
221 alias old_add_timer add_timer
222 def add_timer(*args, &block)
223 args[0] = 0
224 old_add_timer(*args, &block)
225 end
226 end
227end
228
229module Minitest
230 class Test
231 def self.property(m, &block)
232 define_method("test_#{m}") do
233 property_of(&block).check { |args| send(m, *args) }
234 end
235 end
236
237 def self.em(m)
238 alias_method "raw_#{m}", m
239 define_method(m) do
240 EM.run do
241 Fiber.new {
242 begin
243 send("raw_#{m}")
244 ensure
245 EM.stop
246 end
247 }.resume
248 end
249 end
250 end
251 end
252end