forked from geldata/gel-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_enum.py
More file actions
78 lines (62 loc) · 2.5 KB
/
test_enum.py
File metadata and controls
78 lines (62 loc) · 2.5 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
66
67
68
69
70
71
72
73
74
75
76
77
78
#
# This source file is part of the EdgeDB open source project.
#
# Copyright 2019-present MagicStack Inc. and the EdgeDB authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import uuid
import edgedb
from edgedb import _testbase as tb
class TestEnum(tb.AsyncQueryTestCase):
ISOLATED_METHODS = False
SETUP = '''
CREATE SCALAR TYPE CellType EXTENDING enum<'red', 'white'>;
CREATE SCALAR TYPE Color EXTENDING enum<'red', 'white'>;
'''
async def test_enum_01(self):
ct_red = await self.con.query_one('SELECT <CellType>"red"')
ct_white = await self.con.query_one('SELECT <CellType>"white"')
c_red = await self.con.query_one('SELECT <Color>"red"')
self.assertTrue(isinstance(ct_red, edgedb.EnumValue))
self.assertTrue(isinstance(ct_red.__tid__, uuid.UUID))
self.assertEqual(repr(ct_red), "<EnumValue 'red'>")
self.assertEqual(str(ct_red), 'red')
self.assertNotEqual(ct_red, 'red')
self.assertFalse(ct_red == 'red')
self.assertFalse(ct_red == c_red)
self.assertEqual(ct_red, ct_red)
self.assertNotEqual(ct_red, ct_white)
self.assertNotEqual(ct_red, c_red)
self.assertLess(ct_red, ct_white)
self.assertLessEqual(ct_red, ct_red)
self.assertGreater(ct_white, ct_red)
self.assertGreaterEqual(ct_white, ct_white)
with self.assertRaises(TypeError):
ct_red < 'red'
with self.assertRaises(TypeError):
ct_red > 'red'
with self.assertRaises(TypeError):
ct_red <= 'red'
with self.assertRaises(TypeError):
ct_red >= 'red'
with self.assertRaises(TypeError):
ct_red < c_red
with self.assertRaises(TypeError):
ct_red > c_red
with self.assertRaises(TypeError):
ct_red <= c_red
with self.assertRaises(TypeError):
ct_red >= c_red
self.assertNotEqual(hash(ct_red), hash(c_red))
self.assertNotEqual(hash(ct_red), hash('red'))