blob: ab2864caa21efd509ededd18b5440fc2fff9f338 (
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
|
#include "watermark_tracker.h"
#include <util/system/types.h>
#include <library/cpp/testing/unittest/registar.h>
namespace NKikimr {
namespace NMiniKQL {
Y_UNIT_TEST_SUITE(TWatermarkTrackerTest) {
Y_UNIT_TEST(WhenNotGranularShouldNotMoveWatermark) {
TWatermarkTracker tracker(15, 10);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(25), 10);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(26), std::nullopt);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(30), std::nullopt);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(34), std::nullopt);
}
Y_UNIT_TEST(WhenGranularShouldMoveWatermark) {
TWatermarkTracker tracker(15, 10);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(35), 20);
UNIT_ASSERT_EQUAL(tracker.HandleNextEventTime(45), 30);
}
}
} // NMiniKQL
} // NKikimr
|