Revision: 3274
Author: pekka.klarck
Date: Wed May 12 03:58:31 2010
Log: imports
http://code.google.com/p/robotframework/source/detail?r=3274
Modified:
/trunk/src/robot/parsing/newmodel.py
/trunk/utest/parsing/test_model.py
=======================================
--- /trunk/src/robot/parsing/newmodel.py Wed May 12 03:43:07 2010
+++ /trunk/src/robot/parsing/newmodel.py Wed May 12 03:58:31 2010
@@ -49,6 +49,15 @@
self.default_tags = Tags()
self.imports = []
+ def add_library(self, value):
+ self.imports.append(Library(value))
+
+ def add_resource(self, value):
+ self.imports.append(Resource(value))
+
+ def add_variables(self, value):
+ self.imports.append(Variables(value))
+
def __iter__(self):
for setting in [self.doc, self.suite_setup, self.suite_teardown,
self.metadata, self.test_setup, self.test_teardown,
@@ -125,6 +134,20 @@
class Return(Setting):
pass
+class Import(Setting):
+
+ def __init__(self, value):
+ self.value = value
+
+class Library(Import):
+ pass
+
+class Resource(Import):
+ pass
+
+class Variables(Import):
+ pass
+
class Variable(object):
=======================================
--- /trunk/utest/parsing/test_model.py Wed May 12 03:43:07 2010
+++ /trunk/utest/parsing/test_model.py Wed May 12 03:58:31 2010
@@ -50,6 +50,14 @@
self.table.doc.set(['hello', 'world'])
assert_equal(self.table.doc.value, 'hello world')
+ def test_imports(self):
+ self.table.add_library(['Name', 'arg'])
+ self.table.add_resource(['reso.txt'])
+ self.table.add_variables(['varz.py', 'a1', 'a2'])
+ self.table.add_resource(['reso2.txt'])
+ assert_true(len(self.table.imports), 4)
+ assert_true(all(isinstance(im, Import) for im in
self.table.imports))
+
class TestVariableTable(unittest.TestCase):
