From 82adc85562603c00fa466f65d90b9247129dcd26 Mon Sep 17 00:00:00 2001
From: Adam Harrison <harrisonadamw@gmail.com>
Date: Tue, 8 Mar 2016 16:03:37 +0000
Subject: [PATCH] Router and proxy run with --restart=always

---
 test/090_docker_restart_policy_2_test.sh | 34 ++++++++++++++++++++++++
 weave                                    |  2 ++
 2 files changed, 36 insertions(+)
 create mode 100755 test/090_docker_restart_policy_2_test.sh

diff --git a/test/090_docker_restart_policy_2_test.sh b/test/090_docker_restart_policy_2_test.sh
new file mode 100755
index 00000000..b8b9a5a7
--- /dev/null
+++ b/test/090_docker_restart_policy_2_test.sh
@@ -0,0 +1,34 @@
+#! /bin/bash
+
+. ./config.sh
+
+start_suite "Test docker restart policy"
+
+pidof() {
+    docker_on $1 inspect -f '{{.State.Pid}}' $2
+}
+
+check_restart() {
+    OLD_PID=$(pidof $1 $2)
+
+    run_on $1 sudo kill $OLD_PID
+
+    for i in $(seq 1 30); do
+        NEW_PID=$(pidof $1 $2)
+
+        if [ $NEW_PID != 0 -a $NEW_PID != $OLD_PID ] ; then
+            return 0
+        fi
+
+        sleep 1
+    done
+
+    return 1
+}
+
+weave_on $HOST1 launch
+
+assert_raises "check_restart $HOST1 weave"
+assert_raises "check_restart $HOST1 weaveproxy"
+
+end_suite
diff --git a/weave b/weave
index a307f830..79a17d72 100755
--- a/weave
+++ b/weave
@@ -1627,6 +1627,7 @@ launch_router() {
     # when launching the weave container.
     start_if_necessary router $CONTAINER_NAME $IMAGE \
         $(docker_run_options) \
+        --restart=always \
         -e WEAVE_PASSWORD \
         -e CHECKPOINT_DISABLE \
         $WEAVE_DOCKER_ARGS $IMAGE $COVERAGE_ARGS \
@@ -1683,6 +1684,7 @@ launch_proxy() {
     fi
     start_if_necessary proxy $PROXY_CONTAINER_NAME $EXEC_IMAGE \
         $(docker_run_options) \
+        --restart=always \
         --pid=host \
         $PROXY_VOLUMES \
         --volumes-from $VOLUMES_CONTAINER_NAME \