Getting #nlst working

Gemfile

@@ -1,5 +1,5 @@
 source 'http://rubygems.org'
 
 group :development do
-  gem 'rspec', '~> 1.3.0'
+  gem 'rspec', '>= 2.0.0.beta.17'
 end

Gemfile.lock

@@ -1,10 +1,18 @@
 GEM
   remote: http://rubygems.org/
   specs:
-    rspec (1.3.0)
+    diff-lcs (1.1.2)
+    rspec (2.0.0.beta.17)
+      rspec-core (= 2.0.0.beta.17)
+      rspec-expectations (= 2.0.0.beta.17)
+      rspec-mocks (= 2.0.0.beta.17)
+    rspec-core (2.0.0.beta.17)
+    rspec-expectations (2.0.0.beta.17)
+      diff-lcs (>= 1.1.2)
+    rspec-mocks (2.0.0.beta.17)
 
 PLATFORMS
   ruby
 
 DEPENDENCIES
-  rspec (~> 1.3.0)
+  rspec (>= 2.0.0.beta.17)

Rakefile

@@ -5,15 +5,13 @@ task :default => :spec
 #------------------------------------
 
 begin
-  require 'spec/rake/spectask'
+  require 'rspec/core'
+  require 'rspec/core/rake_task'
 rescue LoadError
-  raise 'Run `gem install rspec` to be able to run specs'
+  raise 'Run `gem install rspec --pre` to be able to run specs'
 else
-  desc 'Run specs'
-  Spec::Rake::SpecTask.new do |t|
-    t.spec_files = FileList['spec/*_spec.rb']
-    t.spec_opts  = %w(-fs --color)
-  end
+  desc 'Run the specs'
+  RSpec::Core::RakeTask.new
 end
 
 #------------------------------------

lib/mock_ftp.rb

@@ -1,5 +1,21 @@
+require 'net/ftp'
+require 'mock_ftp/core_ext/string'
+
 module MockFTP
-  def mock_ftp
+  autoload :File,   'mock_ftp/file'
+  autoload :Folder, 'mock_ftp/folder'
+
+  module Net
+    autoload :FTP, 'mock_ftp/net_ftp'
+  end
+
+  def mock_ftp(root_path = '', &block)
+    @original_net_ftp = ::Net.__send__ :remove_const, :FTP
+    ::Net.const_set :FTP, MockFTP::Net::FTP
+
+    MockFTP::Folder.root(root_path, &block)
 
+    ::Net.__send__ :remove_const, :FTP
+    ::Net.const_set :FTP, @original_net_ftp
   end
 end

spec/mock_ftp_spec.rb

@@ -1,5 +1,43 @@
 require 'spec_helper'
 
 describe MockFTP do
-  pending 'YES!'
+  describe '#nlst' do
+    it 'should list mocked files' do
+      mock_ftp do |f|
+        f.file 'file 1'
+        f.file 'file 2'
+
+        open_ftp do |ftp|
+          ftp.nlst.should == [ 'file 1', 'file 2' ]
+        end
+      end
+    end
+
+    context 'within a folder' do
+      it 'should list mocked files with prefixed folder name' do
+        mock_ftp do |f|
+          f.folder 'folder' do |f|
+            f.file 'file 1'
+            f.file 'file 2'
+          end
+
+          open_ftp do |ftp|
+            ftp.nlst('/folder').should == [ '/folder/file 1', '/folder/file 2' ]
+          end
+        end
+      end
+    end
+
+    context 'without a folder' do
+      it 'should raise an error' do
+        mock_ftp do |f|
+          open_ftp do |ftp|
+            expect {
+              ftp.nlst('blah')
+            }.to raise_error(Net::FTPPermError)
+          end
+        end
+      end
+    end
+  end
 end

spec/spec_helper.rb

@@ -6,6 +6,10 @@
 Bundler.require(:default, :development)
 require 'mock_ftp'
 
-Spec::Runner.configure do |config|
-
+RSpec.configure do |config|
+  config.include MockFTP
+end
+
+def open_ftp(&block)
+  ::Net::FTP.open('www.example.com', &block)
 end