internetee-registry/test/models/registrant_user_test.rb

require 'test_helper'

class RegistrantUserTest < ActiveSupport::TestCase
  def setup
    super

    @user = users(:registrant)
  end

  def teardown
    super
  end

  def test_ident_helper_method
    assert_equal('1234', @user.ident)
  end

  def test_first_name_from_username
    user = RegistrantUser.new(username: 'John Doe')
    assert_equal 'John', user.first_name
  end

  def test_last_name_from_username
    user = RegistrantUser.new(username: 'John Doe')
    assert_equal 'Doe', user.last_name
  end

  def test_returns_country
    user = RegistrantUser.new(registrant_ident: 'US-1234')
    assert_equal Country.new('US'), user.country
  end

  def test_queries_company_register_for_associated_companies
    assert_equal 'US-1234', @user.registrant_ident

    company_register = Minitest::Mock.new
    company_register.expect(:representation_rights, %w[acme ace], [{ citizen_personal_code: '1234',
                                                                     citizen_country_code: 'USA' }])

    assert_equal %w[acme ace], @user.companies(company_register)
    company_register.verify
  end

  def test_returns_contacts
    Contact.stub(:registrant_user_contacts, %w(john jane)) do
      assert_equal %w(john jane), @user.contacts
    end
  end

  def test_returns_direct_contacts
    Contact.stub(:registrant_user_direct_contacts, %w(john jane)) do
      assert_equal %w(john jane), @user.direct_contacts
    end
  end

  def test_returns_domains
    Domain.stub(:registrant_user_domains, %w(shop airport)) do
      assert_equal %w(shop airport), @user.domains
    end
  end

  def test_returns_administered_domains
    Domain.stub(:registrant_user_administered_domains, %w(shop airport)) do
      assert_equal %w(shop airport), @user.administered_domains
    end
  end
end