Register/Unregister for offline I/O in functional tests

microsoft · Aug 5, 2019 · 06f173a · 06f173a
1 parent 8ecfa36
commit 06f173a
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/GVFS/GVFS.FunctionalTests/Tests/EnlistmentPerTestCase/RepairTests.cs b/GVFS/GVFS.FunctionalTests/Tests/EnlistmentPerTestCase/RepairTests.cs
@@ -4,6 +4,7 @@
 using System;
 using System.IO;
 using System.Linq;
+using System.Runtime.InteropServices;
 using System.Text;
 
 namespace GVFS.FunctionalTests.Tests.EnlistmentPerTestCase
@@ -12,6 +13,26 @@ namespace GVFS.FunctionalTests.Tests.EnlistmentPerTestCase
     [Category(Categories.ExtraCoverage)]
     public class RepairTests : TestsWithEnlistmentPerTestCase
     {
+        private const string PrjFSLibPath = "libPrjFSLib.dylib";
+
+        [OneTimeSetUp]
+        public void TurnOfflineIOOn()
+        {
+            if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
+            {
+                 RegisterForOfflineIO();
+            }
+        }
+
+        [OneTimeTearDown]
+        public void TurnOfflineIOOff()
+        {
+            if (RuntimeInformation.IsOSPlatform(OSPlatform.OSX))
+            {
+                UnregisterForOfflineIO();
+            }
+        }
+
         [TestCase]
         public void NoFixesNeeded()
         {
@@ -154,6 +175,12 @@ public void FixesCorruptGitConfig()
             this.Enlistment.MountGVFS();
         }
 
+        [DllImport(PrjFSLibPath, EntryPoint = "PrjFS_RegisterForOfflineIO")]
+        private static extern uint RegisterForOfflineIO();
+
+        [DllImport(PrjFSLibPath, EntryPoint = "PrjFS_UnregisterForOfflineIO")]
+        private static extern uint UnregisterForOfflineIO();
+
         private void CreateCorruptIndexAndRename(string indexPath, Action<FileStream, FileStream> corruptionAction)
         {
             string tempIndexPath = indexPath + ".lock";