Add tests (and correct existing test)

nextflow-io · Jul 26, 2024 · 316bd84 · 316bd84
1 parent 8beef54
commit 316bd84
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/plugins/nf-amazon/src/test/nextflow/cloud/aws/batch/AwsContainerOptionsMapperTest.groovy b/plugins/nf-amazon/src/test/nextflow/cloud/aws/batch/AwsContainerOptionsMapperTest.groovy
@@ -94,7 +94,25 @@ class AwsContainerOptionsMapperTest extends Specification {
         def map = CmdLineHelper.parseGnuArgs('--shm-size 12048024')
         def properties = AwsContainerOptionsMapper.createContainerProperties(map)
         then:
-        properties.getLinuxParameters().getSharedMemorySize() == 12048024
+        properties.getLinuxParameters().getSharedMemorySize() == 11
+    }
+
+    def 'should set shared memory size with unit in MiB'() {
+
+        when:
+        def map = CmdLineHelper.parseGnuArgs('--shm-size 256m')
+        def properties = AwsContainerOptionsMapper.createContainerProperties(map)
+        then:
+        properties.getLinuxParameters().getSharedMemorySize() == 256
+    }
+
+    def 'should set shared memory size with unit in GiB'() {
+
+        when:
+        def map = CmdLineHelper.parseGnuArgs('--shm-size 1g')
+        def properties = AwsContainerOptionsMapper.createContainerProperties(map)
+        then:
+        properties.getLinuxParameters().getSharedMemorySize() == 1024
     }
 
     def 'should set memory swappiness'() {