package jail

import "testing"

func TestIov(t *testing.T) {
	params := make(map[string]interface{})
	params["name"] = "test"
	params["path"] = "/"

	_, niov, fl := mapToIov(params)

	defer freeIov(fl)

	if niov != 4 {
		t.Error("mapToIov return wrong number of pairs ", niov)
	}
}

func TestRemove(t *testing.T) {
	params := make(map[string]interface{})
	params["name"] = "test_remove"
	params["path"] = "/"
	params["persist"] = true

	jid, err := Set(params, JAIL_CREATE)
	if err != nil {
		t.Error("Failed to create jail: ", err.Error())
	}

	err = Remove(jid)
	if err != nil {
		t.Error("Failed to remove jail: ", err.Error())
	}
}

func TestGet(t *testing.T) {
	params := make(map[string]interface{})
	params["name"] = "test_get"
	params["path"] = "/"
	params["persist"] = true

	jid, err := Set(params, JAIL_CREATE)
	if err != nil {
		t.Error("Failed to create jail: ", err.Error())
	}

	delete(params, "persist")
	params["name"] = "          "
	params["path"] = "          "
	params["jid"] = jid

	jid, err = Get(params, 0)
	if err != nil {
		t.Error("Failed to get jail info: ", err.Error())
	}

	if params["name"] != "test_get" {
		t.Error("Wrong name: ", params["name"])
	}

	if params["path"] != "/" {
		t.Error("Wrong path: ", params["path"])
	}

	err = Remove(jid)
	if err != nil {
		t.Error("Failed to remove jail: ", err.Error())
	}
}

func TestCreate(t *testing.T) {
	params := make(map[string]interface{})
	params["name"] = "test_create"
	params["path"] = "/"

	_, err := Set(params, JAIL_CREATE | JAIL_ATTACH)
	if err != nil {
		t.Error("Failed to create jail: ", err.Error())
	}
}