summaryrefslogtreecommitdiff
path: root/cpp/test/IceStorm/persistent/Client.cpp
blob: 8b8bb5ba6d319dbebdbcab4361a7f56f4953ff35 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// **********************************************************************
//
// Copyright (c) 2003-present ZeroC, Inc. All rights reserved.
//
// This copy of Ice is licensed to you under the terms described in the
// ICE_LICENSE file included in this distribution.
//
// **********************************************************************

#include <Ice/Ice.h>
#include <IceStorm/IceStorm.h>
#include <TestHelper.h>

using namespace std;
using namespace Ice;
using namespace IceStorm;
using namespace Test;

class Client : public Test::TestHelper
{
public:

    void run(int, char**);
};

void
Client::run(int argc, char** argv)
{
    Ice::CommunicatorHolder communicator = initialize(argc, argv);
    PropertiesPtr properties = communicator->getProperties();
    string managerProxy = properties->getProperty("IceStormAdmin.TopicManager.Default");
    if(managerProxy.empty())
    {
        ostringstream os;
        os << argv[0] << ": property `IceStormAdmin.TopicManager.Default' is not set";
        throw invalid_argument(os.str());
    }

    IceStorm::TopicManagerPrx manager =
        IceStorm::TopicManagerPrx::checkedCast(communicator->stringToProxy(managerProxy));
    if(!manager)
    {
        ostringstream os;
        os << argv[0] << ": `" << managerProxy << "' is not running";
        throw invalid_argument(os.str());
    }

    if(argc != 2)
    {
        throw invalid_argument("client create|check");
    }

    string action(argv[1]);
    if(action == "create")
    {
        //
        // Create topics
        //
        cerr << "creating topics and links..." << flush;
        TopicPrx linkTo = 0;
        for(int i = 0; i < 10; ++i)
        {
            ostringstream topicName;
            topicName << "topic" << i;
            TopicPrx topic = manager->create(topicName.str());
            if(linkTo)
            {
                topic->link(linkTo, i + 1);
            }
            linkTo = topic;
        }
        cerr << "ok" << endl;

        //
        // Create a subscriber for each topic
        //
        cerr << "create a susbscriber for each topic... ";
        for(int i = 0; i < 10; ++i)
        {
            ostringstream topicName;
            topicName << "topic" << i;
            TopicPrx topic = manager->retrieve(topicName.str());
            test(topic);

            ostringstream subscriber;
            subscriber << "subscriber" << i << ":default -h 10000";
            topic->subscribeAndGetPublisher(IceStorm::QoS(), communicator->stringToProxy(subscriber.str()));
        }
        cerr << "ok" << endl;
    }
    else
    {
        test(action == "check");
    }
}

DEFINE_TEST(Client)