// runStatus is the code that implements the status command. func runStatus(cmd *cobra.Command, args []string) { cmd.Printf("Status User : Pid[%s] Email[%s] Active[%v]\n", status.pid, status.email, status.active) if status.pid == "" && status.email == "" { cmd.Help() return } db := db.NewMGO() defer db.CloseMGO() var publicID string if status.pid != "" { publicID = status.pid } else { u, err := auth.GetUserByEmail("", db, status.email, false) if err != nil { cmd.Println("Status User : "******"", db, publicID, st); err != nil { cmd.Println("Status User : "******"Status User : Updated") }
// runGet is the code that implements the get command. func runGet(cmd *cobra.Command, args []string) { cmd.Printf("Getting User : Pid[%s] Email[%s]\n", get.pid, get.email) if get.pid == "" && get.email == "" { cmd.Help() return } db, err := db.NewMGO("", mgoSession) if err != nil { cmd.Println("Getting User : "******"") var u *auth.User if get.pid != "" { u, err = auth.GetUserByPublicID("", db, get.pid, false) } else { u, err = auth.GetUserByEmail("", db, get.email, false) } if err != nil { cmd.Println("Getting User : "******"", db, u.PublicID) if err != nil { cmd.Println("Getting User : Unable to retrieve web token : ", err) } data, err := json.MarshalIndent(&u, "", " ") if err != nil { cmd.Println("Getting User : "******"\n%s\n\nToken: %s\n\n", string(data), webTok) return }
// TestCreateUser tests the creation of a user. func TestCreateUser(t *testing.T) { tests.ResetLog() defer tests.DisplayLog() db, err := db.NewMGO(tests.Context, tests.TestSession) if err != nil { t.Fatalf("\t%s\tShould be able to get a Mongo session : %v", tests.Failed, err) } defer db.CloseMGO(tests.Context) var publicID string defer func() { if err := removeUser(db, publicID); err != nil { t.Fatalf("\t%s\tShould be able to remove the test user : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to remove the test user.", tests.Success) }() t.Log("Given the need to create users in the DB.") { t.Log("\tWhen using a test user.") { u1, err := auth.NewUser(auth.NUser{ Status: auth.StatusActive, FullName: "Test Kennedy", Email: "*****@*****.**", Password: "******", }) if err != nil { t.Fatalf("\t%s\tShould be able to build a new user : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to build a new user.", tests.Success) if err := auth.CreateUser(tests.Context, db, u1); err != nil { t.Fatalf("\t%s\tShould be able to create a user : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to create a user.", tests.Success) // We need to do this so we can clean up after. publicID = u1.PublicID u2, err := auth.GetUserByPublicID(tests.Context, db, u1.PublicID, true) if err != nil { t.Fatalf("\t%s\tShould be able to retrieve the user by PublicID : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to retrieve the user by PublicID.", tests.Success) // Remove the objectid to be able to compare the values. u2.ID = "" // Need to remove the nanoseconds to be able to compare the values. u1.DateModified = u1.DateModified.Add(-time.Duration(u1.DateModified.Nanosecond())) u1.DateCreated = u1.DateCreated.Add(-time.Duration(u1.DateCreated.Nanosecond())) u2.DateModified = u2.DateModified.Add(-time.Duration(u2.DateModified.Nanosecond())) u2.DateCreated = u2.DateCreated.Add(-time.Duration(u2.DateCreated.Nanosecond())) if !reflect.DeepEqual(*u1, *u2) { t.Logf("\t%+v", *u1) t.Logf("\t%+v", *u2) t.Fatalf("\t%s\tShould be able to get back the same user.", tests.Failed) } else { t.Logf("\t%s\tShould be able to get back the same user.", tests.Success) } u3, err := auth.GetUserByEmail(tests.Context, db, u1.Email, true) if err != nil { t.Fatalf("\t%s\tShould be able to retrieve the user by Email : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to retrieve the user by Email.", tests.Success) // Remove the objectid to be able to compare the values. u3.ID = "" // Need to remove the nanoseconds to be able to compare the values. u3.DateModified = u3.DateModified.Add(-time.Duration(u3.DateModified.Nanosecond())) u3.DateCreated = u3.DateCreated.Add(-time.Duration(u3.DateCreated.Nanosecond())) if !reflect.DeepEqual(*u1, *u3) { t.Logf("\t%+v", *u1) t.Logf("\t%+v", *u3) t.Fatalf("\t%s\tShould be able to get back the same user.", tests.Failed) } else { t.Logf("\t%s\tShould be able to get back the same user.", tests.Success) } } } }
// TestNoSession tests when a nil session is used. func TestNoSession(t *testing.T) { tests.ResetLog() defer tests.DisplayLog() t.Log("Given the need to test calls with a bad session.") { t.Log("\tWhen using a nil session") { u1, err := auth.NewUser(auth.NUser{ Status: auth.StatusActive, FullName: "Test Kennedy", Email: "*****@*****.**", Password: "******", }) if err != nil { t.Fatalf("\t%s\tShould be able to build a new user : %v", tests.Failed, err) } t.Logf("\t%s\tShould be able to build a new user.", tests.Success) if err := auth.CreateUser(tests.Context, nil, u1); err == nil { t.Errorf("\t%s\tShould Not be able to create a user.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to create a user.", tests.Success) } if _, err := auth.CreateWebToken(tests.Context, nil, u1, time.Second); err == nil { t.Errorf("\t%s\tShould Not be able to create a web token.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to create a web token.", tests.Success) } webTok := "OGY4OGI3YWQtZjc5Ny00ODI1LWI0MmUtMjIwZTY5ZDQxYjMzOmFKT2U1b0pFZlZ4cWUrR0JONEl0WlhmQTY0K3JsN2VGcmM2MVNQMkV1WVE9" if _, err := auth.ValidateWebToken(tests.Context, nil, webTok); err == nil { t.Errorf("\t%s\tShould Not be able to validate a web token.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to validate a web token.", tests.Success) } if _, err := auth.GetUserByPublicID(tests.Context, nil, "6dcda2da-92c3-11e5-8994-feff819cdc9f", true); err == nil { t.Errorf("\t%s\tShould Not be able to get a user by PublicID.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to get a user by PublicID.", tests.Success) } if _, err := auth.GetUserByEmail(tests.Context, nil, "*****@*****.**", true); err == nil { t.Errorf("\t%s\tShould Not be able to get a user by Email.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to get a user by Email.", tests.Success) } uu := auth.UpdUser{ PublicID: "6dcda2da-92c3-11e5-8994-feff819cdc9f", Status: auth.StatusActive, FullName: "Update Kennedy", Email: "*****@*****.**", } if err := auth.UpdateUser(tests.Context, nil, uu); err == nil { t.Errorf("\t%s\tShould Not be able to update a user.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to update a user.", tests.Success) } if err := auth.UpdateUserPassword(tests.Context, nil, u1, "password890"); err == nil { t.Errorf("\t%s\tShould Not be able to update a user pasword.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to update a user password.", tests.Success) } if err := auth.UpdateUserStatus(tests.Context, nil, "6dcda2da-92c3-11e5-8994-feff819cdc9f", auth.StatusDisabled); err == nil { t.Errorf("\t%s\tShould Not be able to disable a user.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to disable a user.", tests.Success) } if _, err := auth.LoginUser(tests.Context, nil, "*****@*****.**", "_pass"); err == nil { t.Errorf("\t%s\tShould Not be able to login a user.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to login a user.", tests.Success) } if _, err := auth.GetUserWebToken(tests.Context, nil, "6dcda2da-92c3-11e5-8994-feff819cdc9f"); err == nil { t.Errorf("\t%s\tShould Not be able to get user web token.", tests.Failed) } else { t.Logf("\t%s\tShould Not be able to get user web token.", tests.Success) } } } }