itsi 0.1.7 → 0.1.9

data/gems/server/test/test_itsi_server.rb

@@ -1,9 +1,281 @@
-# frozen_string_literal: true
-
-require_relative "test_helper"
+require "socket"
+require "net/http"
+require "minitest/autorun"
 
 class TestItsiServer < Minitest::Test
   def test_that_it_has_a_version_number
     refute_nil ::Itsi::Server::VERSION
   end
+
+  def test_hello_world
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain" }, ["Hello, World!"]]
+    end) do |uri|
+      assert_equal "Hello, World!", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_post
+    run_app(lambda do |env|
+      assert_equal env["REQUEST_METHOD"], "POST"
+      assert_equal "data", env["rack.input"].read
+      [200, { "Content-Type" => "text/plain" }, ["Hello, World!"]]
+    end) do |uri|
+      assert_equal "Hello, World!", Net::HTTP.post(uri, "data").body
+    end
+  end
+
+  def test_full_hijack
+    run_app(lambda do |env|
+      io = env["rack.hijack"].call
+      io.write("HTTP/1.1 200 Ok\r\n")
+      io.write("Content-Type: text/plain\r\n")
+      io.write("Transfer-Encoding: chunked\r\n")
+      io.write("\r\n")
+      io.write("7\r\n")
+      io.write("Hello, \r\n")
+      io.write("6\r\n")
+      io.write("World!\r\n")
+      io.write("0\r\n\r\n")
+      io.close
+    end) do |uri|
+      assert_equal "Hello, World!", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_streaming_body
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain" }, lambda { |stream|
+        stream.write("Hello")
+        stream.write(", World!")
+        stream.close
+      }]
+    end) do |uri|
+      assert_equal "Hello, World!", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_partial_hijack
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain", "rack.hijack" => lambda { |stream|
+        stream.write("Hello")
+        stream.write(", World!")
+        stream.close
+      } }, []]
+    end) do |uri|
+      assert_equal "Hello, World!", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_enumerable_body
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "application/json" },
+       %W[one\n two\n three\n]]
+    end) do |uri|
+      assert_equal "one\ntwo\nthree\n", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_scheduler_non_blocking
+    run_app(
+      lambda do |env|
+        sleep 0.25
+        [200, { "Content-Type" => "text/plain" }, "Hello, World!"]
+      end,
+      scheduler_class: "Itsi::Scheduler"
+    ) do |uri|
+      start_time = Time.now
+      20.times.map do
+        Thread.new do
+          assert_equal "Hello, World!", Net::HTTP.get(uri)
+        end
+      end.each(&:join)
+      assert_in_delta 0.25, Time.now - start_time, 0.5
+    end
+  end
+
+  def test_query_params
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain" }, [env["QUERY_STRING"]]]
+    end) do |uri|
+      uri.query = "foo=bar&baz=qux"
+      assert_equal "foo=bar&baz=qux", Net::HTTP.get(uri)
+    end
+  end
+
+  def test_put_request
+    run_app(lambda do |env|
+      body = env["rack.input"].read
+      [200, { "Content-Type" => "text/plain" }, [body]]
+    end) do |uri|
+      uri_obj = URI(uri)
+      req = Net::HTTP::Put.new(uri_obj)
+      req.body = "put data"
+      response = Net::HTTP.start(uri_obj.hostname, uri_obj.port) { |http| http.request(req) }
+      assert_equal "put data", response.body
+    end
+  end
+
+  def test_custom_headers
+    run_app(lambda do |env|
+      header = env["HTTP_X_CUSTOM"] || ""
+      [200, { "Content-Type" => "text/plain" }, [header]]
+    end) do |uri|
+      uri_obj = URI(uri)
+      req = Net::HTTP::Get.new(uri_obj)
+      req["X-Custom"] = "custom-value"
+      response = Net::HTTP.start(uri_obj.hostname, uri_obj.port) { |http| http.request(req) }
+      assert_equal "custom-value", response.body
+    end
+  end
+
+  def test_error_response
+    response = nil
+    capture_subprocess_io do
+      run_app(lambda do |env|
+        raise "Intentional error for testing"
+      end) do |uri|
+        response = Net::HTTP.get_response(uri)
+      end
+    end
+    assert_equal "500", response.code
+  end
+
+  def test_redirect
+    run_app(lambda do |env|
+      [302, { "Location" => "http://example.com" }, []]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "302", response.code
+      assert_equal "http://example.com", response["location"]
+    end
+  end
+
+  def test_not_found
+    run_app(lambda do |env|
+      if env["PATH_INFO"] == "/"
+        [200, { "Content-Type" => "text/plain" }, ["Home"]]
+      else
+        [404, { "Content-Type" => "text/plain" }, ["Not Found"]]
+      end
+    end) do |uri|
+      uri.path = "/nonexistent"
+      response = Net::HTTP.get_response(uri)
+      assert_equal "404", response.code
+      assert_equal "Not Found", response.body
+    end
+  end
+
+  def test_head_request
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain", "Content-Length" => "13" }, ["Hello, World!"]]
+    end) do |uri|
+      uri_obj = URI(uri)
+      response = Net::HTTP.start(uri_obj.hostname, uri_obj.port) do |http|
+        http.head("/")
+      end
+      assert_equal "200", response.code
+      assert_empty response.body.to_s
+      assert_equal "13", response["content-length"]
+    end
+  end
+
+  def test_options_request
+    run_app(lambda do |env|
+      [200, { "Allow" => "GET,POST,OPTIONS", "Content-Type" => "text/plain" }, ["Options Response"]]
+    end) do |uri|
+      uri_obj = URI(uri)
+      req = Net::HTTP::Options.new(uri_obj)
+      response = Net::HTTP.start(uri_obj.hostname, uri_obj.port) { |http| http.request(req) }
+      assert_equal "200", response.code
+      assert_equal "GET,POST,OPTIONS", response["allow"]
+      assert_equal "Options Response", response.body
+    end
+  end
+
+  def test_cookie_handling
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain", "Set-Cookie" => "session=abc123; Path=/" }, ["Cookie Test"]]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "200", response.code
+      assert_match(/session=abc123/, response["set-cookie"])
+      assert_equal "Cookie Test", response.body
+    end
+  end
+
+  def test_multiple_headers
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain", "X-Example" => "one, two, three" }, ["Multiple Headers"]]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "200", response.code
+      assert_equal "one, two, three", response["x-example"]
+      assert_equal "Multiple Headers", response.body
+    end
+  end
+
+  def test_large_body
+    large_text = "A" * 10_000
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain", "Content-Length" => large_text.bytesize.to_s }, [large_text]]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "200", response.code
+      assert_equal large_text, response.body
+    end
+  end
+
+  def test_custom_status_code
+    run_app(lambda do |env|
+      [201, { "Content-Type" => "text/plain" }, ["Created"]]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "201", response.code
+      assert_equal "Created", response.body
+    end
+  end
+
+  def test_empty_body
+    run_app(lambda do |env|
+      [204, { "Content-Type" => "text/plain" }, []]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "204", response.code
+      assert_nil response.body
+    end
+  end
+
+  def test_utf8_response
+    utf8_text = "こんにちは世界"
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain; charset=utf-8" }, [utf8_text]]
+    end) do |uri|
+      response = Net::HTTP.get_response(uri)
+      assert_equal "200", response.code
+      assert_equal utf8_text, response.body.force_encoding("UTF-8")
+    end
+  end
+
+  def test_custom_request_header
+    run_app(lambda do |env|
+      header_value = env["HTTP_X_MY_HEADER"] || ""
+      [200, { "Content-Type" => "text/plain" }, [header_value]]
+    end) do |uri|
+      uri_obj = URI(uri)
+      req = Net::HTTP::Get.new(uri_obj)
+      req["X-My-Header"] = "test-header"
+      response = Net::HTTP.start(uri_obj.hostname, uri_obj.port) { |http| http.request(req) }
+      assert_equal "test-header", response.body
+    end
+  end
+
+  def test_url_encoded_query_params
+    run_app(lambda do |env|
+      [200, { "Content-Type" => "text/plain" }, [env["QUERY_STRING"]]]
+    end) do |uri|
+      uri.query = "param=%C3%A9" # %C3%A9 represents 'é'
+      assert_equal "param=%C3%A9", Net::HTTP.get(uri)
+    end
+  end
 end

data/lib/itsi/version.rb

@@ -1,3 +1,3 @@
 module Itsi
-  VERSION = "0.1.7"
+  VERSION = '0.1.9'
 end

data/sandbox/deploy/main.tf

@@ -0,0 +1,238 @@
+terraform {
+  required_providers {
+    aws = {
+      source  = "hashicorp/aws"
+      version = "~> 5.0"
+    }
+    cloudflare = {
+      source  = "cloudflare/cloudflare"
+      version = "~> 5.0.0"
+    }
+  }
+
+  required_version = ">= 0.14.9"
+}
+
+provider "aws" {
+  region = var.aws_region
+}
+
+
+data "aws_region" "current" {
+  name = var.aws_region
+}
+
+data "aws_ami" "ubuntu" {
+  most_recent = true
+
+  filter {
+    name   = "name"
+    values = ["ubuntu/images/hvm-ssd-gp3/*ubuntu-noble-24.04-arm64-server-*"]
+  }
+
+  filter {
+    name   = "virtualization-type"
+    values = ["hvm"]
+  }
+
+  owners = ["099720109477"] # Canonical
+}
+
+# Create VPC
+# terraform aws create vpc
+resource "aws_vpc" "vpc" {
+  cidr_block           = "10.0.0.0/16"
+  instance_tenancy     = "default"
+  enable_dns_hostnames = true
+  tags = {
+    Name = "${var.environment}-vpc"
+  }
+}
+resource "aws_subnet" "public_subnet_a" {
+  vpc_id                  = aws_vpc.vpc.id
+  cidr_block              = "10.0.1.0/24"
+  availability_zone       = var.subnet_az
+  map_public_ip_on_launch = true
+
+  tags = {
+    Name = "${var.environment} Public Subnet A"
+  }
+}
+
+resource "aws_subnet" "private_subnet_a" {
+  vpc_id            = aws_vpc.vpc.id
+  cidr_block        = "10.0.2.0/24"
+  availability_zone = var.subnet_az
+
+  tags = {
+    Name = "${var.environment} Private Subnet A"
+  }
+}
+
+resource "aws_internet_gateway" "ig_a" {
+  vpc_id = aws_vpc.vpc.id
+
+  tags = {
+    Name = "${var.environment} Internet Gateway A"
+  }
+}
+
+resource "aws_route_table" "public_rt" {
+  vpc_id = aws_vpc.vpc.id
+
+  route {
+    cidr_block = "0.0.0.0/0"
+    gateway_id = aws_internet_gateway.ig_a.id
+  }
+
+  route {
+    ipv6_cidr_block = "::/0"
+    gateway_id      = aws_internet_gateway.ig_a.id
+  }
+
+  tags = {
+    Name = "${var.environment} Public Route Table"
+  }
+}
+
+resource "aws_route_table_association" "public_1_rt_a" {
+  subnet_id      = aws_subnet.public_subnet_a.id
+  route_table_id = aws_route_table.public_rt.id
+}
+
+resource "aws_security_group" "web_sg" {
+  name   = "HTTP and SSH"
+  vpc_id = aws_vpc.vpc.id
+
+  ingress {
+    from_port   = 443
+    to_port     = 443
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+
+
+  ingress {
+    from_port   = 80
+    to_port     = 80
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+
+  ingress {
+    from_port   = 22
+    to_port     = 22
+    protocol    = "tcp"
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+
+  egress {
+    from_port   = 0
+    to_port     = 0
+    protocol    = -1
+    cidr_blocks = ["0.0.0.0/0"]
+  }
+
+  tags = {
+    Name = "${var.environment} Web Security Group"
+  }
+}
+
+
+resource "aws_instance" "web" {
+  ami                  = data.aws_ami.ubuntu.id
+  instance_type        = "t4g.micro"
+  key_name             = "key-pair-aws-admin"
+  iam_instance_profile = aws_iam_instance_profile.web.name
+
+  tags = {
+    Name = "web-${var.environment}"
+  }
+
+  subnet_id                   = aws_subnet.public_subnet_a.id
+  security_groups             = [aws_security_group.web_sg.id]
+  associate_public_ip_address = true
+
+  user_data = <<EOF
+#!/bin/bash
+apt-get update
+apt-get install zlib1g-dev libyaml-dev libssl-dev libffi-dev libgmp3-dev libclang-dev build-essential -y  && \
+apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove -y
+curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+curl https://mise.run | sh
+export HOME=/root
+echo "eval \"\$(/root/.local/bin/mise activate bash)\"" >> ~/.bashrc
+eval "$(/root/.local/bin/mise activate bash)"
+mise use ruby@3.4.2
+gem install itsi
+EOF
+
+  root_block_device {
+    encrypted   = true
+    volume_type = "gp3"
+    volume_size = 12
+    tags = {
+      Name = "web-root-block-device-${var.environment}"
+    }
+  }
+
+  depends_on = [aws_route_table_association.public_1_rt_a]
+}
+
+locals {
+  role_policy_arns = [
+    "arn:aws:iam::aws:policy/service-role/AmazonEC2RoleforSSM",
+    "arn:aws:iam::aws:policy/CloudWatchAgentServerPolicy"
+  ]
+}
+
+resource "aws_iam_instance_profile" "web" {
+  name = "EC2-Profile-${var.environment}"
+  role = aws_iam_role.web.name
+}
+
+resource "aws_iam_role_policy_attachment" "web" {
+  count = length(local.role_policy_arns)
+
+  role       = aws_iam_role.web.name
+  policy_arn = element(local.role_policy_arns, count.index)
+}
+
+resource "aws_iam_role_policy" "web" {
+  name = "EC2-Inline-Policy-${var.environment}"
+  role = aws_iam_role.web.id
+  policy = jsonencode(
+    {
+      "Version" : "2012-10-17",
+      "Statement" : [
+        {
+          "Effect" : "Allow",
+          "Action" : [
+            "ssm:GetParameter"
+          ],
+          "Resource" : "*"
+        }
+      ]
+    }
+  )
+}
+
+resource "aws_iam_role" "web" {
+  name = "EC2-Role-${var.environment}"
+  path = "/"
+
+  assume_role_policy = jsonencode(
+    {
+      "Version" : "2012-10-17",
+      "Statement" : [
+        {
+          "Action" : "sts:AssumeRole",
+          "Principal" : {
+            "Service" : "ec2.amazonaws.com"
+          },
+          "Effect" : "Allow"
+        }
+      ]
+    }
+  )
+}

data/sandbox/deploy/outputs.tf

@@ -0,0 +1,4 @@
+output "ec2_public_ip" {
+  description = "Public IP of the EC2 instance"
+  value       = aws_instance.web.public_ip
+}

data/sandbox/deploy/vars.tf

@@ -0,0 +1,11 @@
+variable "aws_region" {
+  default = "ap-southeast-2"
+}
+
+variable "environment" {
+  default = "itsi-production"
+}
+
+variable "subnet_az" {
+  default = "ap-southeast-2a"
+}

data/sandbox/falcon_benchmark/Gemfile

@@ -0,0 +1,10 @@
+source 'https://rubygems.org'
+
+gem 'iodine'
+gem 'unicorn'
+gem 'puma'
+gem 'passenger'
+gem 'agoo'
+gem 'falcon'
+gem "itsi-server", path: "../../gems/server"
+gem "itsi-scheduler", path: "../../gems/scheduler"

data/sandbox/falcon_benchmark/Gemfile.lock

@@ -0,0 +1,140 @@
+PATH
+  remote: ../../gems/scheduler
+  specs:
+    itsi-scheduler (0.1.0)
+      libclang (~> 14.0)
+      rb_sys (~> 0.9.91)
+
+PATH
+  remote: ../../gems/server
+  specs:
+    itsi-server (0.1.1)
+      libclang (~> 14.0)
+      rack (~> 2.2)
+      rb_sys (~> 0.9.91)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    agoo (2.15.13)
+    async (2.23.1)
+      console (~> 1.29)
+      fiber-annotation
+      io-event (~> 1.9)
+      metrics (~> 0.12)
+      traces (~> 0.15)
+    async-container (0.24.0)
+      async (~> 2.22)
+    async-container-supervisor (0.5.1)
+      async-container (~> 0.22)
+      async-service
+      io-endpoint
+      memory-leak (~> 0.5)
+    async-http (0.87.0)
+      async (>= 2.10.2)
+      async-pool (~> 0.9)
+      io-endpoint (~> 0.14)
+      io-stream (~> 0.6)
+      metrics (~> 0.12)
+      protocol-http (~> 0.49)
+      protocol-http1 (~> 0.30)
+      protocol-http2 (~> 0.22)
+      traces (~> 0.10)
+    async-http-cache (0.4.5)
+      async-http (~> 0.56)
+    async-pool (0.10.3)
+      async (>= 1.25)
+    async-service (0.13.0)
+      async
+      async-container (~> 0.16)
+    console (1.30.2)
+      fiber-annotation
+      fiber-local (~> 1.1)
+      json
+    falcon (0.51.1)
+      async
+      async-container (~> 0.20)
+      async-container-supervisor (~> 0.5.0)
+      async-http (~> 0.75)
+      async-http-cache (~> 0.4)
+      async-service (~> 0.10)
+      bundler
+      localhost (~> 1.1)
+      openssl (~> 3.0)
+      protocol-http (~> 0.31)
+      protocol-rack (~> 0.7)
+      samovar (~> 2.3)
+    fiber-annotation (0.2.0)
+    fiber-local (1.1.0)
+      fiber-storage
+    fiber-storage (1.0.0)
+    io-endpoint (0.15.2)
+    io-event (1.10.0)
+    io-stream (0.6.1)
+    iodine (0.7.58)
+    json (2.10.2)
+    kgio (2.11.4)
+    libclang (14.0.6-aarch64-linux)
+    libclang (14.0.6-arm64-darwin)
+    libclang (14.0.6-x86_64-darwin)
+    libclang (14.0.6-x86_64-linux)
+    libclang (14.0.6-x86_64-linux-musl)
+    localhost (1.3.1)
+    mapping (1.1.1)
+    memory-leak (0.5.2)
+    metrics (0.12.2)
+    nio4r (2.7.4)
+    openssl (3.3.0)
+    passenger (6.0.23)
+      rack (>= 1.6.13)
+      rackup
+      rake (>= 12.3.3)
+    protocol-hpack (1.5.1)
+    protocol-http (0.49.0)
+    protocol-http1 (0.30.0)
+      protocol-http (~> 0.22)
+    protocol-http2 (0.22.1)
+      protocol-hpack (~> 1.4)
+      protocol-http (~> 0.47)
+    protocol-rack (0.11.2)
+      protocol-http (~> 0.43)
+      rack (>= 1.0)
+    puma (6.6.0)
+      nio4r (~> 2.0)
+    rack (2.2.13)
+    rackup (1.0.1)
+      rack (< 3)
+      webrick
+    raindrops (0.20.1)
+    rake (13.2.1)
+    rake-compiler-dock (1.9.1)
+    rb_sys (0.9.111)
+      rake-compiler-dock (= 1.9.1)
+    samovar (2.3.0)
+      console (~> 1.0)
+      mapping (~> 1.0)
+    traces (0.15.2)
+    unicorn (6.1.0)
+      kgio (~> 2.6)
+      raindrops (~> 0.7)
+    webrick (1.9.1)
+
+PLATFORMS
+  aarch64-linux
+  arm64-darwin
+  x86_64-darwin
+  x86_64-linux
+  x86_64-linux-musl
+
+DEPENDENCIES
+  agoo
+  falcon
+  iodine
+  itsi-scheduler!
+  itsi-server!
+  passenger
+  puma
+  unicorn
+
+BUNDLED WITH
+   2.6.3