new tests

basic globbing support
geeklearningio · Jul 5, 2016 · 690bc1c · 690bc1c
1 parent ca7481a
commit 690bc1c
Show file tree

Hide file tree

Showing 9 changed files with 103 additions and 17 deletions.
diff --git a/src/GeekLearning.Storage.Azure/AzureStore.cs b/src/GeekLearning.Storage.Azure/AzureStore.cs
@@ -147,10 +147,23 @@ public async Task<IFileReference[]> ListAsync(string path, bool recursive)
 
         public async Task<IFileReference[]> ListAsync(string path, string searchPattern, bool recursive)
         {
+            if (string.IsNullOrWhiteSpace(path))
+            {
+                path = null;
+            }
+            else
+            {
+                if (!path.EndsWith("/"))
+                {
+                    path = path + "/";
+                }
+            }
+
+            string prefix = path;
             var firstWildCard = searchPattern.IndexOf('*');
             if (firstWildCard >= 0)
             {
-                path += searchPattern.Substring(0, firstWildCard);
+                prefix += searchPattern.Substring(0, firstWildCard);
                 searchPattern = searchPattern.Substring(firstWildCard);
             }
 
@@ -162,19 +175,19 @@ public async Task<IFileReference[]> ListAsync(string path, string searchPattern,
             List<IListBlobItem> results = new List<IListBlobItem>();
             do
             {
-                var response = await this.container.Value.ListBlobsSegmentedAsync(path, recursive, BlobListingDetails.None, null, continuationToken, new BlobRequestOptions(), new OperationContext());
+                var response = await this.container.Value.ListBlobsSegmentedAsync(prefix, recursive, BlobListingDetails.None, null, continuationToken, new BlobRequestOptions(), new OperationContext());
                 continuationToken = response.ContinuationToken;
                 results.AddRange(response.Results);
             }
             while (continuationToken != null);
 
-            var pathMap = results.Select(blob => new Internal.AzureFileReference(blob)).ToDictionary(x => x.Path);
-            throw new NotSupportedException();
-            //var filteredResults = matcher.Execute(
-            //    new Internal.AzureListDirectoryWrapper(path,
-            //    pathMap.Values));
+            var pathMap = results.OfType<ICloudBlob>().Select(blob => new Internal.AzureFileReference(blob)).ToDictionary(x => x.Path);
+
+            var filteredResults = matcher.Execute(
+                new Internal.AzureListDirectoryWrapper(path,
+                pathMap));
 
-            //return filteredResults.Files.Select(x => pathMap[x.Path]).ToArray();
+            return filteredResults.Files.Select(x => pathMap[path + x.Path]).ToArray();
         }
 
         public async Task DeleteAsync(IPrivateFileReference file)

diff --git a/src/GeekLearning.Storage.Azure/Internal/AzureListDirectoryWrapper.cs b/src/GeekLearning.Storage.Azure/Internal/AzureListDirectoryWrapper.cs
@@ -9,14 +9,33 @@ namespace GeekLearning.Storage.Azure.Internal
 {
     public class AzureListDirectoryWrapper : DirectoryInfoBase
     {
+        private string name;
         private string fullName;
+        private string path;
+        private Dictionary<string, AzureFileReference> files;
 
         public AzureListDirectoryWrapper(FileSystemInfoBase childrens)
         {
             this.fullName = "root";
             this.ParentDirectory = null;
         }
 
+        public AzureListDirectoryWrapper(string path, Dictionary<string, AzureFileReference> files)
+        {
+            this.path = path;
+            this.files = files;
+            this.fullName = path;
+            var lastSlash = path.LastIndexOf('/');
+            if (lastSlash >= 0)
+            {
+                this.name = path.Substring(lastSlash + 1);
+            }
+            else
+            {
+                this.name = path;
+            }
+        }
+
         public AzureListDirectoryWrapper(CloudBlobDirectory blobDirectory, AzureListDirectoryWrapper parent = null)
         {
             this.ParentDirectory = parent;
@@ -35,7 +54,7 @@ public override string Name
         {
             get
             {
-                return fullName;
+                return name;
             }
         }
 
@@ -46,7 +65,7 @@ public override DirectoryInfoBase ParentDirectory
 
         public override IEnumerable<FileSystemInfoBase> EnumerateFileSystemInfos()
         {
-            throw new NotImplementedException();
+            return this.files.Values.Select(file => new AzureListFileWrapper(file.CloudBlob, this));
         }
 
         public override DirectoryInfoBase GetDirectory(string path)
@@ -56,7 +75,7 @@ public override DirectoryInfoBase GetDirectory(string path)
 
         public override FileInfoBase GetFile(string path)
         {
-            throw new NotImplementedException();
+            return new AzureListFileWrapper(this.files[path].CloudBlob, this);
         }
     }
 }
diff --git a/src/GeekLearning.Storage.Azure/Internal/AzureListFileWrapper.cs b/src/GeekLearning.Storage.Azure/Internal/AzureListFileWrapper.cs
@@ -9,11 +9,23 @@ namespace GeekLearning.Storage.Azure.Internal
 {
     public class AzureListFileWrapper : FileInfoBase
     {
-        private CloudBlob blob;
+        private ICloudBlob blob;
+        private string name;
+        private AzureListDirectoryWrapper parent;
 
-        public AzureListFileWrapper(CloudBlob blob)
+        public AzureListFileWrapper(ICloudBlob blob, AzureListDirectoryWrapper parent)
         {
             this.blob = blob;
+            var lastSlash = blob.Name.LastIndexOf('/');
+            if (lastSlash >= 0)
+            {
+                this.name = blob.Name.Substring(lastSlash + 1);
+            }
+            else
+            {
+                this.name = blob.Name;
+            }
+            this.parent = parent;
         }
 
         public override string FullName
@@ -28,15 +40,15 @@ public override string Name
         {
             get
             {
-                return this.blob.Name;
+                return name;
             }
         }
 
         public override DirectoryInfoBase ParentDirectory
         {
             get
             {
-                throw new NotImplementedException();
+                return this.parent;
             }
         }
     }

diff --git a/src/GeekLearning.Storage.FileSystem/FileSystemStore.cs b/src/GeekLearning.Storage.FileSystem/FileSystemStore.cs
@@ -73,10 +73,10 @@ public Task<IFileReference[]> ListAsync(string path, string searchPattern, bool
             Microsoft.Extensions.FileSystemGlobbing.Matcher matcher = new Microsoft.Extensions.FileSystemGlobbing.Matcher(StringComparison.Ordinal);
             matcher.AddInclude(searchPattern);
 
-            var results = matcher.Execute(new Microsoft.Extensions.FileSystemGlobbing.Abstractions.DirectoryInfoWrapper(new DirectoryInfo(path)));
+            var results = matcher.Execute(new Microsoft.Extensions.FileSystemGlobbing.Abstractions.DirectoryInfoWrapper(new DirectoryInfo(directoryPath)));
 
             return Task.FromResult(results.Files
-                .Select(match => (IFileReference)new Internal.FileSystemFileReference(match.Path, match.Path.Replace(this.absolutePath, "").Trim('/', '\\')))
+                .Select(match => (IFileReference)new Internal.FileSystemFileReference(Path.Combine(directoryPath, match.Path), Path.Combine(path, match.Path).Trim('/', '\\')))
                 .ToArray());
         }
 

diff --git a/tests/GeekLearning.Integration.Test/ListTests.cs b/tests/GeekLearning.Integration.Test/ListTests.cs
@@ -94,5 +94,43 @@ public async Task ListSubDirectoryFilesWithTrailingSlash(string storeName)
             Assert.Empty(missingFiles);
             Assert.Empty(unexpectedFiles);
         }
+
+        [Theory(DisplayName = nameof(ExtensionGlobbing)), InlineData("azure"), InlineData("filesystem")]
+        public async Task ExtensionGlobbing(string storeName)
+        {
+            var storageFactory = this.storeFixture.Services.GetRequiredService<IStorageFactory>();
+
+            var store = storageFactory.GetStore(storeName);
+
+            var expected = new string[] { "Globbing/template.hbs", "Globbing/template-header.hbs" };
+
+            var results = await store.ListAsync("Globbing", "*.hbs");
+
+            var missingFiles = expected.Except(results.Select(f => f.Path)).ToArray();
+
+            var unexpectedFiles = results.Select(f => f.Path).Except(expected).ToArray();
+
+            Assert.Empty(missingFiles);
+            Assert.Empty(unexpectedFiles);
+        }
+
+        [Theory(DisplayName = nameof(FileNameGlobbing)), InlineData("azure"), InlineData("filesystem")]
+        public async Task FileNameGlobbing(string storeName)
+        {
+            var storageFactory = this.storeFixture.Services.GetRequiredService<IStorageFactory>();
+
+            var store = storageFactory.GetStore(storeName);
+
+            var expected = new string[] { "Globbing/template.hbs", "Globbing/template.mustache" };
+
+            var results = await store.ListAsync("Globbing", "template.*");
+
+            var missingFiles = expected.Except(results.Select(f => f.Path)).ToArray();
+
+            var unexpectedFiles = results.Select(f => f.Path).Except(expected).ToArray();
+
+            Assert.Empty(missingFiles);
+            Assert.Empty(unexpectedFiles);
+        }
     }
 }
diff --git a/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template-header.hbs b/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template-header.hbs
@@ -0,0 +1 @@
+template.header.hbs
diff --git a/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template-header.mustache b/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template-header.mustache
@@ -0,0 +1 @@
+template.header.mustache
diff --git a/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template.hbs b/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template.hbs
@@ -0,0 +1 @@
+template.hbs
diff --git a/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template.mustache b/tests/GeekLearning.Integration.Test/SampleDirectory/Globbing/template.mustache
@@ -0,0 +1 @@
+template.mustache