feature: add mock mgr for the convenience of testing

Signed-off-by: Starnop <starnop@163.com>
dragonflyoss · Apr 18, 2019 · 3c6bfd3 · 3c6bfd3
1 parent 20f8a4a
commit 3c6bfd3
Show file tree

Hide file tree

Showing 12 changed files with 1,688 additions and 0 deletions.
diff --git a/hack/generate-super-mock-mgr.sh b/hack/generate-super-mock-mgr.sh
@@ -0,0 +1,25 @@
+#!/bin/bash
+set -e
+
+# This script is used to generate mock files for interfaces.
+
+# Get the absolute path of this file
+DIR="$( cd "$( dirname "$0"  )" && pwd  )"/..
+cd "$DIR"
+
+# install mockgen if it not exists
+mockgen -h >/dev/null 2>&1 || go get github.com/golang/mock/mockgen
+# install goimports if it not exists
+goimports -h >/dev/null 2>&1 || go get golang.org/x/tools/cmd/goimports
+
+
+# generate mock files for supernode mgr interfaces.
+MGR_ARRAY=("cdn_mgr" "dfget_task_mgr" "peer_mgr" "progress_mgr" "scheduler_mgr")
+for name in "${MGR_ARRAY[@]}"
+do
+	mockgen -destination "./supernode/daemon/mgr/mock/mock_$name.go" -source "supernode/daemon/mgr/$name.go" -package mock
+	goimports -w --local "github.com/dragonflyoss/Dragonfly" "./supernode/daemon/mgr/mock"
+	echo "Mock file ./supernode/daemon/mgr/mock/mock_$name.go generated successfully for $name"
+done
+
+
diff --git a/supernode/daemon/mgr/mock/mock_cdn_mgr.go b/supernode/daemon/mgr/mock/mock_cdn_mgr.go
diff --git a/supernode/daemon/mgr/mock/mock_dfget_task_mgr.go b/supernode/daemon/mgr/mock/mock_dfget_task_mgr.go
diff --git a/supernode/daemon/mgr/mock/mock_peer_mgr.go b/supernode/daemon/mgr/mock/mock_peer_mgr.go