mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
2d8fcd9bf4
Co-authored-by: Malte Poll <mp@edgeless.systems> Co-authored-by: katexochen <katexochen@users.noreply.github.com> Co-authored-by: Daniel Weiße <dw@edgeless.systems> Co-authored-by: Thomas Tendyck <tt@edgeless.systems> Co-authored-by: Benedict Schlueter <bs@edgeless.systems> Co-authored-by: leongross <leon.gross@rub.de> Co-authored-by: Moritz Eckert <m1gh7ym0@gmail.com>
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package ec2
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestIDs(t *testing.T) {
|
|
assert := assert.New(t)
|
|
testState := testInstances()
|
|
expectedIDs := []string{"id-9", "id-10", "id-11", "id-12"}
|
|
assert.ElementsMatch(expectedIDs, testState.IDs())
|
|
}
|
|
|
|
func TestPublicIPs(t *testing.T) {
|
|
assert := assert.New(t)
|
|
testState := testInstances()
|
|
expectedIPs := []string{"192.0.2.1", "192.0.2.3", "192.0.2.5", "192.0.2.7"}
|
|
assert.ElementsMatch(expectedIPs, testState.PublicIPs())
|
|
}
|
|
|
|
func TestPrivateIPs(t *testing.T) {
|
|
assert := assert.New(t)
|
|
testState := testInstances()
|
|
expectedIPs := []string{"192.0.2.2", "192.0.2.4", "192.0.2.6", "192.0.2.8"}
|
|
assert.ElementsMatch(expectedIPs, testState.PrivateIPs())
|
|
}
|
|
|
|
func TestGetOne(t *testing.T) {
|
|
assert := assert.New(t)
|
|
testState := testInstances()
|
|
id, instance, err := testState.GetOne()
|
|
assert.NoError(err)
|
|
assert.Contains(testState, id)
|
|
assert.Equal(testState[id], instance)
|
|
}
|
|
|
|
func TestGetOthers(t *testing.T) {
|
|
assert := assert.New(t)
|
|
testCases := testInstances().IDs()
|
|
|
|
for _, id := range testCases {
|
|
others := testInstances().GetOthers(id)
|
|
assert.NotContains(others, id)
|
|
expectedInstances := testInstances()
|
|
delete(expectedInstances, id)
|
|
assert.ElementsMatch(others.IDs(), expectedInstances.IDs())
|
|
}
|
|
}
|
|
|
|
func testInstances() Instances {
|
|
return Instances{
|
|
"id-9": {
|
|
PublicIP: "192.0.2.1",
|
|
PrivateIP: "192.0.2.2",
|
|
},
|
|
"id-10": {
|
|
PublicIP: "192.0.2.3",
|
|
PrivateIP: "192.0.2.4",
|
|
},
|
|
"id-11": {
|
|
PublicIP: "192.0.2.5",
|
|
PrivateIP: "192.0.2.6",
|
|
},
|
|
"id-12": {
|
|
PublicIP: "192.0.2.7",
|
|
PrivateIP: "192.0.2.8",
|
|
},
|
|
}
|
|
}
|