# frozen_string_literal: true
# encoding: utf-8

module Sdam
  class Verifier
    include RSpec::Matchers

    def verify_sdam_event(expected_events, actual_events, i)
      expect(expected_events.length).to be > i
      expect(actual_events.length).to be > i

      expected_event = expected_events[i]
      actual_event = actual_events[i]

      actual_event_name = Utils.underscore(actual_event.class.name.sub(/.*::/, ''))
      actual_event_name = actual_event_name.to_s.sub('topology_changed', 'topology_description_changed') + '_event'
      expect(actual_event_name).to eq(expected_event.name)

      send("verify_#{expected_event.name}", expected_event, actual_event)
    end

    def verify_topology_opening_event(expected, actual)
      expect(actual.topology).not_to be nil
    end

    def verify_topology_description_changed_event(expected, actual)
      verify_topology_matches(expected.data['previousDescription'], actual.previous_topology)
      verify_topology_matches(expected.data['newDescription'], actual.new_topology)
    end

    def verify_topology_matches(expected, actual)
      expected_type = ::Mongo::Cluster::Topology.const_get(expected['topologyType'])
      expect(actual).to be_a(expected_type)

      expect(actual.replica_set_name).to eq(expected['setName'])

      expected['servers'].each do |server|
        desc = actual.server_descriptions[server['address'].to_s]
        expect(desc).not_to be nil
        verify_description_matches(server, desc)
      end

      # Verify actual topology has no servers not also present in the
      # expected topology description.
      expected_addresses = expected['servers'].map do |server|
        server['address']
      end
      actual.server_descriptions.keys.each do |address_str|
        expect(expected_addresses).to include(address_str)
      end
    end

    def verify_server_opening_event(expected, actual)
      expect(actual.address.to_s).to eq(expected.data['address'])
    end

    def verify_server_description_changed_event(expected, actual)
      verify_description_matches(expected.data['previousDescription'], actual.previous_description)
      verify_description_matches(expected.data['newDescription'], actual.new_description)
    end

    def verify_description_matches(server_spec, actual)
      case server_spec['type']
      when 'Standalone'
        expect(actual).to be_standalone
      when 'RSPrimary'
        expect(actual).to be_primary
      when 'RSSecondary'
        expect(actual).to be_secondary
      when 'RSArbiter'
        expect(actual).to be_arbiter
      when 'Mongos'
        expect(actual).to be_mongos
      when 'Unknown', 'PossiblePrimary'
        expect(actual).to be_unknown
      when 'RSGhost'
        expect(actual).to be_ghost
      when 'RSOther'
        expect(actual).to be_other
      end

      if server_spec['arbiters']
        expect(actual.arbiters).to eq(server_spec['arbiters'])
      end
      if server_spec['hosts']
        expect(actual.hosts).to eq(server_spec['hosts'])
      end
      if server_spec['passives']
        expect(actual.passives).to eq(server_spec['passives'])
      end
      if server_spec['primary']
        expect(actual.primary_host).to eq(server_spec['primary'])
      end
      expect(actual.replica_set_name).to eq(server_spec['setName'])

      if server_spec['topologyVersion']
        # In the Ruby TopologyVersion object, the counter is a
        # Ruby integer. It would serialize to BSON int.
        # The expected topology version specifies counter as a
        # BSON long.
        # Parse expected value as extended json and compare
        # Ruby objects.
        expected_tv = server_spec['topologyVersion']
        expect(actual.topology_version).to eq(expected_tv)
      end
    end

    def verify_server_closed_event(expected, actual)
      expect(actual.address.to_s).to eq(expected.data['address'])
    end
  end
end
