blob: ae0a27354ebc6d60443aedf808c634c0307a5e70 (
plain) (
blame)
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
|
#include <yql/essentials/public/purecalc/purecalc.h>
#include <yql/essentials/public/purecalc/common/default_runtime_settings.h>
#include <yql/essentials/public/purecalc/io_specs/protobuf/spec.h>
#include <yql/essentials/public/purecalc/ut/protos/test_structs.pb.h>
#include <yql/essentials/public/purecalc/ut/empty_stream.h>
#include <yql/essentials/minikql/computation/mkql_computation_node.h>
#include <library/cpp/testing/unittest/registar.h>
namespace {
class TWorkerHolderOutputSpec: public NYql::NPureCalc::TOutputSpecBase {
public:
const NYT::TNode& GetSchema() const override {
static const NYT::TNode Entity = NYT::TNode::CreateEntity();
return Entity;
}
};
} // namespace
template <>
struct NYql::NPureCalc::TOutputSpecTraits<TWorkerHolderOutputSpec> {
static const constexpr bool IsPartial = false;
static const constexpr bool SupportPullListMode = true;
using TPullListReturnType = const NYql::TRuntimeSettings*;
static TPullListReturnType ConvertPullListWorkerToOutputType(
const TWorkerHolderOutputSpec&,
NYql::NPureCalc::TWorkerHolder<NYql::NPureCalc::IPullListWorker> worker)
{
return &worker->GetGraph().GetContext().RuntimeSettings;
}
};
Y_UNIT_TEST_SUITE(TestRuntimeSettings) {
Y_UNIT_TEST(DefaultRuntimeSettings) {
using namespace NYql::NPureCalc;
auto factory = MakeProgramFactory();
auto program = factory->MakePullListProgram(
TProtobufInputSpec<NPureCalcProto::TStringMessage>(),
TWorkerHolderOutputSpec(),
"SELECT X FROM Input",
ETranslationMode::SQL);
// The worker must use the default runtime settings singleton, not a custom copy.
UNIT_ASSERT_EQUAL(program->Apply(EmptyStream<NPureCalcProto::TStringMessage*>()),
NYql::NPureCalc::NPrivate::GetDefaultRuntimeSettings().Get());
}
} // Y_UNIT_TEST_SUITE(TestRuntimeSettings)
|
