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
101
102
103
|
#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/unittest/registar.h>
using namespace NYT;
Y_UNIT_TEST_SUITE(JobCounters)
{
Y_UNIT_TEST(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));
UNIT_ASSERT_VALUES_EQUAL(counters.GetTotal(), 105);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompleted().GetTotal(), 6);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedNonInterrupted().GetTotal(), 1);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedInterrupted().GetTotal(), 5);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAborted().GetTotal(), 22);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedNonScheduled().GetTotal(), 9);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedScheduled().GetTotal(), 13);
UNIT_ASSERT_VALUES_EQUAL(counters.GetLost().GetTotal(), 8);
UNIT_ASSERT_VALUES_EQUAL(counters.GetInvalidated().GetTotal(), 9);
UNIT_ASSERT_VALUES_EQUAL(counters.GetFailed().GetTotal(), 10);
UNIT_ASSERT_VALUES_EQUAL(counters.GetRunning().GetTotal(), 11);
UNIT_ASSERT_VALUES_EQUAL(counters.GetSuspended().GetTotal(), 12);
UNIT_ASSERT_VALUES_EQUAL(counters.GetPending().GetTotal(), 13);
UNIT_ASSERT_VALUES_EQUAL(counters.GetBlocked().GetTotal(), 14);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedInterrupted().GetValue("whatever_interrupted"), 2);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedInterrupted().GetValue("whatever_else_interrupted"), 3);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedNonScheduled().GetValue("whatever_non_scheduled"), 4);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedNonScheduled().GetValue("whatever_else_non_scheduled"), 5);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedScheduled().GetValue("whatever_scheduled"), 6);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedScheduled().GetValue("whatever_else_scheduled"), 7);
UNIT_ASSERT_EXCEPTION(counters.GetCompletedInterrupted().GetValue("Nothingness"), yexception);
}
Y_UNIT_TEST(Empty)
{
const TString input = "{}";
TJobCounters counters(NodeFromYsonString(input));
UNIT_ASSERT_VALUES_EQUAL(counters.GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompleted().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedNonInterrupted().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetCompletedInterrupted().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAborted().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedNonScheduled().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetAbortedScheduled().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetLost().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetInvalidated().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetFailed().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetRunning().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetSuspended().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetPending().GetTotal(), 0);
UNIT_ASSERT_VALUES_EQUAL(counters.GetBlocked().GetTotal(), 0);
}
Y_UNIT_TEST(Broken)
{
UNIT_ASSERT_EXCEPTION_CONTAINS(TJobCounters(TNode()), yexception, "TJobCounters");
UNIT_ASSERT_EXCEPTION_CONTAINS(TJobCounters(TNode(1)), yexception, "TJobCounters");
UNIT_ASSERT_EXCEPTION_CONTAINS(TJobCounters(TNode(1.0)), yexception, "TJobCounters");
UNIT_ASSERT_EXCEPTION_CONTAINS(TJobCounters(TNode("Whatever")), yexception, "TJobCounters");
}
}
|
