Add test for get

1 files changed, 35 insertions, 0 deletions

diff --git a/jail_test.go b/jail_test.go
index 1967175..72e9fa3 100644
--- a/jail_test.go
+++ b/jail_test.go
@@ -33,6 +33,41 @@ func TestRemove(t *testing.T) {
 	}
 }
 
+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"