derrick 0.0.10

A tool for provisioning workspaces to run code projects in.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

#!/usr/bin/env ruby

require 'json'
require 'net/http'
require 'socket'

def request(method, path, body=nil)
  body_json = JSON.generate(body) if body
  uri = URI("http://localhost:50080#{path}")
  response = Net::HTTP.start(uri.host, uri.port) do |http|
    request = Net::HTTP.const_get(method.capitalize).new(uri)
    request.body = body_json
    request['Content-Type'] = 'application/json'
    http.request(request)
  end

  if response.code.to_i >= 400
    raise "Request failed with status #{response.code}: #{response.body}"
  end

  JSON.parse(response.body)
end

def run_with_derrick(options)
  command = "cargo run -- #{options.join(' ')}"
  pid = Process.spawn(command)
  puts "Running command: #{command} (PID: #{pid})"

  connected = false
  while !connected do
    begin
      TCPSocket.open('localhost', 50080) { connected = true }
    rescue Errno::ECONNREFUSED
      sleep 0.3
    end
  end

  yield pid
ensure
  Process.kill('TERM', pid)
end

def run_tests(provisioner_mode:)
  file_dir = File.dirname(__FILE__)

  options = ["-p", provisioner_mode, "-s", "http", "-w", "#{file_dir}/test_config.json"]
  run_with_derrick(options) do |pid|
    puts "Running tests..."

    # Test the API
    response = request(:get, '/workspaces')
    raise "Expected empty workspaces, got #{response.inspect}" unless response["workspaces"] == []

    response = request(:post, '/workspaces', { 'env' => {} })
    raise "Expected workspace ID, got #{response.inspect}" unless response['id']

    id = response['id']

    puts "Test that we can get the workspace"
    response = request(:get, '/workspaces')
    raise "Expected empty workspaces, got #{response.inspect}" unless response.dig("workspaces", 0, "id") == id

    puts "Test that we can run a command"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => 'echo hello' })
    raise "Expected output, got #{response.inspect}" unless response == "hello\n"

    puts "Test that we can list files in the workspace"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => 'ls ./code/swiftide-ask' })
    raise "Expected output, got #{response.inspect}" unless response.include?("Cargo.toml")

    puts "Test that we can run a command in the workspace"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => 'cd ./code/swiftide-ask && ls' })
    raise "Expected output, got #{response.inspect}" unless response.include?("Cargo.toml")

    puts "Test that we can run multiline commands"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => "echo hello\necho world" })
    raise "Expected output, got #{response.inspect}" unless response.include?("hello\nworld\n")

    puts "Test that the setup script ran successfully"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => 'cat /tmp/hello.txt' })
    raise "Expected output, got #{response.inspect}" unless response.include?("Hello World")

    puts "Test that we can set environment variables"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => 'echo $HELLO', 'env' => { 'HELLO' => 'WORLD' } })
    raise "Expected output, got #{response.inspect}" unless response.include?("WORLD")

    script_with_heredoc = <<~SCRIPT
      cat <<-"EOF" > /tmp/hello.txt
      hello
      world
      EOF
      cat /tmp/hello.txt
    SCRIPT

    puts "Test that we can use HEREDOCs"
    response = request(:post, "/workspaces/#{id}/cmd_with_output", { 'cmd' => script_with_heredoc })
    raise "Expected output, got #{response.inspect}" unless response.include?("hello\nworld\n")
  end
end

["docker"].each do |provisioner_mode|
  puts "Running tests in #{provisioner_mode} mode..."
  run_tests(provisioner_mode: provisioner_mode)
end

puts "\n\nAll tests passed!\n\n"