# frozen_string_literal: true

require "assert/context_info"
require "assert/macro"
require "assert/suite"
require "assert/test"

module Assert; end

class Assert::Context
  module TestDSL
    def test(desc_or_macro, called_from = nil, first_caller = nil, &block)
      if desc_or_macro.is_a?(Assert::Macro)
        instance_eval(&desc_or_macro)
      elsif block_given?
        # create a test from the given code block
        desc =
          if desc_or_macro.is_a?(Assert::Macro)
            desc_or_macro.name
          else
            desc_or_macro
          end
        suite.on_test(
          Assert::Test.for_block(
            desc,
            Assert::ContextInfo.new(
              self,
              called_from,
              first_caller || caller_locations.first,
            ),
            suite.config,
            &block
          ),
        )
      else
        test_eventually(
          desc_or_macro,
          called_from,
          first_caller || caller_locations.first,
          &block
        )
      end
    end

    def test_eventually(desc_or_macro, called_from = nil, first_caller = nil)
      # create a test from a proc that just skips
      ci =
        Assert::ContextInfo.new(
          self,
          called_from,
          first_caller || caller_locations.first,
        )
      suite.on_test(Assert::Test.for_block(
        desc_or_macro.is_a?(Assert::Macro) ? desc_or_macro.name : desc_or_macro,
        ci,
        suite.config,
        &proc{ skip("TODO", [ci.called_from.to_s]) }
      ))
    end
    alias_method :test_skip, :test_eventually

    def should(desc_or_macro, called_from = nil, first_caller = nil, &block)
      unless desc_or_macro.is_a?(Assert::Macro)
        desc_or_macro = "should #{desc_or_macro}"
      end
      test(
        desc_or_macro,
        called_from,
        first_caller || caller_locations.first,
        &block
      )
    end

    def should_eventually(
          desc_or_macro,
          called_from = nil,
          first_caller = nil,
          &block)
      unless desc_or_macro.is_a?(Assert::Macro)
        desc_or_macro = "should #{desc_or_macro}"
      end
      test_eventually(
        desc_or_macro,
        called_from,
        first_caller || caller_locations.first,
        &block
      )
    end
    alias_method :should_skip, :should_eventually
  end
end