From 52d68913e5893e79ee871250e7d9d3e352d2e8b7 Mon Sep 17 00:00:00 2001
From: Greg Anderson <greg.1.anderson@greenknowe.org>
Date: Fri, 13 Oct 2017 17:25:21 -0700
Subject: [PATCH] When creating the sut, also create aliases @sut.dev and
 @sut.stage.

---
 sut           | 16 +++++++++++-----
 unish.sut.php | 14 +++++++++++++-
 2 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/sut b/sut
index 9b792321ff..88c1ecc561 100755
--- a/sut
+++ b/sut
@@ -16,14 +16,20 @@ $vendor = $sut . '/vendor';
 $arguments = $GLOBALS['argv'];
 // Shift off argv[0] which contains the name of this script.
 array_shift($arguments);
-// Add uri option
-array_unshift($arguments, "--uri=dev");
+// Add alias if not already specified
+if (!empty($arguments) && ($arguments[0][0] != '@')) {
+  array_unshift($arguments, "@sut.dev");
+}
 
-// Make it easy to just call `/.sut si -y testing`
+// Make it easy to just call `./sut si -y testing`, or `./sut @sut.stage -y testing`
 if (in_array('si', $arguments) || in_array('site-install', $arguments)) {
+    $subdir = 'dev';
+    if (in_array('@sut.stage', $arguments)) {
+      $subdir = 'stage';
+    }
     $db_url = getenv('UNISH_DB_URL') ?: 'mysql://root:@127.0.0.1';
-    $arguments[] = "--db-url=$db_url/unish_dev";
-    $arguments[] = '--sites-subdir=dev';
+    $arguments[] = "--db-url=$db_url/unish_$subdir";
+    $arguments[] = "--sites-subdir=$subdir";
 }
 
 /**
diff --git a/unish.sut.php b/unish.sut.php
index d7e45a196c..b902e5771e 100755
--- a/unish.sut.php
+++ b/unish.sut.php
@@ -28,6 +28,7 @@ function unish_validate() {
  */
 function unish_setup_sut($unish_sandbox) {
   $working_dir = dirname($unish_sandbox) . DIRECTORY_SEPARATOR . 'build-drush-sut';
+  $target_dir = dirname($working_dir) . DIRECTORY_SEPARATOR . 'drush-sut';
   drush_delete_dir($working_dir, TRUE);
   $codebase = 'tests/resources/codebase';
   drush_copy_dir($codebase, $working_dir);
@@ -41,6 +42,18 @@ function unish_setup_sut($unish_sandbox) {
     }
   }
 
+  $alias_contents = <<<EOT
+dev:
+  root: $target_dir/web
+  uri: dev
+stage:
+  root: $target_dir/web
+  uri: stage
+EOT;
+  mkdir("$working_dir/drush");
+  mkdir("$working_dir/drush/site-aliases");
+  file_put_contents("$working_dir/drush/site-aliases/sut.alias.yml", $alias_contents);
+
   $verbose = unishIsVerbose();
   $cmd = "composer $verbose install --prefer-dist --no-progress --no-suggest --working-dir " . escapeshellarg($working_dir);
   fwrite(STDERR, 'Executing: ' . $cmd . "\n");
@@ -54,7 +67,6 @@ function unish_setup_sut($unish_sandbox) {
   }
 
   // Move the sut into place
-  $target_dir = dirname($working_dir) . DIRECTORY_SEPARATOR . 'drush-sut';
   drush_delete_dir($target_dir, TRUE);
   rename($working_dir, $target_dir);