common libs go into the lib/common directory

spec/lib/common/cache_file_store_spec.rb

@@ -0,0 +1,93 @@
+# encoding: UTF-8
+#--
+# WPScan - WordPress Security Scanner
+# Copyright (C) 2012-2013
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#++
+
+require 'spec_helper'
+
+describe CacheFileStore do
+
+  before :all do
+    @cache_dir = SPEC_CACHE_DIR + '/cache_file_store'
+  end
+
+  before :each do
+    Dir.delete(@cache_dir) rescue nil
+
+    @cache = CacheFileStore.new(@cache_dir)
+  end
+
+  after :each do
+    @cache.clean
+  end
+
+  describe '#storage_path' do
+    it 'returns the storage path given in the #new' do
+      @cache.storage_path.should == @cache_dir
+    end
+  end
+
+  describe '#serializer' do
+    it 'should return the default serializer : Marshal' do
+      @cache.serializer.should == Marshal
+      @cache.serializer.should_not == YAML
+    end
+  end
+
+  describe '#clean' do
+    it "should remove all files from the cache dir (#{@cache_dir}" do
+      # let's create some files into the directory first
+      (0..5).each do |i|
+        File.new(@cache_dir + "/file_#{i}.txt", File::CREAT)
+      end
+
+      count_files_in_dir(@cache_dir, 'file_*.txt').should == 6
+      @cache.clean
+      count_files_in_dir(@cache_dir).should == 0
+    end
+  end
+
+  describe '#read_entry (nonexistent entry)' do
+    it 'should return nil' do
+      @cache.read_entry(Digest::SHA1.hexdigest('hello world')).should be_nil
+    end
+  end
+
+  describe '#write_entry, #read_entry' do
+
+    after :each do
+      @cache.write_entry(@key, @data, @timeout)
+      @cache.read_entry(@key).should === @expected
+    end
+
+    it 'should get the correct entry (string)' do
+      @timeout = 10
+      @key = 'some_key'
+      @data = 'Hello World !'
+      @expected = @data
+    end
+
+    it 'should not write the entry' do
+      @timeout = 0
+      @key = 'another_key'
+      @data = 'Another Hello World !'
+      @expected = nil
+    end
+
+    ## TODO write / read for an object
+  end
+end