@@ -9,30 +9,31 @@ import (
99 "testing"
1010 "time"
1111
12- "code.gitea.io/gitea/modules/test"
1312 "code.gitea.io/gitea/tests"
1413
1514 "github.com/stretchr/testify/assert"
1615)
1716
1817func TestViewTimetrackingControls (t * testing.T ) {
1918 defer tests .PrepareTestEnv (t )()
20- session := loginUser (t , "user2" )
21- testViewTimetrackingControls (t , session , "user2" , "repo1" , "1" , true )
22- // user2/repo1
23- }
2419
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+ })
3125
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+ })
3637}
3738
3839func testViewTimetrackingControls (t * testing.T , session * TestSession , user , repo , issue string , canTrackTime bool ) {
@@ -41,35 +42,36 @@ func testViewTimetrackingControls(t *testing.T, session *TestSession, user, repo
4142
4243 htmlDoc := NewHTMLParser (t , resp .Body )
4344
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 )
4647
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 {
4850 "_csrf" : htmlDoc .GetCSRF (),
4951 })
5052 if canTrackTime {
51- resp = session .MakeRequest (t , req , http .StatusSeeOther )
53+ session .MakeRequest (t , req , http .StatusOK )
5254
53- req = NewRequest (t , "GET" , test . RedirectURL ( resp ) )
55+ req = NewRequest (t , "GET" , issueLink )
5456 resp = session .MakeRequest (t , req , http .StatusOK )
5557 htmlDoc = NewHTMLParser (t , resp .Body )
5658
5759 events := htmlDoc .doc .Find (".event > span.text" )
5860 assert .Contains (t , events .Last ().Text (), "started working" )
5961
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 )
6264
6365 // Sleep for 1 second to not get wrong order for stopping timer
6466 time .Sleep (time .Second )
6567
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 {
6769 "_csrf" : htmlDoc .GetCSRF (),
6870 })
69- resp = session .MakeRequest (t , req , http .StatusSeeOther )
71+ session .MakeRequest (t , req , http .StatusOK )
7072
71- req = NewRequest (t , "GET" , test . RedirectURL ( resp ) )
72- resp = session .MakeRequest (t , req , http .StatusOK )
73+ req = NewRequest (t , "GET" , issueLink )
74+ session .MakeRequest (t , req , http .StatusOK )
7375 htmlDoc = NewHTMLParser (t , resp .Body )
7476
7577 events = htmlDoc .doc .Find (".event > span.text" )
0 commit comments