@@ -9,30 +9,31 @@ import (
9
9
"testing"
10
10
"time"
11
11
12
- "code.gitea.io/gitea/modules/test"
13
12
"code.gitea.io/gitea/tests"
14
13
15
14
"github.com/stretchr/testify/assert"
16
15
)
17
16
18
17
func TestViewTimetrackingControls (t * testing.T ) {
19
18
defer tests .PrepareTestEnv (t )()
20
- session := loginUser (t , "user2" )
21
- testViewTimetrackingControls (t , session , "user2" , "repo1" , "1" , true )
22
- // user2/repo1
23
- }
24
19
25
- func TestNotViewTimetrackingControls (t * testing.T ) {
26
- defer tests .PrepareTestEnv (t )()
27
- session := loginUser (t , "user5" )
28
- testViewTimetrackingControls (t , session , "user2" , "repo1" , "1" , false )
29
- // user2/repo1
30
- }
20
+ t .Run ("Exist" , func (t * testing.T ) {
21
+ defer tests .PrintCurrentTest (t )()
22
+ session := loginUser (t , "user2" )
23
+ testViewTimetrackingControls (t , session , "user2" , "repo1" , "1" , true )
24
+ })
31
25
32
- func TestViewTimetrackingControlsDisabled (t * testing.T ) {
33
- defer tests .PrepareTestEnv (t )()
34
- session := loginUser (t , "user2" )
35
- testViewTimetrackingControls (t , session , "org3" , "repo3" , "1" , false )
26
+ t .Run ("Non-exist" , func (t * testing.T ) {
27
+ defer tests .PrintCurrentTest (t )()
28
+ session := loginUser (t , "user5" )
29
+ testViewTimetrackingControls (t , session , "user2" , "repo1" , "1" , false )
30
+ })
31
+
32
+ t .Run ("Disabled" , func (t * testing.T ) {
33
+ defer tests .PrintCurrentTest (t )()
34
+ session := loginUser (t , "user2" )
35
+ testViewTimetrackingControls (t , session , "org3" , "repo3" , "1" , false )
36
+ })
36
37
}
37
38
38
39
func testViewTimetrackingControls (t * testing.T , session * TestSession , user , repo , issue string , canTrackTime bool ) {
@@ -41,34 +42,35 @@ func testViewTimetrackingControls(t *testing.T, session *TestSession, user, repo
41
42
42
43
htmlDoc := NewHTMLParser (t , resp .Body )
43
44
44
- htmlDoc .AssertElement (t , ".timetrack . issue-start-time" , canTrackTime )
45
- htmlDoc .AssertElement (t , ".timetrack . issue-add-time" , canTrackTime )
45
+ htmlDoc .AssertElement (t , ".issue-start-time" , canTrackTime )
46
+ htmlDoc .AssertElement (t , ".issue-add-time" , canTrackTime )
46
47
47
- req = NewRequestWithValues (t , "POST" , path .Join (user , repo , "issues" , issue , "times" , "stopwatch" , "toggle" ), map [string ]string {
48
+ issueLink := path .Join (user , repo , "issues" , issue )
49
+ req = NewRequestWithValues (t , "POST" , path .Join (issueLink , "times" , "stopwatch" , "toggle" ), map [string ]string {
48
50
"_csrf" : htmlDoc .GetCSRF (),
49
51
})
50
52
if canTrackTime {
51
- resp = session .MakeRequest (t , req , http .StatusSeeOther )
53
+ resp = session .MakeRequest (t , req , http .StatusOK )
52
54
53
- req = NewRequest (t , "GET" , test . RedirectURL ( resp ) )
55
+ req = NewRequest (t , "GET" , issueLink )
54
56
resp = session .MakeRequest (t , req , http .StatusOK )
55
57
htmlDoc = NewHTMLParser (t , resp .Body )
56
58
57
59
events := htmlDoc .doc .Find (".event > span.text" )
58
60
assert .Contains (t , events .Last ().Text (), "started working" )
59
61
60
- htmlDoc .AssertElement (t , ".timetrack . issue-stop-time" , true )
61
- htmlDoc .AssertElement (t , ".timetrack . issue-cancel-time" , true )
62
+ htmlDoc .AssertElement (t , ".issue-stop-time" , true )
63
+ htmlDoc .AssertElement (t , ".issue-cancel-time" , true )
62
64
63
65
// Sleep for 1 second to not get wrong order for stopping timer
64
66
time .Sleep (time .Second )
65
67
66
- req = NewRequestWithValues (t , "POST" , path .Join (user , repo , "issues" , issue , "times" , "stopwatch" , "toggle" ), map [string ]string {
68
+ req = NewRequestWithValues (t , "POST" , path .Join (issueLink , "times" , "stopwatch" , "toggle" ), map [string ]string {
67
69
"_csrf" : htmlDoc .GetCSRF (),
68
70
})
69
- resp = session .MakeRequest (t , req , http .StatusSeeOther )
71
+ resp = session .MakeRequest (t , req , http .StatusOK )
70
72
71
- req = NewRequest (t , "GET" , test . RedirectURL ( resp ) )
73
+ req = NewRequest (t , "GET" , issueLink )
72
74
resp = session .MakeRequest (t , req , http .StatusOK )
73
75
htmlDoc = NewHTMLParser (t , resp .Body )
74
76
0 commit comments