|
@@ -5,6 +5,7 @@ import (
|
|
|
|
|
|
composetypes "github.com/aanand/compose-file/types"
|
|
|
"github.com/docker/docker/api/types/mount"
|
|
|
+ "github.com/docker/docker/api/types/swarm"
|
|
|
"github.com/docker/docker/pkg/testutil/assert"
|
|
|
)
|
|
|
|
|
@@ -28,3 +29,26 @@ func TestConvertVolumeToMountInvalidFormat(t *testing.T) {
|
|
|
assert.Error(t, err, "invalid volume: "+vol)
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+func TestConvertResourcesOnlyMemory(t *testing.T) {
|
|
|
+ source := composetypes.Resources{
|
|
|
+ Limits: &composetypes.Resource{
|
|
|
+ MemoryBytes: composetypes.UnitBytes(300000000),
|
|
|
+ },
|
|
|
+ Reservations: &composetypes.Resource{
|
|
|
+ MemoryBytes: composetypes.UnitBytes(200000000),
|
|
|
+ },
|
|
|
+ }
|
|
|
+ resources, err := convertResources(source)
|
|
|
+ assert.NilError(t, err)
|
|
|
+
|
|
|
+ expected := &swarm.ResourceRequirements{
|
|
|
+ Limits: &swarm.Resources{
|
|
|
+ MemoryBytes: 300000000,
|
|
|
+ },
|
|
|
+ Reservations: &swarm.Resources{
|
|
|
+ MemoryBytes: 200000000,
|
|
|
+ },
|
|
|
+ }
|
|
|
+ assert.DeepEqual(t, resources, expected)
|
|
|
+}
|