Resolve some unit test issues.

Author: EmmaZhu

src/Common/Storage/Commands.Storage.Test/File/Cmdlet/GetAzureStorageFileTest.cs

@@ -76,33 +76,34 @@ public void GetFileFromSubDirectoryTest_UsingShareObjectWithPath()
         {
             var share = this.MockChannel.GetShareReference(ShareName);
             var dir = share.GetRootDirectoryReference().GetDirectoryReference("dir");
-            var listItems = Enumerable.Range(0, 10).Select(x => dir.GetFileReference(string.Format("file{0}", x))).Cast<IListFileItem>()
-                .Concat(Enumerable.Range(0, 5).Select(x => dir.GetDirectoryReference(string.Format("dir{0}", x)))).ToArray();
 
-            this.ListFilesAndAssertResults(
-                () => this.CmdletInstance.RunCmdlet(
-                    Constants.ShareParameterSetName,
-                    new KeyValuePair<string, object>("Share", share),
-                    new KeyValuePair<string, object>("Path", dir.Name)),
-                listItems,
-                dir);
+            this.MockChannel.SetsAvailableDirectories(dir.Name);
+
+            this.CmdletInstance.RunCmdlet(
+                    Constants.ShareNameParameterSetName,
+                    new KeyValuePair<string, object>("ShareName", ShareName),
+                    new KeyValuePair<string, object>("Path", dir.Name));
+
+            List<IListFileItem> fileList = new List<IListFileItem>();
+            fileList.Add(dir);
+            this.MockCmdRunTime.OutputPipeline.AssertListFileItems(fileList);
         }
 
         [TestMethod]
         public void GetFileFromSubDirectoryTest_UsingShareNameWithPath()
         {
             var share = this.MockChannel.GetShareReference(ShareName);
             var dir = share.GetRootDirectoryReference().GetDirectoryReference("dir");
-            var listItems = Enumerable.Range(0, 10).Select(x => dir.GetFileReference(string.Format("file{0}", x))).Cast<IListFileItem>()
-                .Concat(Enumerable.Range(0, 5).Select(x => dir.GetDirectoryReference(string.Format("dir{0}", x)))).ToArray();
+            this.MockChannel.SetsAvailableDirectories(dir.Name);
 
-            this.ListFilesAndAssertResults(
-                () => this.CmdletInstance.RunCmdlet(
+            this.CmdletInstance.RunCmdlet(
                     Constants.ShareNameParameterSetName,
                     new KeyValuePair<string, object>("ShareName", ShareName),
-                    new KeyValuePair<string, object>("Path", dir.Name)),
-                listItems,
-                dir);
+                    new KeyValuePair<string, object>("Path", dir.Name));
+
+            List<IListFileItem> fileList = new List<IListFileItem>();
+            fileList.Add(dir);
+            this.MockCmdRunTime.OutputPipeline.AssertListFileItems(fileList);
         }
 
         [TestMethod]

src/Common/Storage/Commands.Storage.Test/Service/MockStorageFileManagement.cs

@@ -171,12 +171,26 @@ public void SetSharePermissions(CloudFileShare share, FileSharePermissions permi
 
         public Task FetchFileAttributesAsync(CloudFile file, AccessCondition accessCondition, FileRequestOptions options, OperationContext operationContext, CancellationToken token)
         {
-            throw new NotImplementedException();
+            if (this.availableDirectoryNames.Contains(file.Name))
+            {
+                return TaskEx.FromResult(true);
+            }
+            else
+            {
+                throw new MockupException("FileNotFound");
+            }
         }
 
         public Task FetchDirectoryAttributesAsync(CloudFileDirectory directory, AccessCondition accessCondition, FileRequestOptions options, OperationContext operationContext, CancellationToken cancellationToken)
         {
-            throw new NotImplementedException();
+            if (this.availableDirectoryNames.Contains(directory.Name))
+            {
+                return TaskEx.FromResult(true);
+            }
+            else
+            {
+                throw new MockupException("DirectoryNotFound");
+            }
         }
 
         public Task AbortCopyAsync(CloudFile file, string copyId, AccessCondition accessCondition, FileRequestOptions requestOptions, OperationContext operationContext, CancellationToken cancellationToken)