From 9f7f18c2c471ff00ad14205ea42d1cca9b1dffb3 Mon Sep 17 00:00:00 2001
From: Jiajie Chen <jiegec@qq.com>
Date: Thu, 10 Sep 2020 21:58:31 +0800
Subject: [PATCH] Fix missing method in mock test

---
 manager/server_test.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/manager/server_test.go b/manager/server_test.go
index 1db70de..7fc1432 100644
--- a/manager/server_test.go
+++ b/manager/server_test.go
@@ -462,6 +462,15 @@ func (b *mockDBAdapter) CreateWorker(w WorkerStatus) (WorkerStatus, error) {
 	return w, nil
 }
 
+func (b *mockDBAdapter) RefreshWorker(workerID string) (w WorkerStatus, err error) {
+	w, err = b.GetWorker(workerID)
+	if err == nil {
+		w.LastOnline = time.Now()
+		w, err = b.CreateWorker(w)
+	}
+	return w, err
+}
+
 func (b *mockDBAdapter) GetMirrorStatus(workerID, mirrorID string) (MirrorStatus, error) {
 	id := mirrorID + "/" + workerID
 	status, ok := b.statusStore[id]