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
|
"""
Test case that checks the working of the utils/command/gen_uml.py module.
"""
import unittest
from utils.command.gen_uml import generate
class PseudoFile(object):
def __init__(self):
self.data = ''
def write(self, data):
self.data += data
def close(self):
pass
class GenUmlTestCase(unittest.TestCase):
def test_loading(self):
model_file = 'tests/test-model.gaphor'
outfile = PseudoFile()
generate(model_file, outfile)
assert outfile.data == GENERATED, '"""%s"""' % outfile.data
GENERATED = """# This file is generated by build_uml.py. DO NOT EDIT!
from properties import association, attribute, enumeration, derived, derivedunion, redefine
# class 'ValSpec' has been stereotyped as 'SimpleAttribute'
# class 'ShouldNotShowUp' has been stereotyped as 'SimpleAttribute' too
class C(object): pass
class D(C): pass
class Element(object): pass
class SubClass(Element): pass
C.attr = attribute('attr', str)
# 'SubClass.value' is a simple attribute
SubClass.value = attribute('value', str)
C.name1 = association('name1', SubClass, opposite='name2')
SubClass.name2 = association('name2', C, opposite='name1')
C.base = association('base', SubClass, opposite='abstract')
D.name3 = association('name3', SubClass, opposite='name4')
D.subbase = association('subbase', SubClass, opposite='concrete')
SubClass.concrete = association('concrete', D, opposite='subbase')
SubClass.abstract = derivedunion('abstract', C, 0, '*', SubClass.concrete)
SubClass.name4 = redefine(SubClass, 'name4', D, name2)
"""
# vim:sw=4:et:ai
|
