Test: model.Tags / modifications

pkg/aria2/aria2_test.go

@@ -36,6 +36,7 @@ func TestDummyAria2(t *testing.T) {
 }
 
 func TestInit(t *testing.T) {
+	MAX_RETRY = 0
 	asserts := assert.New(t)
 	cache.Set("setting_aria2_token", "1", 0)
 	cache.Set("setting_aria2_call_timeout", "5", 0)

pkg/aria2/monitor.go

@@ -32,6 +32,8 @@ type StatusEvent struct {
 	Status int
 }
 
+var MAX_RETRY = 10
+
 // NewMonitor 新建上传状态监控
 func NewMonitor(task *model.Download) {
 	monitor := &Monitor{
@@ -73,7 +75,7 @@ func (monitor *Monitor) Update() bool {
 		util.Log().Warning("无法获取下载任务[%s]的状态，%s", monitor.Task.GID, err)
 
 		// 十次重试后认定为任务失败
-		if monitor.retried > 10 {
+		if monitor.retried > MAX_RETRY {
 			util.Log().Warning("无法获取下载任务[%s]的状态，超过最大重试次数限制，%s", monitor.Task.GID, err)
 			monitor.setErrorStatus(err)
 			monitor.RemoveTempFolder()

pkg/aria2/monitor_test.go

@@ -50,6 +50,7 @@ func TestNewMonitor(t *testing.T) {
 func TestMonitor_Loop(t *testing.T) {
 	asserts := assert.New(t)
 	notifier := make(chan StatusEvent)
+	MAX_RETRY = 0
 	monitor := &Monitor{
 		Task:     &model.Download{GID: "gid"},
 		Interval: time.Duration(1) * time.Second,
@@ -72,11 +73,13 @@ func TestMonitor_Update(t *testing.T) {
 
 	// 无法获取状态
 	{
+		MAX_RETRY = 1
 		testInstance := new(InstanceMock)
 		testInstance.On("Status", testMock.Anything).Return(rpc.StatusInfo{}, errors.New("error"))
 		file, _ := util.CreatNestedFile("TestMonitor_Update/1")
 		file.Close()
 		Instance = testInstance
+		asserts.False(monitor.Update())
 		asserts.True(monitor.Update())
 		testInstance.AssertExpectations(t)
 		asserts.False(util.Exists("TestMonitor_Update"))