1 files changed, 24 insertions, 23 deletions
diff --git a/bin/test b/bin/test
index 1f2a038..12dde30 100755
--- a/bin/test
+++ b/bin/test
@@ -9,28 +9,29 @@ end
 
 require "minitest/autorun"
 
-def assert(condition)
-  raise "heck" unless condition
+class ServerTest < Minitest::Test
+  attr_reader :base_url, :client
+
+  def setup
+    @base_url = "http://127.0.0.1:7878"
+    @client = Net::Hippie::Client.new
+    pid = Process.spawn("cargo run")
+    sleep 1
+  end
+
+  def teardown
+    # Process.kill(-9, pid)
+    system("killall sts")
+  end
+
+  def test_homepage
+    response = client.get(base_url + "/")
+    assert_equal response.code, "200"
+  end
+
+  def test_not_found
+    response = client.get(base_url + "/404")
+    assert_equal "404", response.code
+  end
 end
 
-def assert_equal(expected, actual)
-  assert(expected == actual)
-end
-
-def run(base_url)
-  client = Net::Hippie::Client.new
-
-  response = client.get(base_url + "/")
-  assert_equal "200", response.code
-
-  response = client.get(base_url + "/404")
-  assert_equal "404", response.code
-end
-
-begin
-  pid = Process.spawn("cargo run")
-  sleep 1
-  run("http://127.0.0.1:7878")
-ensure
-  system("killall sts")
-end