forked from feldera/feldera
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_program.py
More file actions
65 lines (50 loc) · 1.86 KB
/
test_program.py
File metadata and controls
65 lines (50 loc) · 1.86 KB
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
import unittest
import uuid
import logging
import sys
from tests import TEST_CLIENT
from feldera.rest.program import Program
from feldera.enums import CompilationProfile
from typing import Optional
NAME = str(uuid.uuid4())
class TestProgram(unittest.TestCase):
root = logging.getLogger()
root.setLevel(logging.DEBUG)
handler = logging.StreamHandler(sys.stdout)
handler.setLevel(logging.DEBUG)
root.addHandler(handler)
def test_compile_program(self, name: str = NAME, config: Optional[dict] = None, delete: bool = True):
sql = """
CREATE TABLE test_table (
id INT,
name VARCHAR(255)
);
CREATE VIEW V AS SELECT * FROM test_table;
"""
program = Program(name, sql)
TEST_CLIENT.compile_program(program, config)
assert program.version == 1
if delete:
TEST_CLIENT.delete_program(name)
def test_delete_program(self, name: str = NAME):
self.test_compile_program(name, delete=False)
TEST_CLIENT.delete_program(name)
def test_list_programs(self):
self.test_compile_program(NAME, delete=False)
programs = TEST_CLIENT.programs()
assert len(programs) > 0
assert NAME in [p.name for p in programs]
self.test_delete_program(NAME)
def test_get_program(self):
self.test_compile_program(NAME, delete=False)
program = TEST_CLIENT.get_program(NAME)
assert program.name == NAME
assert program.version == 1
self.test_delete_program(NAME)
def __test_compilation_profile(self, profile: CompilationProfile = None):
self.test_compile_program(str(uuid.uuid4()), config={"profile": profile.value}, delete=True)
def test_compilation_profiles(self):
for profile in CompilationProfile:
self.__test_compilation_profile(profile)
if __name__ == '__main__':
unittest.main()