speechTranscription_nws_yarp_test.cpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2023-2023 Istituto Italiano di Tecnologia (IIT)
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#include <yarp/os/Network.h>
#include <yarp/dev/PolyDriver.h>
#include <yarp/dev/WrapperSingle.h>
 
#include <catch2/catch_amalgamated.hpp>
#include <harness.h>
 
#include <yarp/sig/Sound.h>
 
using namespace yarp::dev;
using namespace yarp::os;
 
TEST_CASE("dev::speechTranscription_nws_yarp", "[yarp::dev]")
{
    YARP_REQUIRE_PLUGIN("fakeSpeechTranscription", "device");
    YARP_REQUIRE_PLUGIN("speechTranscription_nws_yarp", "device");
 
    Network::setLocalMode(true);
 
    SECTION("Checking speechTranscription_nws_yarp device alone")
    {
        PolyDriver dd;
 
        {
            Property pnws_cfg;
            pnws_cfg.put("device", "speechTranscription_nws_yarp");
            pnws_cfg.put("name", "/speechTranscription_nws");
            REQUIRE(dd.open(pnws_cfg));
        }
 
        yarp::os::Time::delay(1.0);
 
        //"Close all polydrivers and check"
        {
            CHECK(dd.close());
        }
    }
 
    SECTION("Checking speechTranscription_nws_yarp device attached to fake device")
    {
        PolyDriver ddnws;
        PolyDriver ddfake;
 
        {
            Property pnws_cfg;
            pnws_cfg.put("device", "speechTranscription_nws_yarp");
            pnws_cfg.put("name", "/speechTranscription_nws");
            REQUIRE(ddnws.open(pnws_cfg));
 
            Property pdev_cfg;
            pdev_cfg.put("device", "fakeSpeechTranscription");
            REQUIRE(ddfake.open(pdev_cfg));
 
            {yarp::dev::WrapperSingle* ww_nws=nullptr; ddnws.view(ww_nws);
            REQUIRE(ww_nws);
            bool result_att = ww_nws->attach(&ddfake);
            REQUIRE(result_att); }
        }
 
        //tests
        {
            Port po;
            Port pi;
            CHECK(po.open("/test:o"));
            CHECK(Network::connect("/test:o","/speechTranscription_nws/sound:i"));
            CHECK(pi.open("/test:i"));
            CHECK(Network::connect("/speechTranscription_nws/text:o", "/test:i"));
            yarp::os::Time::delay(0.5);
 
            yarp::sig::Sound s;
            s.resize(100, 2);
            CHECK(po.write(s));
            yarp::os::Time::delay(0.5);
            yarp::os::Bottle b;
            CHECK(pi.read(b));
            CHECK(b.get(0).asString() == "hello world");
            CHECK(b.get(1).asFloat64() == 1);
 
            po.close();
            pi.close();
        }
        yarp::os::Time::delay(0.5);
 
        //"Close all polydrivers and check"
        {
            CHECK(ddnws.close());
            CHECK(ddfake.close());
        }
    }
 
    Network::setLocalMode(false);
}