d32631cc54
An AttributeError is raised when `direction=Vector(0,0,0)` and obj is
an Arch::Space on line: bead9bb938/src/Mod/Draft/draftfunctions/svg.py (L799)
This patch checks if early on if the direction vector and raises a
ValueError with a description of what has gone wrong.
A caveat with this solution is that this new behaviour might break old
code which depends on that invalid directions can be used.
120 lines
4.8 KiB
Python
120 lines
4.8 KiB
Python
# ***************************************************************************
|
|
# * Copyright (c) 2013 Yorik van Havre <yorik@uncreated.net> *
|
|
# * Copyright (c) 2019 Eliud Cabrera Castillo <e.cabrera-castillo@tum.de> *
|
|
# * *
|
|
# * This file is part of the FreeCAD CAx development system. *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * FreeCAD is distributed in the hope that it will be useful, *
|
|
# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
# * GNU Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with FreeCAD; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
"""Unit tests for the Draft Workbench, SVG import and export tests."""
|
|
## @package test_svg
|
|
# \ingroup drafttests
|
|
# \brief Unit tests for the Draft Workbench, SVG import and export tests.
|
|
|
|
## \addtogroup drafttests
|
|
# @{
|
|
import os
|
|
import unittest
|
|
|
|
import FreeCAD as App
|
|
import Draft
|
|
import drafttests.auxiliary as aux
|
|
|
|
from draftutils.messages import _msg
|
|
|
|
|
|
class DraftSVG(unittest.TestCase):
|
|
"""Test reading and writing of SVGs with Draft."""
|
|
|
|
def setUp(self):
|
|
"""Set up a new document to hold the tests.
|
|
|
|
This is executed before every test, so we create a document
|
|
to hold the objects.
|
|
"""
|
|
aux.draw_header()
|
|
self.doc_name = self.__class__.__name__
|
|
if App.ActiveDocument:
|
|
if App.ActiveDocument.Name != self.doc_name:
|
|
App.newDocument(self.doc_name)
|
|
else:
|
|
App.newDocument(self.doc_name)
|
|
App.setActiveDocument(self.doc_name)
|
|
self.doc = App.ActiveDocument
|
|
_msg(" Temporary document '{}'".format(self.doc_name))
|
|
|
|
def test_read_svg(self):
|
|
"""Read an SVG file and import its elements as Draft objects."""
|
|
operation = "importSVG.import"
|
|
_msg(" Test '{}'".format(operation))
|
|
_msg(" This test requires an SVG file to read.")
|
|
|
|
file = 'Mod/Draft/drafttest/test.svg'
|
|
in_file = os.path.join(App.getResourceDir(), file)
|
|
_msg(" file={}".format(in_file))
|
|
_msg(" exists={}".format(os.path.exists(in_file)))
|
|
|
|
Draft.import_svg = aux.fake_function
|
|
obj = Draft.import_svg(in_file)
|
|
self.assertTrue(obj, "'{}' failed".format(operation))
|
|
|
|
def test_export_svg(self):
|
|
"""Create some figures and export them to an SVG file."""
|
|
operation = "importSVG.export"
|
|
_msg(" Test '{}'".format(operation))
|
|
|
|
file = 'Mod/Draft/drafttest/out_test.svg'
|
|
out_file = os.path.join(App.getResourceDir(), file)
|
|
_msg(" file={}".format(out_file))
|
|
_msg(" exists={}".format(os.path.exists(out_file)))
|
|
|
|
Draft.export_svg = aux.fake_function
|
|
obj = Draft.export_svg(out_file)
|
|
self.assertTrue(obj, "'{}' failed".format(operation))
|
|
|
|
def test_get_svg_from_arch_space_with_zero_vector(self):
|
|
"""Try to get a svg string from an Arch Space with a zero-vector as direction."""
|
|
import Part
|
|
import Arch
|
|
import Draft
|
|
|
|
sb = Part.makeBox(1,1,1)
|
|
b = App.ActiveDocument.addObject('Part::Feature','Box')
|
|
b.Shape = sb
|
|
|
|
s = Arch.makeSpace(b)
|
|
App.ActiveDocument.recompute()
|
|
|
|
try:
|
|
Draft.get_svg(s, direction=App.Vector(0,0,0))
|
|
except AttributeError as err:
|
|
self.fail("Cryptic exception thrown: {}".format(err))
|
|
except ValueError as err:
|
|
App.Console.PrintLog("Exception thrown, OK: {}".format(err))
|
|
else:
|
|
self.fail("no exception thrown")
|
|
|
|
def tearDown(self):
|
|
"""Finish the test.
|
|
|
|
This is executed after each test, so we close the document.
|
|
"""
|
|
App.closeDocument(self.doc_name)
|
|
|
|
## @}
|