1 import os, sys
2 import marshal
3
4 from ModelParser import *
5
7
9 """Test the class def extractor"""
10 ucr = UMLClassReader()
11 dl = ucr.parse("sample/sample.xmi")
12
13 assert ("User" == dl['classDefs'][0]['name'])
14
15 assert ("uid", dl['classDefs'][0]['attrs'][0])
16 assert ("userName", dl['classDefs'][0]['attrs'][1])
17
18 assert ("Role" == dl['classDefs'][1]['name'])
19
20 assert ("rid", dl['classDefs'][1]['attrs'][0])
21
22
23
25 """Test parser and marshaling of data"""
26 mp = ModelParser()
27 mp.parse("sample/sample.xmi")
28
29 inf = open('data/db.model')
30 a = marshal.load(inf)
31 inf.close()
32 assert ("User" == a['classDefs'][0]['name'])
33