diff --git a/socket_client.c b/socket_client.c index 76c9a7aa..ab0415d9 100644 --- a/socket_client.c +++ b/socket_client.c @@ -36,7 +36,10 @@ int main (int argc, char **argv) { printf ("Socket created\n"); address.sun_family = AF_LOCAL; - strcpy(address.sun_path, UDS_FILE); + if(argc == 2 && strcmp(argv[1], "travis") == 0) + strcpy(address.sun_path,"pihole-FTL.sock"); + else + strcpy(address.sun_path,"/var/run/pihole/FTL.sock"); ret = connect(socketfd, (struct sockaddr *) &address, sizeof (address)); if (ret != 0) { diff --git a/test/test_suite.sh b/test/test_suite.sh index 69de9c31..a4907c1f 100644 --- a/test/test_suite.sh +++ b/test/test_suite.sh @@ -196,6 +196,12 @@ load 'libs/bats-support/load' [[ ${lines[0]} == "pihole-FTL - The Pi-hole FTL engine" ]] } +@test "Unix socket returning data" { + run bash -c './pihole-FTL help' + echo "output: ${lines[@]}" + [[ ${lines[0]} == "pihole-FTL - The Pi-hole FTL engine" ]] +} + @test "Final part of the tests: Killing pihole-FTL process" { run bash -c 'echo ">kill" | nc -v 127.0.0.1 4711' echo "output: ${lines[@]}"