From d7a28d12abb40a2334309c955432f75880a0c0c6 Mon Sep 17 00:00:00 2001
From: Lorenz Bateman <lorenz.bateman@zalando.de>
Date: Tue, 20 Jun 2017 12:35:50 +0200
Subject: [PATCH] Update test case for instance type modification

---
 tests/test_update_cluster.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_update_cluster.py b/tests/test_update_cluster.py
index 3fc6874..13f70c2 100644
--- a/tests/test_update_cluster.py
+++ b/tests/test_update_cluster.py
@@ -51,14 +51,15 @@ def test_build_run_instances_params():
     options = {
         'cluster_name': 'my-cluster-name',
         'docker_image': 'docker.registry/cassandra:123',
-        'taupage_ami_id': 'ami-654321'
+        'taupage_ami_id': 'ami-654321',
+        'instance_type': 'm4.xlarge'
     }
     expected = {
         'MinCount': 1,
         'MaxCount': 1,
         'ImageId': 'ami-654321',
         'SecurityGroupIds': ['sg-abcdef', 'sg-654321'],
-        'InstanceType': 't2.micro',
+        'InstanceType': 'm4.xlarge',
         'SubnetId': 'sn-123',
         'PrivateIpAddress': '172.31.128.11',
         'BlockDeviceMappings': [],