HELLO v3!!!

spec/app/models/theme_spec.rb

@@ -0,0 +1,165 @@
+require 'spec_helper'
+
+describe WPScan::Theme do
+  subject(:theme)  { described_class.new(slug, blog, opts) }
+  let(:slug)       { 'spec' }
+  let(:blog)       { WPScan::Target.new('http://wp.lab/') }
+  let(:opts)       { {} }
+  let(:fixtures)   { File.join(FIXTURES, 'models', 'theme') }
+
+  before { expect(blog).to receive(:content_dir).at_least(1).and_return('wp-content') }
+
+  describe '#new' do
+    before do
+      stub_request(:get, /.*\.css\z/)
+        .to_return(body: File.read(File.join(fixtures, 'style.css')))
+    end
+
+    its(:url) { should eql 'http://wp.lab/wp-content/themes/spec/' }
+    its(:style_url) { should eql 'http://wp.lab/wp-content/themes/spec/style.css' }
+
+    its(:style_name)  { should eql 'Twenty Fifteen' }
+    its(:style_uri)   { should eql 'https://wordpress.org/themes/twentyfifteen' }
+    its(:author)      { should eql 'the WordPress team' }
+    its(:author_uri)  { should eql nil }
+    its(:template)    { should eql nil }
+    its(:description) { should eql 'Our 2015 default theme is clean, blog-focused.' }
+    its(:license)     { should eql 'GNU General Public License v2 or later' }
+    its(:license_uri) { should eql 'http://www.gnu.org/licenses/gpl-2.0.html' }
+    its(:tags)        { should eql 'black, blue, gray, pink, purple, white, yellow.' }
+    its(:text_domain) { should eql 'twentyfifteen' }
+
+    context 'when opts[:style_url]' do
+      let(:opts) { super().merge(style_url: 'http://wp.lab/wp-content/themes/spec/custom.css') }
+
+      its(:style_url) { should eql opts[:style_url] }
+    end
+  end
+
+  describe '#version' do
+    after do
+      stub_request(:get, /.*\.css\z/)
+        .to_return(body: File.read(File.join(fixtures, 'style.css')))
+
+      expect(WPScan::Finders::ThemeVersion::Base).to receive(:find).with(theme, @expected_opts)
+      theme.version(version_opts)
+    end
+
+    let(:default_opts) { {} }
+
+    context 'when no :detection_mode' do
+      context 'when no :mode opt supplied' do
+        let(:version_opts) { { something: 'k' } }
+
+        it 'calls the finder with the correct parameters' do
+          @expected_opts = version_opts
+        end
+      end
+
+      context 'when :mode supplied' do
+        let(:version_opts) { { mode: :passive } }
+
+        it 'calls the finder with the correct parameters' do
+          @expected_opts = default_opts.merge(mode: :passive)
+        end
+      end
+    end
+
+    context 'when :detection_mode' do
+      let(:opts) { super().merge(mode: :passive) }
+
+      context 'when no :mode' do
+        let(:version_opts) { {} }
+
+        it 'calls the finder without mode' do
+          @expected_opts = version_opts
+        end
+      end
+
+      context 'when :mode' do
+        let(:version_opts) { { mode: :mixed } }
+
+        it 'calls the finder with the :mixed mode' do
+          @expected_opts = default_opts.merge(mode: :mixed)
+        end
+      end
+    end
+  end
+
+  describe '#vulnerabilities' do
+    xit
+  end
+
+  describe '#parent_theme' do
+    before do
+      stub_request(:get, blog.url('wp-content/themes/spec/style.css'))
+        .to_return(body: File.read(File.join(fixtures, main_theme)))
+    end
+
+    context 'when no template' do
+      let(:main_theme) { 'style.css' }
+
+      it 'returns nil' do
+        expect(theme.parent_theme).to eql nil
+      end
+    end
+
+    context 'when a template' do
+      let(:main_theme) { 'child_style.css' }
+      let(:parent_url) { blog.url('wp-content/themes/twentyfourteen/custom.css') }
+
+      before do
+        stub_request(:get, parent_url)
+          .to_return(body: File.read(File.join(fixtures, 'style.css')))
+      end
+
+      %w[child_style windows_line_endings].each do |fixture|
+        context "when #{fixture}" do
+          let(:main_theme) { "#{fixture}.css" }
+
+          it 'returns the expected theme' do
+            parent = theme.parent_theme
+
+            expect(parent).to eql described_class.new(
+              'twentyfourteen', blog,
+              style_url: parent_url,
+              confidence: 100,
+              found_by: 'Parent Themes (Passive Detection)'
+            )
+            expect(parent.style_url).to eql parent_url
+          end
+        end
+      end
+    end
+  end
+
+  describe '#parent_themes' do
+    xit
+  end
+
+  describe '#==' do
+    before { stub_request(:get, /.*\.css\z/) }
+
+    context 'when default style' do
+      it 'returns true when equal' do
+        expect(theme == described_class.new(slug, blog, opts)).to be true
+      end
+
+      it 'returns false when not equal' do
+        expect(theme == described_class.new(slug, blog, opts.merge(style_url: 'spec.css'))).to be false
+      end
+    end
+
+    context 'when custom style' do
+      let(:opts) { super().merge(style_url: 'spec.css') }
+
+      it 'returns true when equal' do
+        expect(theme == described_class.new(slug, blog, opts.merge(style_url: 'spec.css'))).to be true
+      end
+
+      it 'returns false when not equal' do
+        expect(theme == described_class.new(slug, blog, opts.merge(style_url: 'spec2.css'))).to be false
+      end
+    end
+  end
+end