// Copyright (c) 2014-2016 Josh Blum
// SPDX-License-Identifier: BSL-1.0

#include "PythonSupport.hpp"
#include <Pothos/Framework.hpp>
#include <Pothos/Managed.hpp>
#include <Pothos/Testing.hpp>
#include <Pothos/Proxy.hpp>
#include <Poco/JSON/Object.h>
#include <iostream>

POTHOS_TEST_BLOCK("/proxy/" POTHOS_PYNAME "/tests", python_module_import)
{
    auto env = Pothos::ProxyEnvironment::make(POTHOS_PYNAME);
    env->findProxy("Pothos");
}

POTHOS_TEST_BLOCK("/proxy/" POTHOS_PYNAME "/tests", test_python_module)
{
    auto env = Pothos::ProxyEnvironment::make(POTHOS_PYNAME);
    env->findProxy("Pothos.TestPothos").callProxy("main");
}

POTHOS_TEST_BLOCK("/proxy/" POTHOS_PYNAME "/tests", test_python_block)
{
    auto env = Pothos::ProxyEnvironment::make("managed");
    auto reg = env->findProxy("Pothos/BlockRegistry");
    auto feeder = reg.callProxy("/blocks/feeder_source", "int");
    auto collector = reg.callProxy("/blocks/collector_sink", "int");
    auto forwarder = reg.callProxy("/" POTHOS_PYNAME "/forwarder", Pothos::DType("int"));

    //create a test plan
    Poco::JSON::Object::Ptr testPlan(new Poco::JSON::Object());
    testPlan->set("enableBuffers", true);
    testPlan->set("enableLabels", true);
    testPlan->set("enableMessages", true);
    auto expected = feeder.callProxy("feedTestPlan", testPlan);

    //run the topology
    {
        Pothos::Topology topology;
        topology.connect(feeder, 0, forwarder, 0);
        topology.connect(forwarder, 0, collector, 0);
        std::cout << "topology commit\n";
        topology.commit();
        POTHOS_TEST_TRUE(topology.waitInactive(0.5, 5.0));
    }

    collector.callVoid("verifyTestPlan", expected);
    std::cout << "run done\n";
}

POTHOS_TEST_BLOCK("/proxy/" POTHOS_PYNAME "/tests", test_signals_and_slots)
{
    auto env = Pothos::ProxyEnvironment::make("managed");
    auto reg = env->findProxy("Pothos/BlockRegistry");
    auto emitter = reg.callProxy("/" POTHOS_PYNAME "/simple_signal_emitter");
    auto acceptor = reg.callProxy("/" POTHOS_PYNAME "/simple_slot_acceptor");

    //run the topology
    {
        Pothos::Topology topology;
        topology.connect(emitter, "activateCalled", acceptor, "activateHandler");
        std::cout << "topology commit\n";
        topology.commit();
        POTHOS_TEST_TRUE(topology.waitInactive());
    }

    auto lastWord = acceptor.call<std::string>("getLastWord");
    POTHOS_TEST_EQUAL(lastWord, "hello");
}