def setUp(self): from octoprint.settings import settings from octoprint.gcodefiles import GcodeManager self.settings = settings(True) self.manager = GcodeManager() self.filenames = []
class FileManipulationTestCase(unittest.TestCase): def setUp(self): from octoprint.settings import settings from octoprint.gcodefiles import GcodeManager self.settings = settings(True) self.manager = GcodeManager() self.filenames = [] def tearDown(self): for filename in self.filenames: self.manager.removeFile(filename) logging.info("REMOVED %s filenames" % str(len(self.filenames))) @patch('octoprint.settings.Settings.getBoolean') @patch('octoprint.slicers.cura.Cura.process_file') def test_add_stl_file_curaDisabled(self, process, getterMock): getterMock.return_value = False fake = Mock() fake.filename = "test_stl.stl" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) self.assertFalse(process.called) self.assertIsNone(result) self.assertTrue(done) @patch('octoprint.settings.Settings.getBoolean') @patch('octoprint.slicers.cura.Cura.process_file') def test_add_stl_file_curaEnabled(self, process, getterMock): getterMock.return_value = True fake = Mock() fake.filename = "test_stl.stl" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) getterMock.assert_called_once_with(["cura", "enabled"]) self.assertTrue(process.called) self.assertTrue("test_stl.gcode" == result) self.assertFalse(done) def test_add_gcode_file(self): fake = Mock() fake.filename = "test_stl.gcode" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) self.assertTrue(fake.filename == result) self.assertTrue(done)
class FileManipulationTestCase(unittest.TestCase): def setUp(self): from octoprint.settings import settings from octoprint.gcodefiles import GcodeManager self.settings = settings(True) self.manager = GcodeManager() self.filenames = [] def tearDown(self): for filename in self.filenames: self.manager.removeFile(filename) logging.info("REMOVED %s filenames" % str(len(self.filenames))) @patch('octoprint.settings.Settings.getBoolean') @patch('octoprint.slicers.cura.Cura.process_file') def test_add_stl_file_curaDisabled(self, process, getterMock): getterMock.return_value = False fake = Mock() fake.filename = "test_stl.stl" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) self.assertFalse(process.called) self.assertIsNone(result) self.assertTrue(done) @patch('octoprint.settings.Settings.getBoolean') @patch('octoprint.slicers.cura.Cura.process_file') def test_add_stl_file_curaEnabled(self, process, getterMock): getterMock.return_value = True fake = Mock() fake.filename = "test_stl.stl" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) getterMock.assert_called_once_with(["cura", "enabled"]) self.assertTrue(process.called) self.assertTrue(fake.filename == result) self.assertFalse(done) def test_add_gcode_file(self): fake = Mock() fake.filename = "test_stl.gcode" self.filenames.append(fake.filename) fake.__getitem__ = "SOMETHING" result, done = self.manager.addFile(fake, FileDestinations.LOCAL) logging.info("RESULT:%s" % str(result)) self.assertTrue(fake.filename == result) self.assertTrue(done)