123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745 |
- package session
- import (
- "fmt"
- "io/ioutil"
- "net/http"
- "net/http/httptest"
- "os"
- "testing"
- )
- var SessionId string = ""
- func TestSessionBool(t *testing.T) {
- // Is set (false)
- request, err := http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetBool("some_bool")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- // Set value
- request, err = http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetBool("some_bool", true)
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- if SessionId == "" && len(recorder.Result().Cookies()) > 0 {
- SessionId = recorder.Result().Cookies()[0].Value
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.GetBool("some_bool", false)))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "true" {
- t.Fatal("bad body response, not match")
- }
- // Is set (true)
- request, err = http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetBool("some_bool")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "true" {
- t.Fatal("bad body response, not match")
- }
- // Del
- request, err = http.NewRequest("GET", "/del", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/del" {
- sess.DelBool("some_bool")
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetBool("some_bool")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- }
- func TestSessionInt(t *testing.T) {
- // Is set (false)
- request, err := http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt("some_int")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- // Set value
- request, err = http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetInt("some_int", 5)
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- if SessionId == "" && len(recorder.Result().Cookies()) > 0 {
- SessionId = recorder.Result().Cookies()[0].Value
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(fmt.Sprintf("%d", sess.GetInt("some_int", 0)))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "5" {
- t.Fatal("bad body response, not match")
- }
- // Is set (true)
- request, err = http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt("some_int")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "true" {
- t.Fatal("bad body response, not match")
- }
- // Del
- request, err = http.NewRequest("GET", "/del", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/del" {
- sess.DelInt("some_int")
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt("some_int")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- }
- func TestSessionInt64(t *testing.T) {
- // Is set (false)
- request, err := http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt64("some_int64")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- // Set value
- request, err = http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetInt64("some_int64", 10)
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- if SessionId == "" && len(recorder.Result().Cookies()) > 0 {
- SessionId = recorder.Result().Cookies()[0].Value
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(fmt.Sprintf("%d", sess.GetInt64("some_int64", 0)))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "10" {
- t.Fatal("bad body response, not match")
- }
- // Is set (true)
- request, err = http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt64("some_int64")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "true" {
- t.Fatal("bad body response, not match")
- }
- // Del
- request, err = http.NewRequest("GET", "/del", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/del" {
- sess.DelInt64("some_int64")
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetInt64("some_int64")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- }
- func TestSessionString(t *testing.T) {
- // Is set (false)
- request, err := http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetString("some_str")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- // Set value
- request, err = http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetString("some_str", "test")
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- if SessionId == "" && len(recorder.Result().Cookies()) > 0 {
- SessionId = recorder.Result().Cookies()[0].Value
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(sess.GetString("some_str", ""))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "test" {
- t.Fatal("bad body response, not match")
- }
- // Is set (true)
- request, err = http.NewRequest("GET", "/isset", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/isset" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetString("some_str")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "true" {
- t.Fatal("bad body response, not match")
- }
- // Del
- request, err = http.NewRequest("GET", "/del", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+SessionId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/del" {
- sess.DelString("some_str")
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.IsSetString("some_str")))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "false" {
- t.Fatal("bad body response, not match")
- }
- }
- func TestSessionActualFile(t *testing.T) {
- if SessionId == "" {
- t.Fatal("SessionId is empty")
- }
- fname := "./../tmp" + string(os.PathSeparator) + SessionId
- bytes, err := ioutil.ReadFile(fname)
- if err != nil {
- t.Fatal(err)
- }
- if string(bytes) != `{"Bool":{},"Int":{},"Int64":{},"String":{}}` {
- t.Fatal("actual file content, not match")
- }
- err = os.Remove(fname)
- if err != nil {
- t.Fatal(err)
- }
- }
- func TestSessionDoNotCreateSessionFileForDefValues(t *testing.T) {
- // Set default values
- request, err := http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetBool("some_bool", false)
- sess.SetInt("some_int", 0)
- sess.SetInt64("some_int64", 0)
- sess.SetString("some_str", "")
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- var sessId string
- if len(recorder.Result().Cookies()) > 0 {
- sessId = recorder.Result().Cookies()[0].Value
- }
- if sessId == "" {
- t.Fatal("session identifier is not defined")
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+sessId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(fmt.Sprintf(
- "(%v)(%v)(%v)(%v)",
- sess.GetBool("some_bool", false),
- sess.GetInt("some_int", 0),
- sess.GetInt64("some_int64", 0),
- sess.GetString("some_str", ""),
- ))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "(false)(0)(0)()" {
- t.Fatal("bad body response, not match")
- }
- // Check session file
- fname := "./../tmp" + string(os.PathSeparator) + sessId
- _, err = ioutil.ReadFile(fname)
- if err == nil {
- _ = os.Remove(fname)
- t.Fatal("session file in tmp folder do not must exists")
- }
- }
- func TestSessionDestroy(t *testing.T) {
- // Set value
- request, err := http.NewRequest("GET", "/set", nil)
- if err != nil {
- t.Fatal(err)
- }
- recorder := httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/set" {
- sess.SetInt("some_var", 1)
- if _, err := w.Write([]byte(`ok`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "ok" {
- t.Fatal("bad body response, not match")
- }
- // Remember session id
- var sessId string
- if len(recorder.Result().Cookies()) > 0 {
- sessId = recorder.Result().Cookies()[0].Value
- }
- if sessId == "" {
- t.Fatal("session identifier is not defined")
- }
- // Get value
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+sessId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- if _, err := w.Write([]byte(fmt.Sprintf("%v", sess.GetInt("some_var", 0)))); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "1" {
- t.Fatal("bad body response, not match")
- }
- // Check destroy
- request, err = http.NewRequest("GET", "/get", nil)
- if err != nil {
- t.Fatal(err)
- }
- request.Header.Set("Cookie", "session="+sessId)
- recorder = httptest.NewRecorder()
- http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
- sess, _ := New(w, r, "./../tmp")
- defer sess.Close()
- if r.URL.Path == "/get" {
- sess.SetInt("some_var", 2)
- err := sess.Destroy()
- if err == nil {
- if _, err := w.Write([]byte(`OK`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- } else {
- if _, err := w.Write([]byte(`ERROR`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- } else {
- w.WriteHeader(http.StatusNotFound)
- if _, err := w.Write([]byte(`404`)); err != nil {
- fmt.Printf("%s\n", err.Error())
- }
- }
- }).ServeHTTP(recorder, request)
- if recorder.Body.String() != "OK" {
- t.Fatal("bad body response, not match")
- }
- // Check session file
- fname := "./../tmp" + string(os.PathSeparator) + sessId
- _, err = ioutil.ReadFile(fname)
- if err == nil {
- _ = os.Remove(fname)
- t.Fatal("session file in tmp folder do not must exists")
- }
- }
|