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
|
from unittest import TestCase
from a38 import fields, models
class Sample(models.Model):
name = fields.StringField()
value = fields.IntegerField()
class Sample1(models.Model):
name = fields.StringField()
type = fields.StringField(choices=("A", "B"))
class TestModel(TestCase):
def test_assignment(self):
o = Sample()
# Values are cleaned on assignment
o.name = 12
o.value = "42"
self.assertEqual(o.name, "12")
self.assertEqual(o.value, 42)
def test_misspelled_field(self):
o = Sample()
with self.assertRaises(AttributeError):
o.nome = "foo"
def test_clean_value(self):
# Assign from a model
val = Sample.clean_value(Sample1("foo", "A"))
self.assertIsInstance(val, Sample)
self.assertEqual(val.name, "foo")
self.assertIsNone(val.value)
# Assign from a dict
val = Sample.clean_value({"name": "foo", "type": "A"})
self.assertIsInstance(val, Sample)
self.assertEqual(val.name, "foo")
self.assertIsNone(val.value)
self.assertIsNone(Sample.clean_value(None))
with self.assertRaises(TypeError):
Sample.clean_value("foo")
def test_compare(self):
self.assertEqual(Sample("test", 7), Sample("test", 7))
self.assertEqual(Sample(), None)
self.assertNotEqual(Sample("test", 7), Sample("test", 6))
self.assertNotEqual(Sample("test", 7), None)
self.assertLess(Sample("test", 6), Sample("test", 7))
self.assertLessEqual(Sample("test", 6), Sample("test", 7))
self.assertLessEqual(Sample("test", 7), Sample("test", 7))
self.assertLessEqual(Sample(), None)
self.assertGreater(Sample("test", 7), Sample("test", 6))
self.assertGreater(Sample("test", 7), None)
self.assertGreaterEqual(Sample("test", 7), Sample("test", 6))
self.assertGreaterEqual(Sample("test", 7), Sample("test", 7))
self.assertGreaterEqual(Sample("test", 7), None)
self.assertGreaterEqual(Sample(), None)
|
