def test__load_params_kmskeyid_without_encrypted():
"""
Test that "KmsKeyId" is not added to params when encrypted is not passed
"""
kwargs = {"zone": "us-west-2", "kmskeyid": "keyid"}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected
def test__load_params_encrypted():
"""
Test that "Encrypted" is added to params
"""
kwargs = {"zone": "us-west-2", "encrypted": True}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
"Encrypted": True,
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected
def test__load_params_size():
"""
Test that "Size" is added to params
"""
kwargs = {"zone": "us-west-2", "size": 10}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
"Size": 10
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected
def test__load_params_type():
"""
Test that "VolumeType" is added to params
"""
kwargs = {"zone": "us-west-2", "type": "gp2"}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
"VolumeType": "gp2",
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected
def test__load_params_snapshot():
"""
Test that "SnapshotID" is added to params
"""
kwargs = {"zone": "us-west-2", "snapshot": 1234}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
"SnapshotId": 1234,
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected
def test__load_params_type_io2():
"""
Test that "Iops" is added to params when type is io2
"""
kwargs = {"zone": "us-west-2", "type": "io2", "iops": 1000}
expected = {
"Action": "CreateVolume",
"AvailabilityZone": "us-west-2",
"VolumeType": "io2",
"Iops": 1000,
}
result = ec2._load_params(kwargs=kwargs)
assert result == expected