TESTS="memory_basic memory_cluster_nodememory "\
"memory_queue_nodememory memory_cluster_and_queue_nodememory_1 "\
"memory_cluster_and_queue_nodememory_2"

function test_memory_basic() {
read -r -d '' job_description_input <<'EOF'
&(executable = "/bin/true")
 (memory = 100)
EOF

read -r -d '' job_script_patch <<'EOF'
--- a/basic-script.sh 2014-01-28 14:14:58.640281603 +0100
+++ b/basic-script.sh 2014-01-28 14:14:58.636281603 +0100
@@ -8,6 +8,7 @@
 #SBATCH --get-user-env=10L
 #SBATCH -n 1
 #SBATCH 
+#SBATCH --mem-per-cpu=100
 
 # Overide umask of execution node (sometime values are really strange)
 umask 077
EOF
echo "${job_script_patch}" | patch -sf -p1 -d ${SRCDIR} -o ${testdir}/expected_lrms_job_script.tmpl
}

function test_memory_cluster_nodememory() {
read -r -d '' arc_test_configuration <<'EOF'
[lrms]
defaultmemory=300
EOF

read -r -d '' job_description_input <<'EOF'
&(executable = "/bin/true")
EOF

read -r -d '' job_script_patch <<'EOF'
--- a/basic-script.sh 2014-01-28 14:13:39.296280413 +0100
+++ b/basic-script.sh 2014-01-28 14:13:39.288280412 +0100
@@ -8,6 +8,7 @@
 #SBATCH --get-user-env=10L
 #SBATCH -n 1
 #SBATCH 
+#SBATCH --mem-per-cpu=300
 
 # Overide umask of execution node (sometime values are really strange)
 umask 077
EOF
echo "${job_script_patch}" | patch -sf -p1 -d ${SRCDIR} -o ${testdir}/expected_lrms_job_script.tmpl
}

function test_memory_queue_nodememory() {
read -r -d '' arc_test_configuration <<'EOF'
[queue:nordugrid]
defaultmemory=200
EOF

read -r -d '' job_description_input <<'EOF'
&(executable = "/bin/true")
 (queue = "nordugrid")
EOF

read -r -d '' job_script_patch <<'EOF'
--- a/basic-script.sh 2014-01-28 14:10:18.764277404 +0100
+++ b/basic-script.sh 2014-01-28 14:10:18.756277403 +0100
@@ -4,11 +4,13 @@
 #SBATCH -e /my/session/directory.comment
 #SBATCH -o /my/session/directory.comment
 
+#SBATCH -p nordugrid
 #SBATCH --nice=50
 #SBATCH -J 'gridjob'
 #SBATCH --get-user-env=10L
 #SBATCH -n 1
 #SBATCH 
+#SBATCH --mem-per-cpu=200
 
 # Overide umask of execution node (sometime values are really strange)
 umask 077
EOF
echo "${job_script_patch}" | patch -sf -p1 -d ${SRCDIR} -o ${testdir}/expected_lrms_job_script.tmpl
}

function test_memory_cluster_and_queue_nodememory_1() {
read -r -d '' arc_test_configuration <<'EOF'
[lrms]
defaultmemory=200

[queue:nordugrid]
defaultmemory=700
EOF

read -r -d '' job_description_input <<'EOF'
&(executable = "/bin/true")
 (queue = "nordugrid")
EOF

read -r -d '' job_script_patch <<'EOF'
--- a/basic-script.sh 2014-01-28 14:08:57.116276178 +0100
+++ b/basic-script.sh 2014-01-28 14:08:57.116276178 +0100
@@ -4,10 +4,12 @@
 #SBATCH -e /my/session/directory.comment
 #SBATCH -o /my/session/directory.comment
 
+#SBATCH -p nordugrid
 #SBATCH --nice=50
 #SBATCH -J 'gridjob'
 #SBATCH --get-user-env=10L
 #SBATCH -n 1
 #SBATCH 
+#SBATCH --mem-per-cpu=700
 
 # Overide umask of execution node (sometime values are really strange)
EOF
echo "${job_script_patch}" | patch -sf -p1 -d ${SRCDIR} -o ${testdir}/expected_lrms_job_script.tmpl
}

function test_memory_cluster_and_queue_nodememory_2() {
read -r -d '' arc_test_configuration <<'EOF'
[lrms]
defaultmemory=200

[queue:nordugrid]
defaultmemory=700
EOF

read -r -d '' job_description_input <<'EOF'
&(executable = "/bin/true")
EOF

read -r -d '' job_script_patch <<'EOF'
--- a/basic-script.sh 2014-01-28 14:06:57.320274381 +0100
+++ b/basic-script.sh 2014-01-28 14:06:57.308274381 +0100
@@ -8,6 +8,7 @@
 #SBATCH --get-user-env=10L
 #SBATCH -n 1
 #SBATCH 
+#SBATCH --mem-per-cpu=200
 
 # Overide umask of execution node (sometime values are really strange)
 umask 077
EOF
echo "${job_script_patch}" | patch -sf -p1 -d ${SRCDIR} -o ${testdir}/expected_lrms_job_script.tmpl
}
