class TestDocument(TestCase):
def setUp(self):
self.document = Document()
def tearDown(self):
try:
os.remove("test.xml")
except OSError:
pass
def test_record_returns_object_of_type_Record(self):
record = self.document.record(model='test')
self.assertEqual(type(record), Record)
def test_write_creates_file_and_writes_document(self):
self.document.write("test.xml")
tree = ElementTree()
tree.parse("test.xml")
root = tree.getroot()
data = root.find("data")
self.assertEqual(root.tag, "openerp")
self.assertEqual(data.tag, "data")
def test_pretty_print_return_string(self):
xml = ('<?xml version="1.0" ?>\n<openerp>\n'
'\t<data noupdate="1"/>\n</openerp>\n')
pretty_string = self.document._pretty_print()
self.assertEqual(pretty_string, xml)
class TestDocument(TestCase):
def setUp(self):
self.document = Document()
def tearDown(self):
try:
os.remove("test.xml")
except OSError:
pass
def test_record_returns_object_of_type_Record(self):
record = self.document.record(model='test')
self.assertEqual(type(record), Record)
def test_write_creates_file_and_writes_document(self):
self.document.write("test.xml")
tree = ElementTree()
tree.parse("test.xml")
root = tree.getroot()
data = root.find("data")
self.assertEqual(root.tag, "openerp")
self.assertEqual(data.tag, "data")
def test_pretty_print_return_string(self):
xml = ('<?xml version="1.0" ?>\n<openerp>\n'
'\t<data noupdate="1"/>\n</openerp>\n')
pretty_string = self.document._pretty_print()
self.assertEqual(pretty_string, xml)
def setUp(self):
self.document = Document()
def setUp(self):
self.document = Document()
