From 47eb206c6a410a7524806bbd9687a5f88de066d3 Mon Sep 17 00:00:00 2001
From: Bruce Campbell <v-brucc@microsoft.com>
Date: Thu, 1 Feb 2018 11:23:15 -0800
Subject: [PATCH] Add additional v3 vm sizes permitted for dcos master and
 agent

---
 pkg/acsengine/Get-AzureConstants.py |  2 +-
 pkg/acsengine/azureconst.go         | 39 +++++++++++++++++++++++++++++
 2 files changed, 40 insertions(+), 1 deletion(-)

diff --git a/pkg/acsengine/Get-AzureConstants.py b/pkg/acsengine/Get-AzureConstants.py
index e632ba56e4..1652ee5b83 100644
--- a/pkg/acsengine/Get-AzureConstants.py
+++ b/pkg/acsengine/Get-AzureConstants.py
@@ -20,7 +20,7 @@ def getAllSizes():
 
 min_cores_dcos = 2
 min_cores_k8s = 1
-dcos_masters_ephemeral_disk_min = 102400
+dcos_masters_ephemeral_disk_min = 16384
 
 def getDcosMasterMap(sizeMap):
     masterMap = {}
diff --git a/pkg/acsengine/azureconst.go b/pkg/acsengine/azureconst.go
index b66c864bc3..ca01a6bf19 100644
--- a/pkg/acsengine/azureconst.go
+++ b/pkg/acsengine/azureconst.go
@@ -53,13 +53,22 @@ func GetDCOSMasterAllowedSizes() string {
         "Standard_A10",
         "Standard_A11",
         "Standard_A2",
+        "Standard_A2_v2",
+        "Standard_A2m_v2",
         "Standard_A3",
         "Standard_A4",
+        "Standard_A4_v2",
+        "Standard_A4m_v2",
         "Standard_A5",
         "Standard_A6",
         "Standard_A7",
         "Standard_A8",
+        "Standard_A8_v2",
+        "Standard_A8m_v2",
         "Standard_A9",
+        "Standard_B2ms",
+        "Standard_B4ms",
+        "Standard_B8ms",
         "Standard_D11",
         "Standard_D11_v2",
         "Standard_D11_v2_Promo",
@@ -78,6 +87,8 @@ func GetDCOSMasterAllowedSizes() string {
         "Standard_D2",
         "Standard_D2_v2",
         "Standard_D2_v2_Promo",
+        "Standard_D2_v3",
+        "Standard_D2s_v3",
         "Standard_D3",
         "Standard_D32_v3",
         "Standard_D32s_v3",
@@ -87,11 +98,19 @@ func GetDCOSMasterAllowedSizes() string {
         "Standard_D4_v2",
         "Standard_D4_v2_Promo",
         "Standard_D4_v3",
+        "Standard_D2s_v3",
         "Standard_D5_v2",
         "Standard_D5_v2_Promo",
         "Standard_D64_v3",
         "Standard_D64s_v3",
         "Standard_D8_v3",
+        "Standard_D8s_v3",
+        "Standard_DS11",
+        "Standard_DS11_v2",
+        "Standard_DS11_v2_Promo",
+        "Standard_DS12",
+        "Standard_DS12_v2",
+        "Standard_DS12_v2_Promo",
         "Standard_DS13",
         "Standard_DS13-2_v2",
         "Standard_DS13-4_v2",
@@ -103,15 +122,24 @@ func GetDCOSMasterAllowedSizes() string {
         "Standard_DS14_v2",
         "Standard_DS14_v2_Promo",
         "Standard_DS15_v2",
+        "Standard_DS3",
+        "Standard_DS3_v2",
+        "Standard_DS3_v2_Promo",
+        "Standard_DS4",
+        "Standard_DS4_v2",
+        "Standard_DS4_v2_Promo",
         "Standard_DS5_v2",
         "Standard_DS5_v2_Promo",
         "Standard_E16_v3",
         "Standard_E16s_v3",
+        "Standard_E2_v3",
+        "Standard_E2s_v3",
         "Standard_E32-16s_v3",
         "Standard_E32-8s_v3",
         "Standard_E32_v3",
         "Standard_E32s_v3",
         "Standard_E4_v3",
+        "Standard_E4s_v3",
         "Standard_E64-16s_v3",
         "Standard_E64-32s_v3",
         "Standard_E64_v3",
@@ -119,16 +147,25 @@ func GetDCOSMasterAllowedSizes() string {
         "Standard_E8_v3",
         "Standard_E8s_v3",
         "Standard_F16",
+        "Standard_F16s",
         "Standard_F16s_v2",
+        "Standard_F2",
+        "Standard_F2s_v2",
         "Standard_F32s_v2",
+        "Standard_F4",
+        "Standard_F4s",
+        "Standard_F4s_v2",
         "Standard_F64s_v2",
         "Standard_F72s_v2",
         "Standard_F8",
+        "Standard_F8s",
+        "Standard_F8s_v2",
         "Standard_G1",
         "Standard_G2",
         "Standard_G3",
         "Standard_G4",
         "Standard_G5",
+        "Standard_GS1",
         "Standard_GS2",
         "Standard_GS3",
         "Standard_GS4",
@@ -266,12 +303,14 @@ func GetMasterAgentAllowedSizes() string {
         "Standard_DS2",
         "Standard_DS2_v2",
         "Standard_DS2_v2_Promo",
+        "Standard_D2s_v3",
         "Standard_DS3",
         "Standard_DS3_v2",
         "Standard_DS3_v2_Promo",
         "Standard_DS4",
         "Standard_DS4_v2",
         "Standard_DS4_v2_Promo",
+        "Standard_D2s_v3",
         "Standard_DS5_v2",
         "Standard_DS5_v2_Promo",
         "Standard_E16_v3",