def test_getLayerColor(self, fake_dwg, *arg): app_data = self.fake_appData() assert DxfDraw(app_data).getLayerColor('layer1') == 1 assert DxfDraw(app_data).getLayerColor() == '7' pass
def test_getLayer(self, fake_dwg, *arg): app_data = self.fake_appData() assert DxfDraw(app_data).getLayer('layer1') == "FAKE-LAYER-1" assert DxfDraw(app_data).getLayer() == '0' pass
def test_makeDxf(self, *arg): with patch.object(VPort, 'createVPortTable') as fake_VPort, \ patch.object(DxfDraw, "drawEntities") as fake_drawEntities: app_data = self.fake_appData() DxfDraw(app_data).makeDxf({}, 2) fake_VPort.assert_called_once() fake_drawEntities.assert_called_once()
def test_addDxfHatch(self, fake_dwg, fake_getLayer, fake_getLayerColor, *arg): app_data = self.fake_appData() fake_msp = Mock() type(fake_dwg.return_value).modelspace = fake_msp fake_add_hatch = Mock() type(fake_msp.return_value).add_hatch = fake_add_hatch fake_edit_boundary = Mock() type(fake_add_hatch.return_value).edit_boundary = fake_edit_boundary fake_add_polyline_path = Mock() fake_enter = Mock(side_effect=lambda: fake_add_polyline_path) fake_exit = Mock() type(fake_edit_boundary.return_value ).add_polyline_path = fake_add_polyline_path type(fake_edit_boundary.return_value).__enter__ = fake_enter type(fake_edit_boundary.return_value).__exit__ = fake_exit DxfDraw(app_data).addDxfHatch(Mock()) fake_dwg.assert_called_once() fake_getLayer.assert_called_once() fake_getLayerColor.assert_called_once() fake_msp.assert_called_once() fake_add_hatch.assert_called_once() fake_edit_boundary.assert_called_once() fake_enter.assert_called_once() fake_exit.assert_called_once() # fake_add_polyline_path.assert_called_once() pass
def test_addCircle(self, fake_dwg, fake_getLayer, *arg): app_data = self.fake_appData() fake_msp = Mock() type(fake_dwg.return_value).modelspace = fake_msp fake_add_circle = Mock() type(fake_msp.return_value).add_circle = fake_add_circle DxfDraw(app_data).addDxfCircle(Mock()) fake_dwg.assert_called_once() fake_getLayer.assert_called_once() fake_msp.assert_called_once() fake_add_circle.assert_called_once()
def test_addDxfText(self, fake_dwg, fake_getLayer, *arg): app_data = self.fake_appData() fake_msp = Mock() type(fake_dwg.return_value).modelspace = fake_msp fake_add_text = Mock() type(fake_msp.return_value).add_text = fake_add_text DxfDraw(app_data).addDxfText(Mock()) fake_dwg.assert_called_once() fake_getLayer.assert_called_once() fake_msp.assert_called_once() fake_add_text.assert_called_once() pass
def test_addDxfLine(self, fake_dwg, fake_getLayer, *arg): pt1 = (0., 0., 0.) pt2 = (20., 0., 3) line = EntityLine(pt1, pt2, "fake_layer") fake_msp = Mock() type(fake_dwg.return_value).modelspace = fake_msp fake_add_line = Mock() type(fake_msp.return_value).add_line = fake_add_line app_data = self.fake_appData() DxfDraw(app_data).addDxfLine(line) fake_dwg.assert_called_once() fake_getLayer.assert_called_once() fake_msp.assert_called_once() fake_add_line.assert_called_once()
def test_getFileName(self, fake_dateTimeString, *arg): fake_dateTimeString.return_value = "190529_1536" app_data = self.fake_appData() assert DxfDraw(app_data).getFileName() == "190529_1536-detail.dxf"
def test_makeDxfException(self, *arg): with patch.object(VPort, 'createVPortTable'), \ patch.object(DxfDraw, "drawEntities"): app_data = self.fake_appData() assert DxfDraw(app_data).makeDxf({}, 2) == None
def test_addDxfEntity(self, fake_show_dimension): app_data = self.fake_appData() DxfDraw(app_data).addDimEntity(Mock()) pass