-
Notifications
You must be signed in to change notification settings - Fork 73
Expand file tree
/
Copy pathevaluator_stack_test.cc
More file actions
97 lines (78 loc) · 3.14 KB
/
Copy pathevaluator_stack_test.cc
File metadata and controls
97 lines (78 loc) · 3.14 KB
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
#include "eval/eval/evaluator_stack.h"
#include "base/type_factory.h"
#include "base/type_manager.h"
#include "base/type_provider.h"
#include "base/value.h"
#include "base/value_factory.h"
#include "extensions/protobuf/memory_manager.h"
#include "internal/testing.h"
namespace google::api::expr::runtime {
namespace {
using ::cel::TypeFactory;
using ::cel::TypeManager;
using ::cel::TypeProvider;
using ::cel::ValueFactory;
using ::cel::extensions::ProtoMemoryManager;
// Test Value Stack Push/Pop operation
TEST(EvaluatorStackTest, StackPushPop) {
google::protobuf::Arena arena;
ProtoMemoryManager manager(&arena);
TypeFactory type_factory(manager);
TypeManager type_manager(type_factory, TypeProvider::Builtin());
ValueFactory value_factory(type_manager);
google::api::expr::v1alpha1::Expr expr;
expr.mutable_ident_expr()->set_name("name");
CelAttribute attribute(expr, {});
EvaluatorStack stack(10);
stack.Push(value_factory.CreateIntValue(1));
stack.Push(value_factory.CreateIntValue(2), AttributeTrail());
stack.Push(value_factory.CreateIntValue(3), AttributeTrail(expr, manager));
ASSERT_EQ(stack.Peek().As<cel::IntValue>()->value(), 3);
ASSERT_FALSE(stack.PeekAttribute().empty());
ASSERT_EQ(stack.PeekAttribute().attribute(), attribute);
stack.Pop(1);
ASSERT_EQ(stack.Peek().As<cel::IntValue>()->value(), 2);
ASSERT_TRUE(stack.PeekAttribute().empty());
stack.Pop(1);
ASSERT_EQ(stack.Peek().As<cel::IntValue>()->value(), 1);
ASSERT_TRUE(stack.PeekAttribute().empty());
}
// Test that inner stacks within value stack retain the equality of their sizes.
TEST(EvaluatorStackTest, StackBalanced) {
google::protobuf::Arena arena;
ProtoMemoryManager manager(&arena);
TypeFactory type_factory(manager);
TypeManager type_manager(type_factory, TypeProvider::Builtin());
ValueFactory value_factory(type_manager);
EvaluatorStack stack(10);
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.Push(value_factory.CreateIntValue(1));
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.Push(value_factory.CreateIntValue(2), AttributeTrail());
stack.Push(value_factory.CreateIntValue(3), AttributeTrail());
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.PopAndPush(value_factory.CreateIntValue(4), AttributeTrail());
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.PopAndPush(value_factory.CreateIntValue(5));
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.Pop(3);
ASSERT_EQ(stack.size(), stack.attribute_size());
}
TEST(EvaluatorStackTest, Clear) {
google::protobuf::Arena arena;
ProtoMemoryManager manager(&arena);
TypeFactory type_factory(manager);
TypeManager type_manager(type_factory, TypeProvider::Builtin());
ValueFactory value_factory(type_manager);
EvaluatorStack stack(10);
ASSERT_EQ(stack.size(), stack.attribute_size());
stack.Push(value_factory.CreateIntValue(1));
stack.Push(value_factory.CreateIntValue(2), AttributeTrail());
stack.Push(value_factory.CreateIntValue(3), AttributeTrail());
ASSERT_EQ(stack.size(), 3);
stack.Clear();
ASSERT_EQ(stack.size(), 0);
ASSERT_TRUE(stack.empty());
}
} // namespace
} // namespace google::api::expr::runtime