From 3ff1e38f6c3830827507b700e9ad27bf45ad038a Mon Sep 17 00:00:00 2001
From: Arfon Smith <arfon@github.com>
Date: Thu, 25 Sep 2014 12:50:42 -0500
Subject: [PATCH] Adding support for overriding configurations in vendor.yml
 and generated?

---
 lib/linguist/lazy_blob.rb | 23 +++++++++++++++---
 test/test_repository.rb   | 50 +++++++++++++++++++++++++++------------
 2 files changed, 55 insertions(+), 18 deletions(-)

diff --git a/lib/linguist/lazy_blob.rb b/lib/linguist/lazy_blob.rb
index 26e1ae88..1f16999d 100644
--- a/lib/linguist/lazy_blob.rb
+++ b/lib/linguist/lazy_blob.rb
@@ -1,7 +1,7 @@
 require 'linguist/blob_helper'
 require 'linguist/language'
 require 'rugged'
-
+require 'pry'
 module Linguist
   class LazyBlob
     GIT_ATTR = ['linguist-language', 'linguist-vendored', 'linguist-generated']
@@ -30,11 +30,28 @@ module Linguist
     end
 
     def linguist_vendored?
-      vendored? || !!git_attributes['linguist-vendored']
+      if git_attributes['linguist-vendored']
+        return result_for_key('linguist-vendored')
+      else
+        return vendored?
+      end
     end
 
     def linguist_generated?
-      generated? || !!git_attributes['linguist-generated']
+      if git_attributes['linguist-generated']
+        return result_for_key('linguist-generated')
+      else
+        return generated?
+      end
+    end
+
+    def result_for_key(keyname)
+      key = git_attributes[keyname]
+      if key == "false"
+        return false
+      else
+        return true
+      end
     end
 
     def overriden_language
diff --git a/test/test_repository.rb b/test/test_repository.rb
index 39bec41a..c0269d5d 100644
--- a/test/test_repository.rb
+++ b/test/test_repository.rb
@@ -49,14 +49,17 @@ class TestRepository < Test::Unit::TestCase
   end
 
   def test_repo_git_attributes
-    # See https://github.com/github/linguist/blob/3770a90251c8a940367300c9f6f97bb64b369bf8/.gitattributes
+    # See https://github.com/github/linguist/blob/b533b682d5d4012ca42f4fc998b45169ec41fe33/.gitattributes
     #
     # It looks like this:
-    # test/*.rb linguist-vendored
+    # Gemfile linguist-vendored=true
     # lib/linguist.rb linguist-language=Java
-    # lib/linguist/classifier.rb linguist-generated
+    # test/*.rb linguist-language=Java
+    # Rakefile linguist-generated
+    # test/fixtures/* linguist-vendored=false
 
-    attr_commit = '178d4756efb647d5f8607d74fe47a852329d7516'
+
+    attr_commit = 'b533b682d5d4012ca42f4fc998b45169ec41fe33'
     repo = linguist_repo(attr_commit)
 
     assert repo.breakdown_by_file.has_key?("Java")
@@ -64,28 +67,45 @@ class TestRepository < Test::Unit::TestCase
 
     assert repo.breakdown_by_file.has_key?("Ruby")
     assert !repo.breakdown_by_file["Ruby"].empty?
-    repo.breakdown_by_file["Ruby"].each do |file|
-      assert !file.start_with?("test/"), "Failing for #{file}"
-    end
   end
 
-  def test_linguist_generated?
-    attr_commit = '178d4756efb647d5f8607d74fe47a852329d7516'
-    file = Linguist::LazyBlob.new(rugged_repository, attr_commit, 'lib/linguist/classifier.rb')
 
+  def test_linguist_generated?
+    attr_commit = 'b533b682d5d4012ca42f4fc998b45169ec41fe33'
+    file = Linguist::LazyBlob.new(rugged_repository, attr_commit, 'Rakefile')
+
+
+    # check we're getting the correct assignment back from .gitattributes
+    assert file.result_for_key('linguist-generated')
     # overridden in .gitattributes
     assert file.linguist_generated?
     # from lib/linguist/generated.rb
     assert !file.generated?
   end
 
-  def test_linguist_vendored?
-    attr_commit = '178d4756efb647d5f8607d74fe47a852329d7516'
-    file = Linguist::LazyBlob.new(rugged_repository, attr_commit, 'test/test_md5.rb')
+  def test_linguist_override_vendored?
+    attr_commit = 'b533b682d5d4012ca42f4fc998b45169ec41fe33'
+    override_vendored = Linguist::LazyBlob.new(rugged_repository, attr_commit, 'Gemfile')
 
+    # check we're getting the correct assignment back from .gitattributes
+    assert override_vendored.result_for_key('linguist-vendored')
     # overridden .gitattributes
-    assert file.linguist_vendored?
+    assert override_vendored.linguist_vendored?
     # from lib/linguist/vendor.yml
-    assert !file.vendored?
+    assert !override_vendored.vendored?
+  end
+
+  def test_linguist_override_unvendored?
+    attr_commit = 'b533b682d5d4012ca42f4fc998b45169ec41fe33'
+
+    # lib/linguist/vendor.yml defines this as vendored.
+    override_unvendored = Linguist::LazyBlob.new(rugged_repository, attr_commit, 'test/fixtures/foo.rb')
+
+    # check we're getting the correct assignment back from .gitattributes
+    assert !override_unvendored.result_for_key('linguist-vendored')
+    # overridden .gitattributes
+    assert !override_unvendored.linguist_vendored?
+    # from lib/linguist/vendor.yml
+    assert override_unvendored.vendored?
   end
 end