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
98
99
100
|
#include <yt/cpp/mapreduce/interface/job_counters.h>
#include <yt/cpp/mapreduce/interface/operation.h>
#include <library/cpp/yson/node/node_io.h>
#include <library/cpp/testing/gtest/gtest.h>
using namespace NYT;
TEST(TJobCountersTest, Full)
{
const TString input = R"""(
{
"completed" = {
"total" = 6;
"non-interrupted" = 1;
"interrupted" = {
"whatever_interrupted" = 2;
"whatever_else_interrupted" = 3;
};
};
"aborted" = {
"non_scheduled" = {
"whatever_non_scheduled" = 4;
"whatever_else_non_scheduled" = 5;
};
"scheduled" = {
"whatever_scheduled" = 6;
"whatever_else_scheduled" = 7;
};
"total" = 22;
};
"lost" = 8;
"invalidated" = 9;
"failed" = 10;
"running" = 11;
"suspended" = 12;
"pending" = 13;
"blocked" = 14;
"total" = 105;
})""";
TJobCounters counters(NodeFromYsonString(input));
EXPECT_EQ(counters.GetTotal(), 105u);
EXPECT_EQ(counters.GetCompleted().GetTotal(), 6u);
EXPECT_EQ(counters.GetCompletedNonInterrupted().GetTotal(), 1u);
EXPECT_EQ(counters.GetCompletedInterrupted().GetTotal(), 5u);
EXPECT_EQ(counters.GetAborted().GetTotal(), 22u);
EXPECT_EQ(counters.GetAbortedNonScheduled().GetTotal(), 9u);
EXPECT_EQ(counters.GetAbortedScheduled().GetTotal(), 13u);
EXPECT_EQ(counters.GetLost().GetTotal(), 8u);
EXPECT_EQ(counters.GetInvalidated().GetTotal(), 9u);
EXPECT_EQ(counters.GetFailed().GetTotal(), 10u);
EXPECT_EQ(counters.GetRunning().GetTotal(), 11u);
EXPECT_EQ(counters.GetSuspended().GetTotal(), 12u);
EXPECT_EQ(counters.GetPending().GetTotal(), 13u);
EXPECT_EQ(counters.GetBlocked().GetTotal(), 14u);
EXPECT_EQ(counters.GetCompletedInterrupted().GetValue("whatever_interrupted"), 2u);
EXPECT_EQ(counters.GetCompletedInterrupted().GetValue("whatever_else_interrupted"), 3u);
EXPECT_EQ(counters.GetAbortedNonScheduled().GetValue("whatever_non_scheduled"), 4u);
EXPECT_EQ(counters.GetAbortedNonScheduled().GetValue("whatever_else_non_scheduled"), 5u);
EXPECT_EQ(counters.GetAbortedScheduled().GetValue("whatever_scheduled"), 6u);
EXPECT_EQ(counters.GetAbortedScheduled().GetValue("whatever_else_scheduled"), 7u);
EXPECT_THROW(counters.GetCompletedInterrupted().GetValue("Nothingness"), yexception);
}
TEST(TJobCountersTest, Empty)
{
const TString input = "{}";
TJobCounters counters(NodeFromYsonString(input));
EXPECT_EQ(counters.GetTotal(), 0u);
EXPECT_EQ(counters.GetCompleted().GetTotal(), 0u);
EXPECT_EQ(counters.GetCompletedNonInterrupted().GetTotal(), 0u);
EXPECT_EQ(counters.GetCompletedInterrupted().GetTotal(), 0u);
EXPECT_EQ(counters.GetAborted().GetTotal(), 0u);
EXPECT_EQ(counters.GetAbortedNonScheduled().GetTotal(), 0u);
EXPECT_EQ(counters.GetAbortedScheduled().GetTotal(), 0u);
EXPECT_EQ(counters.GetLost().GetTotal(), 0u);
EXPECT_EQ(counters.GetInvalidated().GetTotal(), 0u);
EXPECT_EQ(counters.GetFailed().GetTotal(), 0u);
EXPECT_EQ(counters.GetRunning().GetTotal(), 0u);
EXPECT_EQ(counters.GetSuspended().GetTotal(), 0u);
EXPECT_EQ(counters.GetPending().GetTotal(), 0u);
EXPECT_EQ(counters.GetBlocked().GetTotal(), 0u);
}
TEST(TJobCountersTest, Broken)
{
EXPECT_THROW_MESSAGE_HAS_SUBSTR((TJobCounters(TNode())), yexception, "TJobCounters");
EXPECT_THROW_MESSAGE_HAS_SUBSTR((TJobCounters(TNode(1))), yexception, "TJobCounters");
EXPECT_THROW_MESSAGE_HAS_SUBSTR((TJobCounters(TNode(1.0))), yexception, "TJobCounters");
EXPECT_THROW_MESSAGE_HAS_SUBSTR((TJobCounters(TNode("Whatever"))), yexception, "TJobCounters");
}
|
