dns_mock 0.1.0

data/lib/dns_mock/record/factory/ns.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module DnsMock
+  module Record
+    module Factory
+      Ns = ::Class.new(DnsMock::Record::Factory::Base) do
+        record_type :ns
+
+        def instance_params
+          [create_dns_name(record_data)]
+        end
+      end
+    end
+  end
+end

data/lib/dns_mock/record/factory/soa.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module DnsMock
+  module Record
+    module Factory
+      Soa = ::Class.new(DnsMock::Record::Factory::Base) do
+        record_type :soa
+
+        def instance_params
+          record_data[0..1].map(&method(:create_dns_name)) + record_data[2..-1]
+        end
+      end
+    end
+  end
+end

data/lib/dns_mock/record/factory/txt.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module DnsMock
+  module Record
+    module Factory
+      Txt = ::Class.new(DnsMock::Record::Factory::Base) do
+        record_type :txt
+
+        def instance_params
+          record_data
+        end
+      end
+    end
+  end
+end

data/lib/dns_mock/response/answer.rb

@@ -0,0 +1,32 @@
+# frozen_string_literal: true
+
+module DnsMock
+  module Response
+    class Answer
+      TTL = 1
+      REVERSE_TYPE_MAPPER = DnsMock::AVAILABLE_DNS_RECORD_TYPES.each_with_object({}) do |record_type, hash|
+        hash[::Resolv::DNS::Resource::IN.const_get(record_type.upcase)] = record_type
+      end.freeze
+
+      def initialize(records)
+        @records = records
+      end
+
+      def build(hostname, record_class)
+        @hostname = hostname
+        record_by_type(record_class).map { |record| [hostname, DnsMock::Response::Answer::TTL, record] }
+      end
+
+      private
+
+      attr_reader :records, :hostname
+
+      def record_by_type(record_class)
+        record_type = DnsMock::Response::Answer::REVERSE_TYPE_MAPPER[record_class]
+        found_records = records.dig(hostname.to_s, record_type)
+        raise DnsMock::Error::RecordNotFound.new(record_type, hostname) unless found_records
+        found_records
+      end
+    end
+  end
+end

data/lib/dns_mock/response/message.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module DnsMock
+  module Response
+    class Message
+      def initialize(packet, records, dns_answer = DnsMock::Response::Answer, dns_message = ::Resolv::DNS::Message)
+        @dns_answer = dns_answer.new(records)
+        @dns_message = dns_message.decode(packet)
+      end
+
+      def as_binary_string
+        @as_binary_string ||= begin
+          compose_answer
+          dns_message.encode
+        end
+      end
+
+      private
+
+      attr_reader :dns_answer, :dns_message
+
+      def compose_answer
+        dns_message.each_question do |hostname, record_type|
+          dns_message.answer.push(*dns_answer.build(hostname, record_type))
+        end
+      end
+    end
+  end
+end

data/lib/dns_mock/server.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+module DnsMock
+  class Server
+    include DnsMock::Error::Helper
+
+    WARMUP_DELAY = 0.1
+    PACKET_MAX_BYTES_SIZE = 65_535
+
+    attr_reader :port
+
+    def initialize( # rubocop:disable Metrics/ParameterLists
+      socket = ::UDPSocket.new,
+      records_dictionary_builder = DnsMock::Server::RecordsDictionaryBuilder,
+      random_available_port = DnsMock::Server::RandomAvailablePort,
+      thread_class = ::Thread,
+      records: nil,
+      port: nil
+    )
+      @socket = socket
+      @records_dictionary_builder = records_dictionary_builder
+      @thread_class = thread_class
+      @records = records_dictionary_builder.call(records || {})
+      @port = port || random_available_port.call
+      prepare_server_thread
+    end
+
+    def run
+      prepare_socket_for_session
+
+      begin
+        loop do
+          packet, addr = socket.recvfrom(DnsMock::Server::PACKET_MAX_BYTES_SIZE)
+          break if packet.size.zero?
+
+          address, port = addr.values_at(3, 1)
+          socket.send(DnsMock::Response::Message.new(packet, records).as_binary_string, 0, address, port)
+        end
+      ensure
+        socket.close
+      end
+    end
+
+    def assign_mocks(new_records)
+      !!self.records = records_dictionary_builder.call(new_records) if records.empty?
+    end
+
+    def reset_mocks!
+      records.clear.empty?
+    end
+
+    def without_mocks?
+      records.empty?
+    end
+
+    def alive?
+      thread.alive?
+    end
+
+    def stop!
+      thread.kill.stop?
+    end
+
+    private
+
+    attr_reader :socket, :records_dictionary_builder, :thread_class
+    attr_accessor :records, :thread
+
+    def prepare_socket_for_session
+      ::Socket.do_not_reverse_lookup = true
+      socket.setsockopt(::Socket::SOL_SOCKET, ::Socket::SO_REUSEADDR, true)
+      socket.bind(DnsMock::Server::CURRENT_HOST_NAME, port)
+    rescue ::Errno::EADDRINUSE
+      raise DnsMock::Error::PortInUse.new(DnsMock::Server::CURRENT_HOST_NAME, port)
+    end
+
+    def prepare_server_thread
+      self.thread = thread_class.new { run }
+      thread.join(DnsMock::Server::WARMUP_DELAY)
+    end
+  end
+end

