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
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ash/borealis/infra/transition.h"
#include <memory>
#include "base/functional/bind.h"
#include "base/strings/string_number_conversions.h"
#include "base/test/task_environment.h"
#include "chrome/browser/ash/borealis/testing/callback_factory.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace borealis {
namespace {
template <typename F>
using CallbackFactory = StrictCallbackFactory<F>;
class ParseIntTransition : public Transition<std::string, int, bool> {
void Start(std::unique_ptr<std::string> in) override {
int val;
if (base::StringToInt(*in, &val)) {
Succeed(std::make_unique<int>(val));
} else {
Fail(false);
}
}
};
TEST(TransitionTest, TransitionCanTransformInputToOutput) {
base::test::SingleThreadTaskEnvironment task_environment;
ParseIntTransition transition;
CallbackFactory<ParseIntTransition::OnCompleteSignature> callback_handler;
EXPECT_CALL(callback_handler, Call(testing::_))
.WillOnce(testing::Invoke([](ParseIntTransition::Result result) {
ASSERT_TRUE(result.has_value());
EXPECT_EQ(*result.value(), 12345);
}));
transition.Begin(std::make_unique<std::string>("12345"),
callback_handler.BindOnce());
task_environment.RunUntilIdle();
}
TEST(TransitionTest, TransitionCanFail) {
base::test::SingleThreadTaskEnvironment task_environment;
ParseIntTransition transition;
CallbackFactory<ParseIntTransition::OnCompleteSignature> callback_handler;
EXPECT_CALL(callback_handler, Call(testing::_))
.WillOnce(testing::Invoke([](ParseIntTransition::Result result) {
EXPECT_FALSE(result.has_value());
}));
transition.Begin(std::make_unique<std::string>("not a number"),
callback_handler.BindOnce());
task_environment.RunUntilIdle();
}
class MultiCompletionTransition
: public Transition<std::string, std::string, std::string> {
void Start(std::unique_ptr<std::string> in) override {
Fail("foo");
Succeed(std::make_unique<std::string>("bar"));
}
};
TEST(TransitionTest, MultipleCompletionFiresCallbackOnce) {
base::test::SingleThreadTaskEnvironment task_environment;
MultiCompletionTransition transition;
CallbackFactory<MultiCompletionTransition::OnCompleteSignature>
callback_handler;
EXPECT_CALL(callback_handler, Call(testing::_))
.WillOnce(testing::Invoke([](MultiCompletionTransition::Result result) {
// The transition completes twice but only the first one will be used.
EXPECT_FALSE(result.has_value());
EXPECT_EQ(result.error(), "foo");
}));
transition.Begin(nullptr, callback_handler.BindOnce());
task_environment.RunUntilIdle();
}
} // namespace
} // namespace borealis
|
