require 'em_test_helper'

class TestIPv4 < Test::Unit::TestCase
  # Runs a TCP server in the local IPv4 address, connects to it and sends a specific data.
  # Timeout in 2 seconds.
  def test_ipv4_tcp_local_server
    omit_if(!Test::Unit::TestCase.public_ipv4?)

    @@received_data = nil
    @local_port = next_port
    setup_timeout(2)

    EM.run do
      EM::start_server(@@public_ipv4, @local_port) do |s|
        def s.receive_data data
          @@received_data = data
          EM.stop
        end
      end

      EM::connect(@@public_ipv4, @local_port) do |c|
        c.send_data "ipv4/tcp"
      end
    end

    assert_equal "ipv4/tcp", @@received_data
  end

  # Runs a UDP server in the local IPv4 address, connects to it and sends a specific data.
  # Timeout in 2 seconds.
  def test_ipv4_udp_local_server
    omit_if(!Test::Unit::TestCase.public_ipv4?)

    @@received_data = nil
    @local_port = next_port
    setup_timeout(2)

    EM.run do
      EM::open_datagram_socket(@@public_ipv4, @local_port) do |s|
        def s.receive_data data
          @@received_data = data
          EM.stop
        end
      end

      EM::open_datagram_socket(@@public_ipv4, next_port) do |c|
        c.send_datagram "ipv4/udp", @@public_ipv4, @local_port
      end
    end

    assert_equal "ipv4/udp", @@received_data
  end

  # Try to connect via TCP to an invalid IPv4. EM.connect should raise
  # EM::ConnectionError.
  def test_tcp_connect_to_invalid_ipv4
    omit_if(!Test::Unit::TestCase.public_ipv4?)

    invalid_ipv4 = "9.9:9"

    EM.run do
      begin
        error = nil
        EM.connect(invalid_ipv4, 1234)
      rescue => e
        error = e
      ensure
        EM.stop
        assert_equal EM::ConnectionError, (error && error.class)
      end
    end
  end

  # Try to send a UDP datagram to an invalid IPv4. EM.send_datagram should raise
  # EM::ConnectionError.
  def test_udp_send_datagram_to_invalid_ipv4
    omit_if(!Test::Unit::TestCase.public_ipv4?)

    invalid_ipv4 = "9.9:9"

    EM.run do
      begin
        error = nil
        EM.open_datagram_socket(@@public_ipv4, next_port) do |c|
          c.send_datagram "hello", invalid_ipv4, 1234
        end
      rescue => e
        error = e
      ensure
        EM.stop
        assert_equal EM::ConnectionError, (error && error.class)
      end
    end
  end
end