BranchCollection proxy

Author: arturcic

src/GitVersionCore.Tests/Core/RepositoryExtensionsTests.cs

@@ -8,6 +8,7 @@
 using LibGit2Sharp;
 using NSubstitute;
 using NUnit.Framework;
+using BranchCollection = GitVersion.BranchCollection;
 
 namespace GitVersionCore.Tests
 {

src/GitVersionCore.Tests/Mocks/MockBranchCollection.cs

@@ -1,6 +1,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using LibGit2Sharp;
+using BranchCollection = GitVersion.BranchCollection;
 
 namespace GitVersionCore.Tests.Mocks
 {
@@ -38,11 +39,6 @@ public void CopyTo(Branch[] array, int arrayIndex)
             Branches.CopyTo(array, arrayIndex);
         }
 
-        public override void Remove(Branch item)
-        {
-            Branches.Remove(item);
-        }
-
         public int Count => Branches.Count;
         public bool IsReadOnly => false;
     }

src/GitVersionCore.Tests/Mocks/MockRepository.cs

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

src/GitVersionCore/Core/GitModel.cs

@@ -0,0 +1,37 @@
+using System;
+using System.Collections;
+using System.Collections.Generic;
+using LibGit2Sharp;
+
+namespace GitVersion
+{
+    public class BranchCollection : IEnumerable<Branch>
+    {
+        private readonly LibGit2Sharp.BranchCollection innerBranchCollection;
+        private BranchCollection(LibGit2Sharp.BranchCollection branchCollection) => innerBranchCollection = branchCollection;
+
+        protected BranchCollection()
+        {
+        }
+
+        public static implicit operator LibGit2Sharp.BranchCollection(BranchCollection d) => d.innerBranchCollection;
+        public static explicit operator BranchCollection(LibGit2Sharp.BranchCollection b) => new BranchCollection(b);
+
+        public virtual IEnumerator<Branch> GetEnumerator()
+        {
+            foreach (var branch in innerBranchCollection)
+                yield return branch;
+        }
+
+        public virtual Branch Add(string name, Commit commit)
+        {
+            return innerBranchCollection.Add(name, commit);
+        }
+        IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
+        public virtual Branch this[string friendlyName] => innerBranchCollection[friendlyName];
+        public void Update(Branch branch, params Action<BranchUpdater>[] actions)
+        {
+            innerBranchCollection.Update(branch, actions);
+        }
+    }
+}

src/GitVersionCore/Core/GitRepository.cs

@@ -40,7 +40,7 @@ public void Dispose()
 
         public IQueryableCommitLog Commits => repositoryInstance.Commits;
 
-        public BranchCollection Branches => repositoryInstance.Branches;
+        public BranchCollection Branches => (BranchCollection)repositoryInstance.Branches;
 
         public TagCollection Tags => repositoryInstance.Tags;