package storage

import (
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestPrintsStorage(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(storageHandler))
	defer server.Close()
	res, err := http.Get(server.URL)
	if err != nil {
		log.Fatal(err)
	}
	body, err := ioutil.ReadAll(res.Body)
	res.Body.Close()
	if err != nil {
		log.Fatal(err)
	}
	bodyString := string(body)
	if bodyString != "Storage" {
		log.Fatal("Expected 'Storage', Received '" + bodyString + "'")
	}
}

func storageHandler(w http.ResponseWriter, req *http.Request) {
	fmt.Fprintf(w, "Storage")
}