ReferenceCollection proxy

Author: arturcic

src/GitVersionCore.Tests/Core/RepositoryExtensionsTests.cs

@@ -9,6 +9,7 @@
 using NSubstitute;
 using NUnit.Framework;
 using BranchCollection = GitVersion.BranchCollection;
+using ReferenceCollection = GitVersion.ReferenceCollection;
 
 namespace GitVersionCore.Tests
 {
@@ -164,22 +165,14 @@ public TesatbleRemote(string name)
         private class TestableReferenceCollection : ReferenceCollection
         {
             Reference reference;
-
-            public override DirectReference Add(string name, ObjectId targetId)
-            {
-                throw new InvalidOperationException("Update should be invoked when case-insensitively comparing branches.");
-            }
-
             public override Reference Add(string name, string canonicalRefNameOrObjectish)
             {
                 return reference = new TestableReference(canonicalRefNameOrObjectish);
             }
-
             public override Reference UpdateTarget(Reference directRef, ObjectId targetId)
             {
                 return reference;
             }
-
             public override Reference this[string name] => reference;
         }
 

src/GitVersionCore.Tests/Mocks/MockReferenceCollection.cs

@@ -1,5 +1,6 @@
 using System.Collections.Generic;
 using LibGit2Sharp;
+using ReferenceCollection = GitVersion.ReferenceCollection;
 
 namespace GitVersionCore.Tests.Mocks
 {

src/GitVersionCore.Tests/Mocks/MockRepository.cs

@@ -2,6 +2,7 @@
 using GitVersion;
 using LibGit2Sharp;
 using BranchCollection = GitVersion.BranchCollection;
+using ReferenceCollection = GitVersion.ReferenceCollection;
 
 namespace GitVersionCore.Tests.Mocks
 {

src/GitVersionCore/Core/GitModel.cs

@@ -34,4 +34,56 @@ public void Update(Branch branch, params Action<BranchUpdater>[] actions)
             innerBranchCollection.Update(branch, actions);
         }
     }
+
+    public class ReferenceCollection : IEnumerable<Reference>
+    {
+        private readonly LibGit2Sharp.ReferenceCollection innerReferenceCollection;
+        private ReferenceCollection(LibGit2Sharp.ReferenceCollection branchCollection) => innerReferenceCollection = branchCollection;
+
+        protected ReferenceCollection()
+        {
+        }
+
+        public static implicit operator LibGit2Sharp.ReferenceCollection(ReferenceCollection d) => d.innerReferenceCollection;
+        public static explicit operator ReferenceCollection(LibGit2Sharp.ReferenceCollection b) => new ReferenceCollection(b);
+
+        public IEnumerator<Reference> GetEnumerator()
+        {
+            foreach (var reference in innerReferenceCollection)
+                yield return reference;
+        }
+
+        public virtual Reference Add(string name, string canonicalRefNameOrObjectish)
+        {
+            return innerReferenceCollection.Add(name, canonicalRefNameOrObjectish);
+        }
+
+        public virtual DirectReference Add(string name, ObjectId targetId)
+        {
+            return innerReferenceCollection.Add(name, targetId);
+        }
+
+        public virtual DirectReference Add(string name, ObjectId targetId, bool allowOverwrite)
+        {
+            return innerReferenceCollection.Add(name, targetId, allowOverwrite);
+        }
+
+        public virtual Reference UpdateTarget(Reference directRef, ObjectId targetId)
+        {
+            return innerReferenceCollection.UpdateTarget(directRef, targetId);
+        }
+
+        public virtual ReflogCollection Log(string canonicalName)
+        {
+            return innerReferenceCollection.Log(canonicalName);
+        }
+        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
+        public virtual Reference this[string name] => innerReferenceCollection[name];
+        public virtual Reference Head => this["HEAD"];
+
+        public virtual IEnumerable<Reference> FromGlob(string pattern)
+        {
+            return innerReferenceCollection.FromGlob(pattern);
+        }
+    }
 }

src/GitVersionCore/Core/GitRepository.cs

@@ -36,7 +36,7 @@ public void Dispose()
 
         public Branch Head => repositoryInstance.Head;
 
-        public ReferenceCollection Refs => repositoryInstance.Refs;
+        public ReferenceCollection Refs => (ReferenceCollection)repositoryInstance.Refs;
 
         public IQueryableCommitLog Commits => repositoryInstance.Commits;