mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
1.6 KiB
74 lines
1.6 KiB
// Copyright 2021 gRPC authors. |
|
// |
|
// Licensed under the Apache License, Version 2.0 (the "License"); |
|
// you may not use this file except in compliance with the License. |
|
// You may obtain a copy of the License at |
|
// |
|
// http://www.apache.org/licenses/LICENSE-2.0 |
|
// |
|
// Unless required by applicable law or agreed to in writing, software |
|
// distributed under the License is distributed on an "AS IS" BASIS, |
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
// See the License for the specific language governing permissions and |
|
// limitations under the License. |
|
|
|
#include "src/core/lib/promise/loop.h" |
|
|
|
#include <stdint.h> |
|
|
|
#include <utility> |
|
|
|
#include "gtest/gtest.h" |
|
|
|
#include "src/core/lib/promise/seq.h" |
|
|
|
namespace grpc_core { |
|
|
|
TEST(LoopTest, CountToFive) { |
|
int i = 0; |
|
Loop([&i]() -> LoopCtl<int> { |
|
i++; |
|
if (i < 5) return Continue(); |
|
return i; |
|
})(); |
|
EXPECT_EQ(i, 5); |
|
} |
|
|
|
TEST(LoopTest, FactoryCountToFive) { |
|
int i = 0; |
|
Loop([&i]() { |
|
return [&i]() -> LoopCtl<int> { |
|
i++; |
|
if (i < 5) return Continue(); |
|
return i; |
|
}; |
|
})(); |
|
EXPECT_EQ(i, 5); |
|
} |
|
|
|
TEST(LoopTest, LoopOfSeq) { |
|
auto x = |
|
Loop(Seq([]() { return 42; }, [](int i) -> LoopCtl<int> { return i; }))(); |
|
EXPECT_EQ(x, Poll<int>(42)); |
|
} |
|
|
|
TEST(LoopTest, CanAccessFactoryLambdaVariables) { |
|
int i = 0; |
|
auto x = Loop([p = &i]() { |
|
return [q = &p]() -> Poll<LoopCtl<int>> { |
|
++**q; |
|
return Pending{}; |
|
}; |
|
}); |
|
auto y = std::move(x); |
|
auto z = std::move(y); |
|
z(); |
|
EXPECT_EQ(i, 1); |
|
} |
|
|
|
} // namespace grpc_core |
|
|
|
int main(int argc, char** argv) { |
|
::testing::InitGoogleTest(&argc, argv); |
|
return RUN_ALL_TESTS(); |
|
}
|
|
|