The String Calculator Kata in Cobol :)

string-calculator-kata-test.cbl

IDENTIFICATION DIVISION.
PROGRAM-ID. StringCalculatorTest.

DATA DIVISION.
  WORKING-STORAGE SECTION.
*> constants
    01 newline         PIC X   VALUE x'0a'.


*> add parameters
    01 inputString.
      02 prefix       PIC X(2).
      02 newDelimiter PIC X(1).
      02 FILLER       PIC X(99).
    01 result         PIC 9(10).

*> add working storage
    01 stringPointer  PIC 9(10).
    01 addend         PIC 9(10).
    01 theDelimiter   PIC X(1).

*> test working storage
    01 expectedResult PIC 9(10).

PROCEDURE DIVISION.

TestSuite.
  MOVE SPACE TO inputString.
  MOVE 0 TO expectedResult.
  PERFORM "testAdd".

  MOVE "1" TO inputString.
  MOVE 1 TO expectedResult.
  PERFORM "testAdd".

  MOVE "1,3" TO inputString.
  MOVE 4 TO expectedResult.
  PERFORM "testAdd".

  MOVE "1,2,7,33,44" TO inputString.
  MOVE 87 TO expectedResult.
  PERFORM "testAdd".

  MOVE SPACES TO inputString.
  STRING "1" newline "2,3" INTO inputString.
  MOVE 6 TO expectedResult.
  PERFORM "testAdd".

  MOVE SPACES TO inputString.
  STRING "//;" newline "1;2" INTO inputString.
  MOVE 3 TO expectedResult.
  PERFORM "testAdd".


  DISPLAY SPACE.
  GOBACK.

doAdd.
  MOVE 0 TO result.
  IF prefix = "//" THEN
    MOVE 5 TO stringPointer
    MOVE newDelimiter TO theDelimiter
  ELSE
    MOVE 1 TO stringPointer
    MOVE "," TO theDelimiter
  END-IF.
  PERFORM UNTIL stringPointer > LENGTH OF inputString
    UNSTRING inputString
      DELIMITED BY theDelimiter OR newline
      INTO addend
      WITH POINTER stringPointer
    ADD addend TO result
  END-PERFORM.

testAdd.
  PERFORM "doAdd".
  IF result = expectedResult THEN
    DISPLAY "." WITH NO ADVANCING
  ELSE
    DISPLAY SPACE
    DISPLAY "Expected " expectedResult " but was " result
  END-IF.