Client.cpp

/* ===============================
   Client.cpp (Camera Testing)
   =============================== */
#define NOMINMAX
#define DOCTEST_CONFIG_IMPLEMENT_WITH_MAIN
#include "Client.h"
#include <iostream>
#include <winsock2.h>
#include <ws2tcpip.h>
#include "doctest.h"
#include <regex>

#pragma comment(lib, "ws2_32.lib")

Client::Client() {}

std::string Client::sendCommand(const std::string& command) {
    WSADATA wsaData;
    int result = WSAStartup(MAKEWORD(2, 2), &wsaData);
    if (result != 0) {
        throw std::runtime_error("WSAStartup failed");
    }

    SOCKET clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (clientSocket == INVALID_SOCKET) {
        WSACleanup();
        throw std::runtime_error("Socket creation failed");
    }

    sockaddr_in serverAddr = {};
    serverAddr.sin_family = AF_INET;
    serverAddr.sin_port = htons(8080);
    serverAddr.sin_addr.s_addr = inet_addr("127.0.0.1");

    if (connect(clientSocket, (struct sockaddr*)&serverAddr, sizeof(serverAddr)) == SOCKET_ERROR) {
        closesocket(clientSocket);
        WSACleanup();
        throw std::runtime_error("Connection failed");
    }

    result = send(clientSocket, command.c_str(), static_cast<int>(command.length()), 0);
    if (result == SOCKET_ERROR) {
        closesocket(clientSocket);
        WSACleanup();
        throw std::runtime_error("Send failed");
    }

    char buffer[1024] = { 0 };
    int bytesReceived = recv(clientSocket, buffer, sizeof(buffer) - 1, 0);
    std::string response;
    if (bytesReceived > 0) {
        buffer[bytesReceived] = '\0';
        response = buffer;
    }

    closesocket(clientSocket);
    WSACleanup();
    return response;
}

TEST_CASE("Camera function tests") {
    Client client;

    SUBCASE("Status check") {
        std::string response = client.sendCommand("GET /192.168.1.98/status 192.168.2.2");
        CHECK(response.find("Camera is idle") != std::string::npos);
        CHECK(response.find("Night vision is disabled") != std::string::npos);
    }

    SUBCASE("Recording check") {
        // Start recording
        std::string response = client.sendCommand("POST /192.168.1.98/start 192.168.2.2");
        CHECK(response.find("Camera started recording") != std::string::npos);

        // Check status of camera
        response = client.sendCommand("GET /192.168.1.98/status 192.168.2.2");
        CHECK(response.find("Recording is in progress") != std::string::npos);

        // Stop recording
        response = client.sendCommand("POST /192.168.1.98/stop 192.168.2.2");
        CHECK(response.find("Camera stopped recording.") != std::string::npos);

        // Verify stopped status 
        response = client.sendCommand("GET /192.168.1.98/status 192.168.2.2");
        CHECK(response.find("Camera is idle") != std::string::npos);
    }

    SUBCASE("Night vision check") {
        // Toggle night vision on
        std::string response = client.sendCommand("POST /192.168.1.98/toggle/nightvision 192.168.2.2");
        CHECK(response.find("Night vision is now enabled") != std::string::npos);

        // Verify night vision status
        response = client.sendCommand("GET /192.168.1.98/status 192.168.2.2");
        CHECK(response.find("Night vision is enabled") != std::string::npos);

        // Toggle night vision off
        response = client.sendCommand("POST /192.168.1.98/toggle/nightvision 192.168.2.2");
        CHECK(response.find("Night vision is now disabled") != std::string::npos);
    }
}