#!/bin/bash

# to make the script fail safely

set -euo pipefail

out=${1:-output}

echo "" > $out

echo "Below is the feedback for your submission for alcohol.scala" >> $out

echo "" >> $out

# compilation tests

function scala_compile {

  (ulimit -t 30 -m 1024000 ; scala "$1" 2>> $out 1>> $out) 

}

# functional tests

function scala_assert {

  (ulimit -t 30 -m 1024000 ; scala -i "$1" "$2" -e "" 2> /dev/null 1> /dev/null)

}

# purity test

function scala_vars {

   (egrep '\bvar\b|\breturn\b|\.par|ListBuffer|mutable' "$1" 2> /dev/null 1> /dev/null)

}

# var, .par return, ListBuffer test

#

echo "alcohol.scala does not contain vars, returns etc?" >> $out

if (scala_vars alcohol.scala)

then

  echo "  --> fail" >> $out

  tsts0=$(( 1 ))

else

  echo "  --> success" >> $out

  tsts0=$(( 0 )) 

fi

### compilation test

if  [ $tsts0 -eq 0 ]

then 

  echo "alcohol.scala runs?" >> $out

  if (scala_compile alcohol.scala)

  then

    echo "  --> success" >> $out

    tsts=$(( 0 ))

  else

    echo "  --> scala did not run alcohol.scala" >> $out

    tsts=$(( 1 )) 

  fi

else

  tsts=$(( 1 ))     

fi

### alcohol tests

if [ $tsts -eq 0 ]

then

  echo "get_csv tests:" >> $out

  echo "  get_csv_page(url_alcohol).size == 194" >> $out

  echo "  get_csv_file(file_population).size == 216" >> $out

  if (scala_assert "alcohol.scala" "alcohol_test1.scala")

  then

    echo "  --> success" >> $out

  else

    echo "  --> one of the tests failed" >> $out

  fi

fi

### alcohol processing tests

if [ $tsts -eq 0 ]

then

  echo "processing tests:" >> $out  

  echo "  process_alcs(alcs_list.drop(1))(0) == (\"Afghanistan\", 0.0)" >> $out

  echo "  process_pops(pops_list.drop(1))(\"Micronesia\") == 104015" >> $out

  if (scala_assert "alcohol.scala" "alcohol_test2.scala") 

  then

    echo "  --> success" >> $out

  else

    echo "  --> one of the tests failed" >> $out

  fi

fi

### alcohol percentage tests

if [ $tsts -eq 0 ]

then

  echo "calculation tests:" >> $out  

  echo "  sorted_country_consumption().size == 177" >> $out

  echo "  sorted_country_consumption()(9) == (\"United Kingdom\", 671976864)" >> $out

  echo "  percentage(164) == (28771558364L, 28771558364L, 100.0)" >> $out

  if (scala_assert "alcohol.scala" "alcohol_test3.scala") 

  then

    echo "  --> success" >> $out

  else

    echo "  --> one of the tests failed" >> $out

  fi

fi