require "spec_helper"

module Nyanko
  module Unit
    describe Extender do
      before do
        stub_const("ExampleClass", Class.new)
      end

      it "extends instance methods" do
        Extender.new.expand(:ExampleClass) do
          def test
            "test"
          end
        end
        ExampleClass.new.test.should == "test"
      end

      it "extends class methods" do
        Extender.new.expand(:ExampleClass) do
          class_methods do
            def test
              "test"
            end
          end
        end
        ExampleClass.test.should == "test"
      end

      it "extends instance methods with prefix" do
        Extender.new("__prefix_").expand(:ExampleClass) do
          def test
            "test"
          end
        end
        ExampleClass.new.__prefix_test.should == "test"
      end
    end
  end
end