yql/essentials/public/purecalc/ut/fake_spec.h


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

#pragma once

#include <yql/essentials/public/purecalc/purecalc.h>

namespace NYql {
    namespace NPureCalc {
        class TFakeInputSpec: public TInputSpecBase {
        public:
            TVector<NYT::TNode> Schemas = {NYT::TNode::CreateList()};

        public:
            const TVector<NYT::TNode>& GetSchemas() const override {
                return Schemas;
            }
        };

        class TFakeOutputSpec: public TOutputSpecBase {
        public:
            NYT::TNode Schema = NYT::TNode::CreateList();

        public:
            const NYT::TNode& GetSchema() const override {
                return Schema;
            }
        };

        template <>
        struct TInputSpecTraits<TFakeInputSpec> {
            static const constexpr bool IsPartial = false;

            static const constexpr bool SupportPullStreamMode = false;
            static const constexpr bool SupportPullListMode = false;
            static const constexpr bool SupportPushStreamMode = false;

            using TConsumerType = void;
        };

        template <>
        struct TOutputSpecTraits<TFakeOutputSpec> {
            static const constexpr bool IsPartial = false;

            static const constexpr bool SupportPullStreamMode = false;
            static const constexpr bool SupportPullListMode = false;
            static const constexpr bool SupportPushStreamMode = false;

            using TPullStreamReturnType = void;
            using TPullListReturnType = void;
        };

        NYT::TNode MakeFakeSchema(bool pg = false);
        TFakeInputSpec FakeIS(ui32 inputsNumber = 1, bool pg = false);
        TFakeOutputSpec FakeOS(bool pg = false);
    }
}