From f31077898ec86cbce6eac9ecb015197134769274 Mon Sep 17 00:00:00 2001
From: Jiaxun Song <jiaxun.song@outlook.com>
Date: Fri, 13 Nov 2020 22:45:50 +0000
Subject: [PATCH] azurestack

---
 Makefile                     |  6 ++-
 test/integration/README.md   |  8 +++
 test/integration/params.yaml |  3 ++
 test/integration/run-test.sh | 95 ++++++++++++++++++++++++++++++++++++
 test/sanity/run-test.sh      |  2 +
 5 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 test/integration/README.md
 create mode 100644 test/integration/params.yaml
 create mode 100644 test/integration/run-test.sh

diff --git a/Makefile b/Makefile
index 4a99c2f7..062b6c11 100644
--- a/Makefile
+++ b/Makefile
@@ -44,6 +44,10 @@ all: nfs
 sanity-test: nfs
 	./test/sanity/run-test.sh
 
+.PHONY: integration-test
+integration-test: nfs
+	./test/integration/run-test.sh
+
 .PHONY: local-build-push
 local-build-push: nfs
 	docker build -t $(LOCAL_USER)/nfsplugin:latest .
@@ -69,7 +73,7 @@ local-k8s-uninstall:
 
 .PHONY: nfs
 nfs:
-	CGO_ENABLED=0 GOOS=linux go build -a -ldflags ${LDFLAGS} -o bin/nfsplugin ./cmd/nfsplugin
+	CGO_ENABLED=0 GOOS=linux go build -a -ldflags ${LDFLAGS} -mod vendor -o bin/nfsplugin ./cmd/nfsplugin
 
 .PHONY: container
 container: nfs
diff --git a/test/integration/README.md b/test/integration/README.md
new file mode 100644
index 00000000..e74a1e42
--- /dev/null
+++ b/test/integration/README.md
@@ -0,0 +1,8 @@
+## Sanity Tests
+Testing the NFS CSI driver using the [`sanity`](https://github.com/kubernetes-csi/csi-test/tree/master/pkg/sanity) package test suite.
+
+### Run sanity tests
+```
+make sanity-test
+```
+
diff --git a/test/integration/params.yaml b/test/integration/params.yaml
new file mode 100644
index 00000000..d50650b7
--- /dev/null
+++ b/test/integration/params.yaml
@@ -0,0 +1,3 @@
+  
+server: 127.0.0.1
+share: /
\ No newline at end of file
diff --git a/test/integration/run-test.sh b/test/integration/run-test.sh
new file mode 100644
index 00000000..552d31dc
--- /dev/null
+++ b/test/integration/run-test.sh
@@ -0,0 +1,95 @@
+#!/bin/bash
+
+# Copyright 2020 The Kubernetes Authors.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+set -eo pipefail
+
+function cleanup {
+  echo 'pkill -f nfsplugin'
+  pkill -f nfsplugin
+  echo 'Uninstalling NFS server on localhost'
+  docker rm nfs -f
+}
+trap cleanup EXIT
+
+function install_csc_bin {
+  echo 'Installing CSC binary...'
+  export set GOPATH=$HOME/go
+  export set PATH=$PATH:$GOPATH/bin
+  go get github.com/rexray/gocsi/csc
+  pushd $GOPATH/src/github.com/rexray/gocsi/csc
+  make build
+  popd
+}
+
+function provision_nfs_server {
+  echo 'Installing NFS server on localhost'
+  apt-get update -y
+  apt-get install -y nfs-common
+  docker run -d --name nfs --privileged -p 2049:2049 -v $(pwd):/nfsshare -e SHARED_DIRECTORY=/nfsshare itsthenetwork/nfs-server-alpine:latest
+}
+
+provision_nfs_server
+install_csc_bin
+
+readonly volname="citest-$(date +%s)"
+readonly volsize="2147483648"
+readonly endpoint='unix:///tmp/csi.sock'
+readonly staging_target_path="/tmp/stagingtargetpath"
+readonly target_path="/tmp/targetpath"
+
+nodeid='CSINode'
+if [[ "$#" -gt 0 ]] && [[ -n "$1" ]]; then
+  nodeid="$1"
+fi
+
+bin/nfsplugin --endpoint "$endpoint" --nodeid "$nodeid" -v=5 &
+
+echo 'Begin to run integration test...'
+
+# Begin to run CSI functions one by one
+echo "Create volume test:"
+value="$(csc controller new --endpoint "$endpoint" --cap 1,block "$volname" --req-bytes "$volsize")"
+sleep 15
+
+volumeid="$(echo "$value" | awk '{print $1}' | sed 's/"//g')"
+echo "Got volume id: $volumeid"
+storage_account_name="$(echo "$volumeid" | awk -F# '{print $2}')"
+
+csc controller validate-volume-capabilities --endpoint "$endpoint" --cap 1,block "$volumeid"
+
+echo "stage volume test:"
+csc node stage --endpoint "$endpoint" --cap 1,block --staging-target-path "$staging_target_path" "$volumeid"
+
+echo "publish volume test:"
+csc node publish --endpoint "$endpoint" --cap 1,block --staging-target-path "$staging_target_path" --target-path "$target_path" "$volumeid"
+sleep 2
+
+echo "unpublish volume test:"
+csc node unpublish --endpoint "$endpoint" --target-path "$target_path" "$volumeid"
+sleep 2
+
+echo "unstage volume test:"
+csc node unstage --endpoint "$endpoint" --staging-target-path "$staging_target_path" "$volumeid"
+sleep 2
+
+echo "Delete volume test:"
+csc controller del --endpoint "$endpoint" "$volumeid"
+sleep 15
+
+csc identity plugin-info --endpoint "$endpoint"
+csc node get-info --endpoint "$endpoint"
+
+echo "Integration test is completed."
diff --git a/test/sanity/run-test.sh b/test/sanity/run-test.sh
index 9bf2ea58..2d3778ed 100755
--- a/test/sanity/run-test.sh
+++ b/test/sanity/run-test.sh
@@ -21,6 +21,8 @@ function cleanup {
   pkill -f nfsplugin
   echo 'Deleting CSI sanity test binary'
   rm -rf csi-test
+  echo 'Uninstalling NFS server on localhost'
+  docker rm nfs -f
 }
 trap cleanup EXIT