1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
import boto3
from moto import mock_aws
from .test_eks_constants import NODEROLE_ARN_VALUE, SUBNET_IDS
@mock_aws
def test_passing_an_unknown_launchtemplate_is_supported():
eks = boto3.client("eks", "us-east-2")
eks.create_cluster(name="a", roleArn=NODEROLE_ARN_VALUE, resourcesVpcConfig={})
group = eks.create_nodegroup(
clusterName="a",
nodegroupName="b",
launchTemplate={"name": "random"},
nodeRole=NODEROLE_ARN_VALUE,
subnets=SUBNET_IDS,
)["nodegroup"]
assert group["launchTemplate"] == {"name": "random"}
@mock_aws
def test_passing_a_known_launchtemplate_by_name():
ec2 = boto3.client("ec2", region_name="us-east-2")
eks = boto3.client("eks", "us-east-2")
lt_id = ec2.create_launch_template(
LaunchTemplateName="ltn",
LaunchTemplateData={
"TagSpecifications": [
{"ResourceType": "instance", "Tags": [{"Key": "t", "Value": "v"}]}
]
},
)["LaunchTemplate"]["LaunchTemplateId"]
eks.create_cluster(name="a", roleArn=NODEROLE_ARN_VALUE, resourcesVpcConfig={})
group = eks.create_nodegroup(
clusterName="a",
nodegroupName="b",
launchTemplate={"name": "ltn"},
nodeRole=NODEROLE_ARN_VALUE,
subnets=SUBNET_IDS,
)["nodegroup"]
assert group["launchTemplate"] == {"name": "ltn", "id": lt_id}
@mock_aws
def test_passing_a_known_launchtemplate_by_id():
ec2 = boto3.client("ec2", region_name="us-east-2")
eks = boto3.client("eks", "us-east-2")
lt_id = ec2.create_launch_template(
LaunchTemplateName="ltn",
LaunchTemplateData={
"TagSpecifications": [
{"ResourceType": "instance", "Tags": [{"Key": "t", "Value": "v"}]}
]
},
)["LaunchTemplate"]["LaunchTemplateId"]
eks.create_cluster(name="a", roleArn=NODEROLE_ARN_VALUE, resourcesVpcConfig={})
group = eks.create_nodegroup(
clusterName="a",
nodegroupName="b",
launchTemplate={"id": lt_id},
nodeRole=NODEROLE_ARN_VALUE,
subnets=SUBNET_IDS,
)["nodegroup"]
assert group["launchTemplate"] == {"name": "ltn", "id": lt_id}
|