data/lib/dns_mock/server/random_available_port.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+module DnsMock
+  class Server
+    CURRENT_HOST_NAME = 'localhost'
+    CURRENT_HOST_ADDRESS = '127.0.0.1'
+
+    class RandomAvailablePort
+      ATTEMPTS = 100
+      MIN_DYNAMIC_PORT_NUMBER = 49_152
+      MAX_DYNAMIC_PORT_NUMBER = 65_535
+
+      class << self
+        def call
+          DnsMock::Server::RandomAvailablePort::ATTEMPTS.times do
+            port = rand(
+              DnsMock::Server::RandomAvailablePort::MIN_DYNAMIC_PORT_NUMBER..DnsMock::Server::RandomAvailablePort::MAX_DYNAMIC_PORT_NUMBER
+            )
+            return port if port_free?(port)
+          end
+          raise DnsMock::Error::RandomFreePort, DnsMock::Server::RandomAvailablePort::ATTEMPTS
+        end
+
+        private
+
+        def udp_port_free?(port, socket = ::UDPSocket.new)
+          socket.bind(DnsMock::Server::CURRENT_HOST_NAME, port)
+          socket.close
+          true
+        rescue ::Errno::EADDRINUSE
+          false
+        end
+
+        def tcp_port_free?(port, socket = ::Socket.new(::Socket::AF_INET, ::Socket::SOCK_STREAM, 0))
+          socket.bind(::Socket.sockaddr_in(port, DnsMock::Server::CURRENT_HOST_ADDRESS))
+          socket.close
+          true
+        rescue ::Errno::EADDRINUSE
+          false
+        end
+
+        def port_free?(port)
+          udp_port_free?(port) && tcp_port_free?(port)
+        end
+      end
+    end
+  end
+end

data/lib/dns_mock/server/records_dictionary_builder.rb

@@ -0,0 +1,52 @@
+# frozen_string_literal: true
+
+module DnsMock
+  class Server
+    class RecordsDictionaryBuilder
+      include DnsMock::Error::Helper
+
+      TYPE_MAPPER = DnsMock::AVAILABLE_DNS_RECORD_TYPES.zip(
+        [
+          [DnsMock::Record::Builder::A, DnsMock::Record::Factory::A, ::Array],
+          [DnsMock::Record::Builder::Aaaa, DnsMock::Record::Factory::Aaaa, ::Array],
+          [DnsMock::Record::Builder::Cname, DnsMock::Record::Factory::Cname, ::String],
+          [DnsMock::Record::Builder::Mx, DnsMock::Record::Factory::Mx, ::Array],
+          [DnsMock::Record::Builder::Ns, DnsMock::Record::Factory::Ns, ::Array],
+          [DnsMock::Record::Builder::Soa, DnsMock::Record::Factory::Soa, ::Array],
+          [DnsMock::Record::Builder::Txt, DnsMock::Record::Factory::Txt, ::Array]
+        ]
+      ).to_h.freeze
+
+      def self.call(records_to_build)
+        new.build(records_to_build).records
+      end
+
+      attr_reader :records
+
+      def initialize
+        @records = {}
+      end
+
+      def build(records_to_build)
+        raise_unless(DnsMock::Error::ArgumentType.new(records_to_build.class), records_to_build.is_a?(::Hash))
+        records_to_build.each do |hostname, dns_records|
+          raise_unless(DnsMock::Error::RecordHostType.new(hostname, hostname.class), hostname.is_a?(::String))
+          records[hostname] = dns_records.each_with_object({}) do |(record_type, records_data), records_instances_by_type|
+            records_instances_by_type[record_type] = build_records_instances_by_type(record_type, records_data)
+          end
+        end
+
+        self
+      end
+
+      private
+
+      def build_records_instances_by_type(record_type, records_to_build)
+        target_builder, target_factory, expected_type = DnsMock::Server::RecordsDictionaryBuilder::TYPE_MAPPER[record_type]
+        raise_record_type_error(record_type, target_builder)
+        raise_record_context_type_error(record_type, records_to_build, expected_type)
+        target_builder.call(target_factory, records_to_build)
+      end
+    end
+  end
+end

