1# frozen_string_literal: true
2
3require "simplecov"
4SimpleCov.start do
5 add_filter "/test/"
6 enable_coverage :branch
7 command_name ENV.fetch("COVERAGE_NAME", "tests")
8end
9
10require "minitest/autorun"
11require "webmock/minitest"
12require "fakefs/safe"
13
14MMS_PROXY = "https://proxy.test.example.com/"
15BW_MESSAGES_URL = "https://messaging.bandwidth.com/api/v2/users/account/messages"
16
17_saved_argv = ARGV.dup
18
19ARGV[0] = "component"
20ARGV[6] = MMS_PROXY
21
22require_relative "../sgx-bwmsgsv2"
23ARGV.replace(_saved_argv)
24
25LOG.level = :fatal
26
27def SGXbwmsgsv2.write_to_stream(s)
28 @written ||= []
29 @written << s
30end
31
32def reset_stanzas!
33 SGXbwmsgsv2.instance_variable_set(:@written, [])
34end
35
36def reset_redis!
37 REDIS.reset!
38 REDIS.set("catapult_jid-", "HERE")
39 REDIS.set("catapult_jid-+15550000000", "test@example.com")
40 REDIS.set("catapult_cred-test@example.com", [
41 'account', 'user', 'password', '+15550000000'
42 ])
43end
44
45def written
46 SGXbwmsgsv2.instance_variable_get(:@written)
47end
48
49def invoke_webhook(payload, extra_env: {})
50 with_stubs([
51 [
52 SGXbwmsgsv2,
53 :write,
54 ->(data) { SGXbwmsgsv2.write_to_stream(data) }
55 ]
56 ]) do
57 handler = WebhookHandler.new
58 env = {
59 "REQUEST_URI" => "/",
60 "REQUEST_METHOD" => "POST",
61 "params" => { "_json" => [payload] }
62 }.merge(extra_env)
63 handler.instance_variable_set(:@env, env)
64 def handler.params
65 @env["params"]
66 end
67 EMPromise.resolve(nil).then {
68 handler.response(env)
69 }.sync
70 end
71end
72
73def process_stanza(s)
74 SGXbwmsgsv2.send(:client).receive_data(s)
75 raise $panic if $panic
76end
77
78def xmpp_error_name(error)
79 error.find_first(
80 "child::*[name()!='text']",
81 Blather::StanzaError::STANZA_ERR_NS
82 ).element_name
83end
84
85def xmpp_error_text(error)
86 error.find_first("ns:text", ns: Blather::StanzaError::STANZA_ERR_NS)&.text
87end
88
89def with_mms_env(path: "/mms", url: "https://mms.test.example.com")
90 old_path = ENV["MMS_PATH"]
91 old_url = ENV["MMS_URL"]
92 ENV["MMS_PATH"] = path
93 ENV["MMS_URL"] = url
94 yield
95ensure
96 ENV["MMS_PATH"] = old_path
97 ENV["MMS_URL"] = old_url
98end
99
100
101begin
102 require "pry-byebug"
103
104 module Minitest
105 class Test
106 alias old_capture_exceptions capture_exceptions
107 def capture_exceptions
108 old_capture_exceptions do
109 yield
110 rescue Minitest::Skip => e
111 failures << e
112 end
113 end
114 end
115 end
116rescue LoadError, NameError
117 # Just helpers for dev, no big deal if missing
118 nil
119end
120
121
122$VERBOSE = nil
123
124class FakeRedis
125 def initialize(values={})
126 @values = values
127 end
128
129 def reset!(values={})
130 @values = values
131 end
132
133 def set(key, value, *)
134 @values[key] = value
135 EMPromise.resolve("OK")
136 end
137
138 def setex(key, _expiry, value)
139 set(key, value)
140 end
141
142 def del(*keys)
143 keys.each { |key| @values.delete(key) }
144 end
145
146 def mget(*keys)
147 EMPromise.all(keys.map(&method(:get)))
148 end
149
150 def get(key)
151 EMPromise.resolve(@values[key])
152 end
153
154 def getbit(key, bit)
155 get(key).then { |v| v.to_i.to_s(2)[bit].to_i }
156 end
157
158 def bitfield(key, *ops)
159 get(key).then do |v|
160 bits = v.to_i.to_s(2)
161 ops.each_slice(3).map do |(op, encoding, offset)|
162 raise "unsupported bitfield op" unless op == "GET"
163 raise "unsupported bitfield op" unless encoding == "u1"
164
165 bits[offset].to_i
166 end
167 end
168 end
169
170 def hget(key, field)
171 @values.dig(key, field)
172 end
173
174 def hincrby(key, field, incrby)
175 @values[key] ||= {}
176 @values[key][field] ||= 0
177 @values[key][field] += incrby
178 end
179
180 def sadd(key, member)
181 @values[key] ||= Set.new
182 @values[key] << member
183 end
184
185 def srem(key, member)
186 @values[key].delete(member)
187 end
188
189 def scard(key)
190 @values[key]&.size || 0
191 end
192
193 def expire(_, _); end
194
195 def exists(*keys)
196 EMPromise.resolve(
197 @values.select { |k, _| keys.include? k }.size.to_s
198 )
199 end
200
201 def lindex(key, index)
202 get(key).then { |v| v&.fetch(index) }
203 end
204
205 def lrange(key, sindex, eindex)
206 get(key).then { |v| v ? v[sindex..eindex] : [] }
207 end
208
209 def rpush(key, *values)
210 @values[key] ||= []
211 values.each { |v| @values[key].push(v) }
212 end
213
214 def multi
215 end
216
217 def exec
218 end
219
220 def discard
221 end
222
223 def xadd(stream, id, *args)
224 @values[stream] ||= []
225 entry_id = id == "*" ? "#{Time.now.to_i}-0" : id
226 fields = Hash[*args]
227 @values[stream] << { id: entry_id, fields: fields }
228 EMPromise.resolve(entry_id)
229 end
230
231 def stream_entries(stream)
232 EMPromise.resolve(@values[stream] || [])
233 end
234end
235
236REDIS = FakeRedis.new
237
238module Minitest
239 class Test
240 def with_stubs(stubs, &block)
241 if stubs.empty?
242 block.call
243 else
244 obj, method, value = stubs.first
245 obj.stub(method, value) do
246 with_stubs(stubs[1..], &block)
247 end
248 end
249 end
250
251 def self.em(m)
252 alias_method "raw_#{m}", m
253 define_method(m) do
254 $panic = nil
255 e = nil
256 FakeFS do
257 FakeFS::FileSystem.clear
258 FileUtils.mkdir_p("/mms")
259 EM.run do
260 Fiber.new {
261 with_mms_env do
262 ARGV[0] = "component"
263 ARGV[6] = MMS_PROXY
264 begin
265 send("raw_#{m}")
266 rescue
267 e = $!
268 ensure
269 EM.stop
270 end
271 end # with_mms_env do
272 }.resume
273 end # EM.run do
274 end # FakeFs do
275 if e
276 LOG.error("Error in test: #{m}", e)
277 raise e
278 end
279 raise e if e
280 end
281 end
282 end
283end