Skip to content

Commit

Permalink
Merge pull request #1700 from stakwork/feat/test_created_bounties
Browse files Browse the repository at this point in the history 
Refactored GetCreatedBounties tests to use real DB
  • Loading branch information
@elraphty
elraphty committed Jun 17, 2024
2 parents d5f014f + 3f0c977 commit 67b4b3c
Show file tree
Hide file tree
Showing 2 changed files with 85 additions and 94 deletions.
2 changes: 1 addition & 1 deletion db/db.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -828,7 +828,7 @@ func (db database) GetCreatedBounties(r *http.Request) ([]NewBounty, error) {
orderQuery = "ORDER BY created DESC"
}

if offset >= 0 && limit != 0 {
if offset >= 0 && limit > 1 {
limitQuery = fmt.Sprintf("LIMIT %d OFFSET %d", limit, offset)
}

Expand Down
177 changes: 84 additions & 93 deletions handlers/bounty_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -644,57 +644,86 @@ func TestGetPersonAssignedBounties(t *testing.T) {
}

func TestGetPersonCreatedBounties(t *testing.T) {
teardownSuite := setupSuite(t)
defer teardownSuite(t)

ctx := context.Background()
mockDb := dbMocks.NewDatabase(t)
mockHttpClient := mocks.NewHttpClient(t)
bHandler := NewBountyHandler(mockHttpClient, mockDb)
bHandler := NewBountyHandler(mockHttpClient, db.TestDB)

t.Run("should return bounties created by the user", func(t *testing.T) {
mockGenerateBountyResponse := func(bounties []db.NewBounty) []db.BountyResponse {
var bountyResponses []db.BountyResponse
bountyOwner := db.Person{
Uuid: "user_3_uuid",
OwnerAlias: "user3",
UniqueName: "user3",
OwnerPubKey: "user_3_pubkey",
PriceToMeet: 0,
Description: "this is test user 3",
}

for _, bounty := range bounties {
owner := db.Person{
ID: 1,
}
assignee := db.Person{
ID: 1,
}
workspace := db.WorkspaceShort{
Uuid: "uuid",
}
bountyAssignee := db.Person{
Uuid: "user_4_uuid",
OwnerAlias: "user4",
UniqueName: "user4",
OwnerPubKey: "user_4_pubkey",
PriceToMeet: 0,
Description: "this is user 4",
}

bountyResponse := db.BountyResponse{
Bounty: bounty,
Assignee: assignee,
Owner: owner,
Organization: workspace,
Workspace: workspace,
}
bountyResponses = append(bountyResponses, bountyResponse)
}
bounty := db.NewBounty{
Type: "coding",
Title: "first bounty 3",
Description: "first bounty description",
WorkspaceUuid: "work-4",
Assignee: bountyAssignee.OwnerPubKey,
OwnerID: bountyOwner.OwnerPubKey,
Show: true,
}

return bountyResponses
}
bHandler.generateBountyResponse = mockGenerateBountyResponse
bounty2 := db.NewBounty{
Type: "coding 2",
Title: "second bounty 3",
Description: "second bounty description 2",
OrgUuid: "org-4",
WorkspaceUuid: "work-4",
Assignee: bountyAssignee.OwnerPubKey,
OwnerID: bountyOwner.OwnerPubKey,
Show: true,
Created: 11111111,
}

expectedBounties := []db.NewBounty{
{ID: 1, OwnerID: "user1"},
{ID: 2, OwnerID: "user1"},
}
bounty3 := db.NewBounty{
Type: "coding 2",
Title: "second bounty 4",
Description: "second bounty description 2",
WorkspaceUuid: "work-4",
Assignee: "",
OwnerID: bountyOwner.OwnerPubKey,
Show: true,
Created: 2222222,
}

mockDb.On("GetCreatedBounties", mock.Anything).Return(expectedBounties, nil).Once()
mockDb.On("GetPersonByPubkey", mock.Anything).Return(db.Person{}, nil)
mockDb.On("GetWorkspaceByUuid", mock.Anything).Return(db.Workspace{}, nil)
// create users
db.TestDB.CreateOrEditPerson(bountyOwner)
db.TestDB.CreateOrEditPerson(bountyAssignee)

// create bounty
db.TestDB.CreateOrEditBounty(bounty)
db.TestDB.CreateOrEditBounty(bounty2)
db.TestDB.CreateOrEditBounty(bounty3)

t.Run("should return bounties created by the user", func(t *testing.T) {
rr := httptest.NewRecorder()
req, err := http.NewRequest("GET", "/people/wanteds/created/uuid", nil)
req = req.WithContext(ctx)

rctx := chi.NewRouteContext()
rctx.URLParams.Add("uuid", bountyOwner.Uuid)

route := fmt.Sprintf("/people/wanteds/created/%s?sortBy=paid&page=1&limit=20&search=''", bountyOwner.Uuid)
req, err := http.NewRequestWithContext(context.WithValue(context.Background(), chi.RouteCtxKey, rctx), http.MethodGet, route, nil)
if err != nil {
t.Fatal(err)
}

bHandler.GetPersonCreatedBounties(rr, req)

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

var responseData []db.BountyResponse
Expand All @@ -703,25 +732,21 @@ func TestGetPersonCreatedBounties(t *testing.T) {
t.Fatalf("Error decoding JSON response: %s", err)
}

assert.NotEmpty(t, responseData)
assert.Len(t, responseData, 2)
// bounty from db
expectedBounty, _ := db.TestDB.GetCreatedBounties(req)

for i, expectedBounty := range expectedBounties {
assert.Equal(t, expectedBounty.ID, responseData[i].Bounty.ID)
assert.Equal(t, expectedBounty.Assignee, responseData[i].Bounty.Assignee)
}
assert.NotEmpty(t, responseData)
assert.Equal(t, len(expectedBounty), len(responseData))
})

t.Run("should not return bounties created by other users", func(t *testing.T) {
mockGenerateBountyResponse := func(bounties []db.NewBounty) []db.BountyResponse {
return []db.BountyResponse{}
}
bHandler.generateBountyResponse = mockGenerateBountyResponse
rr := httptest.NewRecorder()

mockDb.On("GetCreatedBounties", mock.Anything).Return([]db.NewBounty{}, nil).Once()
rctx := chi.NewRouteContext()
rctx.URLParams.Add("uuid", bountyAssignee.Uuid)

rr := httptest.NewRecorder()
req, err := http.NewRequest("GET", "/people/wanteds/created/uuid", nil)
route := fmt.Sprintf("/people/wanteds/created/%s?sortBy=paid&page=1&limit=20&search=''", bountyAssignee.Uuid)
req, err := http.NewRequest("GET", route, nil)
req = req.WithContext(ctx)
if err != nil {
t.Fatal(err)
Expand All @@ -742,47 +767,13 @@ func TestGetPersonCreatedBounties(t *testing.T) {
})

t.Run("should filter bounties by status and apply pagination", func(t *testing.T) {
mockGenerateBountyResponse := func(bounties []db.NewBounty) []db.BountyResponse {
var bountyResponses []db.BountyResponse

for _, bounty := range bounties {
owner := db.Person{
ID: 1,
}
assignee := db.Person{
ID: 1,
}
workspace := db.WorkspaceShort{
Uuid: "uuid",
}

bountyResponse := db.BountyResponse{
Bounty: bounty,
Assignee: assignee,
Owner: owner,
Organization: workspace,
Workspace: workspace,
}
bountyResponses = append(bountyResponses, bountyResponse)
}

return bountyResponses
}
bHandler.generateBountyResponse = mockGenerateBountyResponse

expectedBounties := []db.NewBounty{
{ID: 1, OwnerID: "user1", Assignee: "assignee1"},
{ID: 2, OwnerID: "user1", Assignee: "assignee2", Paid: true},
{ID: 3, OwnerID: "user1", Assignee: "", Paid: true},
}
rr := httptest.NewRecorder()

mockDb.On("GetCreatedBounties", mock.Anything).Return(expectedBounties, nil).Once()
mockDb.On("GetPersonByPubkey", mock.Anything).Return(db.Person{}, nil)
mockDb.On("GetWorkspaceByUuid", mock.Anything).Return(db.Workspace{}, nil)
rctx := chi.NewRouteContext()
rctx.URLParams.Add("uuid", bountyOwner.Uuid)

rr := httptest.NewRecorder()
req, err := http.NewRequest("GET", "/people/wanteds/created/uuid?Open=true&Assigned=true&Paid=true&offset=0&limit=2", nil)
req = req.WithContext(ctx)
route := fmt.Sprintf("/people/wanteds/created/%s?Assigned=true&page=1&limit=2", bountyOwner.Uuid)
req, err := http.NewRequestWithContext(context.WithValue(context.Background(), chi.RouteCtxKey, rctx), http.MethodGet, route, nil)
if err != nil {
t.Fatal(err)
}
Expand All @@ -797,12 +788,12 @@ func TestGetPersonCreatedBounties(t *testing.T) {
t.Fatalf("Error decoding JSON response: %s", err)
}

assert.Len(t, responseData, 3)
assert.Len(t, responseData, 2)

// Assert that bounties are filtered correctly
assert.Equal(t, expectedBounties[0].ID, responseData[0].Bounty.ID)
assert.Equal(t, expectedBounties[1].ID, responseData[1].Bounty.ID)
assert.Equal(t, expectedBounties[2].ID, responseData[2].Bounty.ID)
// bounty from db
expectedBounty, _ := db.TestDB.GetCreatedBounties(req)
assert.Equal(t, len(expectedBounty), len(responseData))
})
}

Expand Down

0 comments on commit 67b4b3c

Please sign in to comment.