data/lib/dns_mock/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module DnsMock
+  VERSION = '0.1.0'
+end

metadata

@@ -0,0 +1,334 @@
+--- !ruby/object:Gem::Specification
+name: dns_mock
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Vladislav Trotsenko
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2021-01-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: bundler-audit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0.1
+- !ruby/object:Gem::Dependency
+  name: faker
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.15'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.15.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.15'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.15.1
+- !ruby/object:Gem::Dependency
+  name: fasterer
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.3
+- !ruby/object:Gem::Dependency
+  name: overcommit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.57.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.57.0
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.9'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 13.0.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 13.0.3
+- !ruby/object:Gem::Dependency
+  name: reek
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.3
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.10'
+- !ruby/object:Gem::Dependency
+  name: rspec-dns
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.8
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.8
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.8.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.8.1
+- !ruby/object:Gem::Dependency
+  name: rubocop-performance
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.9.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.9.2
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.17.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.17.1
+description: Ruby DNS mock
+email:
+- admin@bestweb.com.ua
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".circleci/config.yml"
+- ".codeclimate.yml"
+- ".github/BRANCH_NAMING_CONVENTION.md"
+- ".github/ISSUE_TEMPLATE/bug_report.md"
+- ".github/ISSUE_TEMPLATE/feature_request.md"
+- ".github/ISSUE_TEMPLATE/issue_report.md"
+- ".github/ISSUE_TEMPLATE/question.md"
+- ".github/PULL_REQUEST_TEMPLATE.md"
+- ".gitignore"
+- ".overcommit.yml"
+- ".reek.yml"
+- ".rspec"
+- ".rubocop.yml"
+- ".ruby-gemset"
+- ".ruby-version"
+- CHANGELOG.md
+- CODE_OF_CONDUCT.md
+- CONTRIBUTING.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- dns_mock.gemspec
+- lib/dns_mock.rb
+- lib/dns_mock/core.rb
+- lib/dns_mock/error/argument_type.rb
+- lib/dns_mock/error/helper.rb
+- lib/dns_mock/error/port_in_use.rb
+- lib/dns_mock/error/random_free_port.rb
+- lib/dns_mock/error/record_context.rb
+- lib/dns_mock/error/record_context_type.rb
+- lib/dns_mock/error/record_host_type.rb
+- lib/dns_mock/error/record_not_found.rb
+- lib/dns_mock/error/record_type.rb
+- lib/dns_mock/record/builder/a.rb
+- lib/dns_mock/record/builder/aaaa.rb
+- lib/dns_mock/record/builder/base.rb
+- lib/dns_mock/record/builder/cname.rb
+- lib/dns_mock/record/builder/mx.rb
+- lib/dns_mock/record/builder/ns.rb
+- lib/dns_mock/record/builder/soa.rb
+- lib/dns_mock/record/builder/txt.rb
+- lib/dns_mock/record/factory/a.rb
+- lib/dns_mock/record/factory/aaaa.rb
+- lib/dns_mock/record/factory/base.rb
+- lib/dns_mock/record/factory/cname.rb
+- lib/dns_mock/record/factory/mx.rb
+- lib/dns_mock/record/factory/ns.rb
+- lib/dns_mock/record/factory/soa.rb
+- lib/dns_mock/record/factory/txt.rb
+- lib/dns_mock/response/answer.rb
+- lib/dns_mock/response/message.rb
+- lib/dns_mock/server.rb
+- lib/dns_mock/server/random_available_port.rb
+- lib/dns_mock/server/records_dictionary_builder.rb
+- lib/dns_mock/version.rb
+homepage: https://github.com/mocktools/ruby-dns-mock
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/mocktools/ruby-dns-mock
+  changelog_uri: https://github.com/mocktools/ruby-dns-mock/blob/master/CHANGELOG.md
+  source_code_uri: https://github.com/mocktools/ruby-dns-mock
+  documentation_uri: https://github.com/mocktools/ruby-dns-mock/blob/master/README.md
+  bug_tracker_uri: https://github.com/mocktools/ruby-dns-mock/issues
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.5.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.7.3
+signing_key:
+specification_version: 4
+summary: dns_mock
+test_files: []