more tcp tests (#370)

* more tcp tests

* fix tests

Author: Anton-4

ci/all_tests.sh

@@ -97,12 +97,13 @@ rm -rf dirExampleE
 rm -rf dirExampleA
 rm -rf dirExampleD
 
+# countdown, echo, form... all require user input or special setup
+ignore_list=("stdin-basic.roc" "stdin-pipe.roc" "command-line-args.roc" "http.roc" "env-var.roc" "bytes-stdin-stdout.roc" "error-handling.roc" "tcp-client.roc" "tcp.roc" "terminal-app-snake.roc")
+
 # roc dev (some expects only run with `roc dev`)
 for roc_file in $EXAMPLES_DIR*.roc; do
     base_file=$(basename "$roc_file")
 
-    # countdown, echo, form... all require user input or special setup
-    ignore_list=("stdin-basic.roc" "stdin-pipe.roc" "command-line-args.roc" "http.roc" "env-var.roc" "bytes-stdin-stdout.roc" "error-handling.roc" "tcp-client.roc" "terminal-app-snake.roc")
 
     # check if base_file matches something from ignore_list
     for file in "${ignore_list[@]}"; do
@@ -130,9 +131,17 @@ for roc_file in $EXAMPLES_DIR*.roc; do
         $ROC dev $roc_file $ROC_BUILD_FLAGS
     fi
 done
+
 for roc_file in $TESTS_DIR*.roc; do
     base_file=$(basename "$roc_file")
 
+    # check if base_file matches something from ignore_list
+    for file in "${ignore_list[@]}"; do
+        if [ "$base_file" == "$file" ]; then
+            continue 2 # continue the outer loop if a match is found
+        fi
+    done
+
     if [ "$base_file" == "sqlite.roc" ]; then
         DB_PATH=${TESTS_DIR}test.db $ROC dev $roc_file $ROC_BUILD_FLAGS
     else

ci/expect_scripts/tcp.exp

@@ -0,0 +1,40 @@
+#!/usr/bin/expect
+
+# uncomment line below for debugging
+# exp_internal 1
+
+set timeout 7
+
+source ./ci/expect_scripts/shared-code.exp
+
+# get path to cat command
+set cat_path [exec which cat]
+# Start echo server
+spawn ncat -e $cat_path -l 8085
+sleep 5
+
+spawn $env(TESTS_DIR)tcp
+
+
+expect "Testing Tcp module functions" {
+    expect "Successfully connected to localhost" {
+        expect "Echo server reply: Hello" {
+            expect "Echo server reply: Test message from Roc!" {
+                expect "Tcp.read_up_to yielded: 'do not read past me'" {
+                    expect "Tcp.read_exactly yielded: 'ABC'" {
+                        expect "Tcp.read_until yielded: 'Line1\r\n'" {
+                            expect "✓ All tests executed! ✓\r\n" {
+                                expect eof {
+                                    check_exit_and_segfault
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+        }
+    }
+}
+
+puts stderr "\nExpect script failed: output was different from expected value."
+exit 1

flake.nix

@@ -30,6 +30,10 @@
         rust =
           pkgs.rust-bin.fromRustupToolchainFile "${toString ./rust-toolchain.toml}";
 
+        aliases = ''
+          alias testcmd='export ROC=roc && export EXAMPLES_DIR=./examples/ && ./ci/all_tests.sh'
+        '';
+
         linuxInputs = with pkgs;
           lib.optionals stdenv.isLinux [
             valgrind
@@ -64,6 +68,10 @@
           # for crti.o, crtn.o, and Scrt1.o
           NIX_GLIBC_PATH =
             if pkgs.stdenv.isLinux then "${pkgs.glibc.out}/lib" else "";
+
+          shellHook = ''
+            ${aliases}
+          '';
         };
 
         formatter = pkgs.nixpkgs-fmt;

tests/.gitignore

@@ -1,4 +1,6 @@
-sqlite
-utc
-url
-env
+# Ignore all files including binary files that have no extension
+*
+# Unignore all files with extensions
+!*.*
+# Unignore all directories
+!*/

tests/tcp.roc

@@ -0,0 +1,73 @@
+app [main!] { pf: platform "../platform/main.roc" }
+
+import pf.Stdout
+import pf.Tcp
+import pf.Arg exposing [Arg]
+
+main! : List Arg => Result {} _
+main! = |_args|
+    Stdout.line!("Testing Tcp module functions...")?
+    Stdout.line!("Note: These tests require a TCP server running on localhost:8085")?
+    Stdout.line!("You can start one with: ncat -e `which cat` -l 8085")?
+    Stdout.line!("")?
+
+    Stdout.line!("Testing Tcp.connect!:")?
+    when Tcp.connect!("127.0.0.1", 8085) is
+        Ok(stream) ->
+            Stdout.line!("✓ Successfully connected to localhost:8085")?
+            test_tcp_functions!(stream)?
+            Stdout.line!("\n✓ All tests executed! ✓")
+        
+        Err(connect_err) ->
+            err_str = Tcp.connect_err_to_str(connect_err)
+            Err(Exit(1, "✗ Failed to connect: ${err_str}"))
+
+
+test_tcp_functions! : Tcp.Stream => Result {} _
+test_tcp_functions! = |stream|
+
+    Stdout.line!("\nTesting Tcp.write!:")?
+    hello_bytes = [72, 101, 108, 108, 111, 10] # "Hello\n" in bytes
+    Tcp.write!(stream, hello_bytes)?
+
+    reply_msg = Tcp.read_line!(stream)?
+    Stdout.line!("Echo server reply: ${reply_msg}")?
+
+
+    Stdout.line!("Testing Tcp.write_utf8!:")?
+    test_message = "Test message from Roc!\n"
+    Tcp.write_utf8!(stream, test_message)?
+    
+    reply_msg_utf8 = Tcp.read_line!(stream)?
+    Stdout.line!("Echo server reply: ${reply_msg_utf8}")?
+
+
+    Stdout.line!("Testing Tcp.read_up_to!:")?
+
+    do_not_read_bytes = [100, 111, 32, 110, 111, 116, 32, 114, 101, 97, 100, 32, 112, 97, 115, 116, 32, 109, 101, 65] # "do not read past meA" in bytes
+    Tcp.write!(stream, do_not_read_bytes)?
+
+    nineteen_bytes = Tcp.read_up_to!(stream, 19) ? |err| FailedReadUpTo(err)
+    nineteen_bytes_as_str = Str.from_utf8(nineteen_bytes) ? |err| ReadUpToFromUtf8(err)
+
+    Stdout.line!("Tcp.read_up_to yielded: '${nineteen_bytes_as_str}'")?
+
+
+    Stdout.line!("\nTesting Tcp.read_exactly!:")?
+    Tcp.write_utf8!(stream, "BC")?
+
+    three_bytes = Tcp.read_exactly!(stream, 3) ? |err| FailedReadExactly(err)
+    three_bytes_as_str = Str.from_utf8(three_bytes) ? |err| ReadExactlyFromUtf8(err)
+
+    Stdout.line!("Tcp.read_exactly yielded: '${three_bytes_as_str}'")?
+
+
+    Stdout.line!("\nTesting Tcp.read_until!:")?
+    Tcp.write_utf8!(stream, "Line1\nLine2\n")?
+
+    bytes_until = Tcp.read_until!(stream, '\n') ? |err| FailedReadUntil(err)
+    bytes_until_as_str = Str.from_utf8(bytes_until) ? |err| ReadUntilFromUtf8(err)
+
+    Stdout.line!("Tcp.read_until yielded: '${bytes_until_as_str}'")?
+
+    Ok({})