79023ee190
Reviewed By: bhamodi Differential Revision: D33331422 fbshipit-source-id: 016e8dcc0c0c7f1fc353a348b54fda0d5e2ddc01
87 lines
2.7 KiB
C++
87 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include <Flipper/FlipperResponderImpl.h>
|
|
#include "yarpl/Flowable.h"
|
|
#include "yarpl/Single.h"
|
|
#include "yarpl/single/SingleTestObserver.h"
|
|
|
|
#include <folly/json.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace facebook {
|
|
namespace flipper {
|
|
namespace test {
|
|
|
|
using folly::dynamic;
|
|
|
|
void assertIsSuccess(folly::dynamic d);
|
|
void assertIsError(folly::dynamic d);
|
|
|
|
TEST(FlipperResponderImplTest, testSuccessWrapper) {
|
|
auto dynamicSingle =
|
|
yarpl::single::Single<folly::dynamic>::create([](auto observer) mutable {
|
|
observer->onSubscribe(yarpl::single::SingleSubscriptions::empty());
|
|
auto responder = std::make_shared<FlipperResponderImpl>(observer);
|
|
responder->success(folly::dynamic::object("my", "object"));
|
|
});
|
|
auto to = yarpl::single::SingleTestObserver<folly::dynamic>::create();
|
|
dynamicSingle->subscribe(to);
|
|
|
|
to->awaitTerminalEvent();
|
|
auto output = to->getOnSuccessValue();
|
|
assertIsSuccess(output);
|
|
EXPECT_EQ(output["success"]["my"], "object");
|
|
}
|
|
|
|
TEST(FlipperResponderImplTest, testErrorWrapper) {
|
|
auto dynamicSingle =
|
|
yarpl::single::Single<folly::dynamic>::create([](auto observer) mutable {
|
|
observer->onSubscribe(yarpl::single::SingleSubscriptions::empty());
|
|
auto responder = std::make_shared<FlipperResponderImpl>(observer);
|
|
responder->error(folly::dynamic::object("my", "object"));
|
|
});
|
|
auto to = yarpl::single::SingleTestObserver<folly::dynamic>::create();
|
|
dynamicSingle->subscribe(to);
|
|
|
|
to->awaitTerminalEvent();
|
|
auto output = to->getOnSuccessValue();
|
|
assertIsError(output);
|
|
EXPECT_EQ(output["error"]["my"], "object");
|
|
}
|
|
|
|
TEST(FlipperResponderImplTest, testNoExplicitResponseReturnsSuccess) {
|
|
auto to = yarpl::single::SingleTestObserver<folly::dynamic>::create();
|
|
{
|
|
auto dynamicSingle = yarpl::single::Single<folly::dynamic>::create(
|
|
[](auto observer) mutable {
|
|
observer->onSubscribe(yarpl::single::SingleSubscriptions::empty());
|
|
auto responder = std::make_shared<FlipperResponderImpl>(observer);
|
|
});
|
|
dynamicSingle->subscribe(to);
|
|
}
|
|
|
|
to->awaitTerminalEvent();
|
|
auto output = to->getOnSuccessValue();
|
|
assertIsSuccess(output);
|
|
EXPECT_TRUE(output["success"].empty());
|
|
}
|
|
|
|
void assertIsSuccess(folly::dynamic d) {
|
|
EXPECT_NE(d.find("success"), d.items().end());
|
|
EXPECT_EQ(d.find("error"), d.items().end());
|
|
}
|
|
|
|
void assertIsError(folly::dynamic d) {
|
|
EXPECT_NE(d.find("error"), d.items().end());
|
|
EXPECT_EQ(d.find("success"), d.items().end());
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace flipper
|
|
} // namespace facebook
|