require 'spec_helper'
require 'ronin/exploits/metadata/os'

describe Ronin::Exploits::Metadata::OS do
  module TestMetadataOS
    class WithNoOSSet
      include Ronin::Exploits::Metadata::OS
    end

    class WithOSSet
      include Ronin::Exploits::Metadata::OS

      os :linux
    end

    class InheritsItsOS < WithOSSet
      include Ronin::Exploits::Metadata::OS
    end

    class OverridesItsInheritedOS < WithOSSet
      include Ronin::Exploits::Metadata::OS

      os :macos
    end

    class WithNoOSVersionSet
      include Ronin::Exploits::Metadata::OS

      os :linux
    end

    class WithOSVersionSet
      include Ronin::Exploits::Metadata::OS

      os :linux
      os_version '5.x'
    end

    class InheritsItsOSVersion < WithOSVersionSet
      include Ronin::Exploits::Metadata::OS
    end

    class OverridesItsInheritedOSVersion < WithOSVersionSet
      include Ronin::Exploits::Metadata::OS

      os :macos
      os_version '13'
    end
  end

  describe ".os" do
    subject { test_class }

    context "and when os is not set in the class" do
      let(:test_class) { TestMetadataOS::WithNoOSSet }

      it "must default to nil" do
        expect(subject.os).to be(nil)
      end
    end

    context "and when os is set in the class" do
      let(:test_class) { TestMetadataOS::WithOSSet }

      it "must return the set os" do
        expect(subject.os).to eq(:linux)
      end
    end

    context "but when the os was set in the superclass" do
      let(:test_class) { TestMetadataOS::InheritsItsOS }

      it "must return the os set in the superclass" do
        expect(subject.os).to eq(:linux)
      end

      context "but the os is overridden in the sub-class" do
        let(:test_class) { TestMetadataOS::OverridesItsInheritedOS }

        it "must return the os set in the sub-class" do
          expect(subject.os).to eq(:macos)
        end
      end
    end
  end

  describe ".os_version" do
    subject { test_class }

    context "and when os is not set in the class" do
      let(:test_class) { TestMetadataOS::WithNoOSVersionSet }

      it "must default to nil" do
        expect(subject.os_version).to be(nil)
      end
    end

    context "and when os_version is set in the class" do
      let(:test_class) { TestMetadataOS::WithOSVersionSet }

      it "must return the set os_version" do
        expect(subject.os_version).to eq('5.x')
      end
    end

    context "but when the os was set in the superclass" do
      let(:test_class) { TestMetadataOS::InheritsItsOSVersion }

      it "must return the os_version set in the superclass" do
        expect(subject.os_version).to eq('5.x')
      end

      context "but the os is overridden in the sub-class" do
        let(:test_class) do
          TestMetadataOS::OverridesItsInheritedOSVersion
        end

        it "must return the os_version set in the sub-class" do
          expect(subject.os_version).to eq('13')
        end
      end
    end
  end

  describe "#os" do
    subject { test_class.new }

    context "when the Exploit class has .os set" do
      let(:test_class) { TestMetadataOS::WithOSSet }

      it "must return the Exploit classes .os" do
        expect(subject.os).to eq(test_class.os)
      end
    end

    context "when the Exploit class does not have .os set" do
      let(:test_class) { TestMetadataOS::WithNoOSSet }

      it "must return nil" do
        expect(subject.os).to be(nil)
      end
    end
  end

  describe "#os_version" do
    subject { test_class.new }

    context "when the Exploit class has .os_version set" do
      let(:test_class) { TestMetadataOS::WithOSVersionSet }

      it "must return the Exploit classes .os_version" do
        expect(subject.os_version).to eq(test_class.os_version)
      end
    end

    context "when the Exploit class does not have .os_version set" do
      let(:test_class) { TestMetadataOS::WithNoOSVersionSet }

      it "must return nil" do
        expect(subject.os_version).to be(nil)
      end
    end
  end
end