internetee-registry/test/models/white_ip_test.rb

require 'test_helper'

class WhiteIpTest < ActiveSupport::TestCase
  def test_either_ipv4_or_ipv6_is_required
    white_ip = valid_white_ip

    white_ip.ipv4 = ''
    white_ip.ipv6 = ''
    assert white_ip.invalid?
    assert_includes white_ip.errors.full_messages, 'IPv4 or IPv6 must be present'

    white_ip.ipv4 = valid_ipv4
    white_ip.ipv6 = ''
    assert white_ip.valid?

    white_ip.ipv4 = ''
    white_ip.ipv6 = valid_ipv6
    assert white_ip.valid?
  end

  def test_validates_ipv4_format
    white_ip = valid_white_ip

    white_ip.ipv4 = 'invalid'
    assert white_ip.invalid?

    white_ip.ipv4 = valid_ipv4
    assert white_ip.valid?
  end

  def test_validates_ipv6_format
    white_ip = valid_white_ip

    white_ip.ipv6 = 'invalid'
    assert white_ip.invalid?

    white_ip.ipv6 = valid_ipv6
    assert white_ip.valid?
  end

  private

  def valid_white_ip
    white_ips(:one)
  end

  def valid_ipv4
    '192.0.2.1'
  end

  def valid_ipv6
    '2001:db8::1'
  end
end