From d54f86ae5850830c1f9eb984b27a51591c643917 Mon Sep 17 00:00:00 2001
From: Adam Roben <adam@roben.org>
Date: Mon, 10 Nov 2014 14:12:29 -0500
Subject: [PATCH] Add some tests for Language.find_by_extension

---
 test/test_language.rb | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/test_language.rb b/test/test_language.rb
index f2bbfc8b..066f577a 100644
--- a/test/test_language.rb
+++ b/test/test_language.rb
@@ -241,6 +241,22 @@ class TestLanguage < Test::Unit::TestCase
     end
   end
 
+  def test_find_by_extension
+    assert_equal [], Language.find_by_extension('.factor-rc')
+    assert_equal [], Language.find_by_extension('rb')
+    assert_equal [], Language.find_by_extension('foo.rb')
+    assert_equal [Language['Ruby']], Language.find_by_extension('.rb')
+    assert_equal [Language['M'], Language['Mathematica'], Language['Matlab'], Language['Mercury'], Language['Objective-C']], Language.find_by_extension('.m')
+  end
+
+  def test_find_all_by_extension
+    Language.all.each do |language|
+      language.extensions.each do |extension|
+        assert_include Language.find_by_extension(extension), language
+      end
+    end
+  end
+
   def test_find_by_filename
     assert_equal [Language['Shell']], Language.find_by_filename('PKGBUILD')
     assert_equal [Language['Ruby']], Language.find_by_filename('foo.rb')