Change summary
lib/parent_code_repo.rb | 7 ++++++-
test/test_registration.rb | 5 +++++
2 files changed, 11 insertions(+), 1 deletion(-)
Detailed changes
@@ -9,7 +9,12 @@ class ParentCodeRepo
end
def find(code)
- @redis.hget("jmp_parent_codes", code)
+ @redis.hget("jmp_parent_codes", code).then do |parent_id|
+ (parent_id ? @redis.get("jmp_customer_trust_level-#{parent_id}") : nil)
+ .then do |tl|
+ tl == "Tomb" ? nil : parent_id
+ end
+ end
end
def find_or_create(customer_id)
@@ -794,6 +794,11 @@ class RegistrationTest < Minitest::Test
EMPromise.resolve("parent_customer"),
["jmp_parent_codes", "pabc"]
)
+ Registration::Payment::InviteCode::REDIS.expect(
+ :get,
+ EMPromise.resolve(nil),
+ ["jmp_customer_trust_level-parent_customer"]
+ )
CustomerPlan::DB.expect(
:query,
[{ "plan_name" => "test_usd" }],