1// SPDX-License-Identifier: AGPL-3.0-or-later
2package kvstore_test
3
4import (
5 "bnema.dev/ermite/internal/domain"
6 "context"
7 "github.com/stretchr/testify/assert"
8 "github.com/stretchr/testify/require"
9 "testing"
10)
11
12func TestIssueLinksReplace(t *testing.T) {
13 ctx := context.Background()
14 s, _, repo := repoWithMRs(t)
15 links := s.IssueLinks()
16 added, removed, err := links.ReplaceLinks(ctx, repo, 9, []int64{3, 2, 3}, []int64{2})
17 require.NoError(t, err)
18 assert.Equal(t, []domain.IssueLink{{Issue: 2, MergeRequest: 9, Closes: true}, {Issue: 3, MergeRequest: 9}}, added)
19 assert.Empty(t, removed)
20 got, err := links.LinksOfMergeRequest(ctx, repo, 9)
21 require.NoError(t, err)
22 assert.Equal(t, added, got)
23 added, removed, err = links.ReplaceLinks(ctx, repo, 9, []int64{2, 3}, []int64{2})
24 require.NoError(t, err)
25 assert.Empty(t, added)
26 assert.Empty(t, removed)
27 added, removed, err = links.ReplaceLinks(ctx, repo, 9, []int64{3, 4}, []int64{3})
28 require.NoError(t, err)
29 assert.Equal(t, []domain.IssueLink{{Issue: 3, MergeRequest: 9, Closes: true}, {Issue: 4, MergeRequest: 9}}, added)
30 assert.Equal(t, []domain.IssueLink{{Issue: 2, MergeRequest: 9, Closes: true}}, removed)
31 got, err = links.LinksOfIssue(ctx, repo, 2, 100)
32 require.NoError(t, err)
33 assert.Empty(t, got)
34 added, removed, err = links.ReplaceLinks(ctx, repo, 9, []int64{3}, nil)
35 require.NoError(t, err)
36 assert.Empty(t, added)
37 assert.Equal(t, []domain.IssueLink{{Issue: 3, MergeRequest: 9, Closes: true}, {Issue: 4, MergeRequest: 9}}, removed)
38 for _, mr := range []int64{12, 1, 5} {
39 _, _, err = links.ReplaceLinks(ctx, repo, mr, []int64{3}, nil)
40 require.NoError(t, err)
41 }
42 got, err = links.LinksOfIssue(ctx, repo, 3, 2)
43 require.NoError(t, err)
44 assert.Equal(t, []domain.IssueLink{{Issue: 3, MergeRequest: 1}, {Issue: 3, MergeRequest: 5}}, got)
45 require.NoError(t, s.Delete(ctx, repo))
46 assert.Zero(t, countKeys(t, s, "isref/"+repo.String()+"/"))
47 assert.Zero(t, countKeys(t, s, "mrref/"+repo.String()+"/"))
48 _, _, err = links.ReplaceLinks(ctx, repo, 9, []int64{2}, nil)
49 assert.ErrorIs(t, err, domain.ErrNotFound)
50}
51
52func TestIssueClosedByRoundTrip(t *testing.T) {
53 ctx := context.Background()
54 s, _, repo := repoWithMRs(t)
55 issue := newIssue(t, repo, "closed by merge")
56 issue.State = domain.IssueStateClosed
57 issue.ClosedByMergeRequest = 17
58 stored, err := s.Issues().Create(ctx, issue)
59 require.NoError(t, err)
60 assert.Equal(t, int64(17), stored.ClosedByMergeRequest)
61 stored.ClosedByMergeRequest = 0
62 require.NoError(t, s.Issues().Update(ctx, stored))
63 stored, err = s.Issues().Get(ctx, repo, stored.Number)
64 require.NoError(t, err)
65 assert.Zero(t, stored.ClosedByMergeRequest)
66}