Skip to content

Commit

Permalink
Merge pull request stakwork#1717 from Ekep-Obasi/refactor/test-delete…
Browse files Browse the repository at this point in the history
…-person

Refactor TestDeletePerson To Use A Real Postgres DB For The Test
  • Loading branch information
elraphty authored Jun 21, 2024
2 parents ac99d06 + f0ecd8c commit 4961847
Showing 1 changed file with 21 additions and 10 deletions.
31 changes: 21 additions & 10 deletions handlers/people_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -243,34 +243,45 @@ func TestGetPersonById(t *testing.T) {
}

func TestDeletePerson(t *testing.T) {
mockDb := mocks.NewDatabase(t)
pHandler := NewPeopleHandler(mockDb)
teardownSuite := SetupSuite(t)
defer teardownSuite(t)

pHandler := NewPeopleHandler(db.TestDB)

t.Run("successful deletion", func(t *testing.T) {
rr := httptest.NewRecorder()
handler := http.HandlerFunc(pHandler.DeletePerson)
person := db.Person{
ID: 1,
Uuid: "test-uuid",
OwnerPubKey: "owner-pub-key",
OwnerAlias: "owner",
ID: 112,
Uuid: "person_112_uuid",
OwnerPubKey: "person_112_pubkey",
OwnerAlias: "owner",
UniqueName: "test_user",
Description: "test user",
Tags: pq.StringArray{},
Extras: db.PropertyMap{},
GithubIssues: db.PropertyMap{},
}

rctx := chi.NewRouteContext()
rctx.URLParams.Add("id", "1")
rctx.URLParams.Add("id", strconv.Itoa(int(person.ID)))

ctx := context.WithValue(context.Background(), chi.RouteCtxKey, rctx)
ctx = context.WithValue(ctx, auth.ContextKey, person.OwnerPubKey)

req, err := http.NewRequestWithContext(ctx, http.MethodDelete, "/person", nil)
assert.NoError(t, err)

mockDb.On("GetPerson", person.ID).Return(person).Once()
mockDb.On("UpdatePerson", person.ID, mock.Anything).Return(true).Once()
db.TestDB.CreateOrEditPerson(person)
fetchedPerson := db.TestDB.GetPerson(person.ID)
assert.Equal(t, person, fetchedPerson)

handler.ServeHTTP(rr, req)

assert.Equal(t, http.StatusOK, rr.Code)
mockDb.AssertExpectations(t)

deletedPerson := db.TestDB.GetPerson(person.ID)
assert.Empty(t, deletedPerson)
})
}

Expand Down

0 comments on commit 4961847

Please sign in to comment.