@@ -98,6 +98,10 @@ public void TestEnumerateDirectories()
9898 Directory . CreateDirectory ( $@ "{ path } subdir2\") ;
9999 Directory . CreateDirectory ( $@ "{ path } subdir3\") ;
100100
101+ // Add files to test we don't see files only directories
102+ File . Create ( $@ "{ path } file1.txt") . Close ( ) ;
103+ File . Create ( $@ "{ path } file2.txt") . Close ( ) ;
104+
101105 var directories = Directory . GetDirectories ( path ) ;
102106
103107 Assert . AreEqual ( 3 , directories . Length ) ;
@@ -119,14 +123,27 @@ public void TestEnumerateFiles()
119123
120124 Directory . CreateDirectory ( path ) ;
121125
122- File . Create ( $@ "{ path } file1.txt") . Close ( ) ;
123- File . Create ( $@ "{ path } file2.txt") . Close ( ) ;
124- File . Create ( $@ "{ path } file3.txt") . Close ( ) ;
126+ string file1 = $@ "{ path } file1.txt";
127+ string file2 = $@ "{ path } file2.txt";
128+ string file3 = $@ "{ path } file3.txt";
129+
130+ // Add a mix of directories and files
131+ Directory . CreateDirectory ( $@ "{ path } TestDir1") ;
132+ Directory . CreateDirectory ( $@ "{ path } TestDir2") ;
133+
134+ File . Create ( file1 ) . Close ( ) ;
135+ File . Create ( file2 ) . Close ( ) ;
136+ File . Create ( file3 ) . Close ( ) ;
125137
126138 var files = Directory . GetFiles ( path ) ;
127139
128140 Assert . AreEqual ( 3 , files . Length ) ;
129141
142+ // Check correct file names returned
143+ Assert . IsTrue ( ( files [ 0 ] == file1 ) , "Invalid file1 in GetFiles()" ) ;
144+ Assert . IsTrue ( ( files [ 1 ] == file2 ) , "Invalid file2 in GetFiles()" ) ;
145+ Assert . IsTrue ( ( files [ 2 ] == file3 ) , "Invalid file3 in GetFiles()" ) ;
146+
130147 // Clean up after the test
131148 Directory . Delete ( path , true ) ;
132149 }
0 commit comments