require File.expand_path('support/test_helper', __dir__)

class ExtendedSchemaTest < Minitest::Test
  class BitwiseAndAttribute < JSON::Schema::Attribute
    def self.validate(current_schema, data, fragments, processor, _validator, options = {})
      return unless data.is_a?(Integer)

      if data & current_schema.schema['bitwise-and'].to_i == 0
        message = "The property '#{build_fragment(fragments)}' did not evaluate to true when bitwise-AND'd with #{current_schema.schema['bitwise-and']}"
        validation_error(processor, message, fragments, current_schema, self, options[:record_errors])
      end
    end
  end

  class ExtendedSchema < JSON::Schema::Draft3
    def initialize
      super
      @attributes['bitwise-and'] = BitwiseAndAttribute
      @names = ['http://test.com/test.json']
      @uri = Addressable::URI.parse('http://test.com/test.json')
      @names = ['http://test.com/test.json']
    end

    JSON::Validator.register_validator(ExtendedSchema.new)
  end

  def test_extended_schema_validation
    schema = {
      '$schema' => 'http://test.com/test.json',
      'properties' => {
        'a' => {
          'bitwise-and' => 1,
        },
        'b' => {
          'type' => 'string',
        },
      },
    }

    assert_valid schema, { 'a' => 1, 'b' => 'taco' }
    refute_valid schema, { 'a' => 0, 'b' => 'taco' }
    refute_valid schema, { 'a' => 1, 'b' => 5 }
  end

  def test_extended_schema_validation_with_fragment
    schema = {
      '$schema' => 'http://test.com/test.json',
      'definitions' => {
        'odd-a' => {
          'properties' => {
            'a' => {
              'bitwise-and' => 1,
            },
          },
        },
      },
    }

    assert_valid schema, { 'a' => 1 }, fragment: '#/definitions/odd-a'
    refute_valid schema, { 'a' => 0 }, fragment: '#/definitions/odd-a'
  end

  def test_unextended_schema
    # Verify that using the original schema disregards the `bitwise-and` property
    schema = {
      'properties' => {
        'a' => {
          'bitwise-and' => 1,
        },
        'b' => {
          'type' => 'string',
        },
      },
    }

    assert_valid schema, { 'a' => 0, 'b' => 'taco' }
    assert_valid schema, { 'a' => 1, 'b' => 'taco' }
    refute_valid schema, { 'a' => 1, 'b' => 5 }
  end
end
