feat(ruby): add add_custom_domains

FGRibreau · Nov 9, 2021 · 167cad2 · 167cad2
1 parent 72b47aa
commit 167cad2
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/platform/ruby/mail_checker.rb b/platform/ruby/mail_checker.rb
@@ -17,6 +17,12 @@ def self.blacklisted?(email)
     extract_all_domain_suffixes(email).any? { |domain| BLACKLIST.include?(domain) }
   end
 
+  def self.add_custom_domains(domains)
+    domains.each do |domain|
+      BLACKLIST.add(domain)
+    end
+  end
+
   def self.extract_all_domain_suffixes(email)
     domain = email.to_s.gsub(/.+@([^.]+)/, '\1').downcase
 

diff --git a/platform/ruby/mail_checker.tmpl.rb b/platform/ruby/mail_checker.tmpl.rb
@@ -17,6 +17,12 @@ def self.blacklisted?(email)
     extract_all_domain_suffixes(email).any? { |domain| BLACKLIST.include?(domain) }
   end
 
+  def self.add_custom_domains(domains)
+    domains.each do |domain|
+      BLACKLIST.add(domain)
+    end
+  end
+
   def self.extract_all_domain_suffixes(email)
     domain = email.to_s.gsub(/.+@([^.]+)/, '\1').downcase
 

diff --git a/test/platform.ruby.test.rb b/test/platform.ruby.test.rb
@@ -60,4 +60,16 @@ def test_extract_all_domain_suffixes
     expected = %w(sub.example.org example.org org)
     assert_equal expected, MailChecker.extract_all_domain_suffixes('test@sub.example.org')
   end
+
+  def test_add_custom_domains
+    valid!('foo@youtube.com')
+    valid!('foo@google.com')
+    valid!('ok@gmail.com')
+
+    MailChecker.add_custom_domains(['youtube.com', 'google.com'])
+
+    invalid!('foo@youtube.com')
+    invalid!('foo@google.com')
+    valid!('ok@gmail.com')
+  end
 end