From ea1fe29a5e6259b7d06ceb52ac4ac8722c7f8120 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= Date: Thu, 16 Apr 2015 13:31:41 +0200 Subject: [PATCH] Fixed unit tests again --- tests/filemanager/test_localstorage.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/filemanager/test_localstorage.py b/tests/filemanager/test_localstorage.py index c722d900..87bc9b9c 100644 --- a/tests/filemanager/test_localstorage.py +++ b/tests/filemanager/test_localstorage.py @@ -7,6 +7,7 @@ __copyright__ = "Copyright (C) 2014 The OctoPrint Project - Released under terms import unittest import os +import mock from ddt import ddt, unpack, data @@ -43,10 +44,27 @@ class LocalStorageTest(unittest.TestCase): self.basefolder = tempfile.mkdtemp() self.storage = octoprint.filemanager.storage.LocalFileStorage(self.basefolder) + # mock file manager module + self.filemanager_patcher = mock.patch("octoprint.filemanager") + self.filemanager = self.filemanager_patcher.start() + + self.filemanager.valid_file_type.return_value = True + + def get_file_type(name): + if name.lower().endswith(".stl"): + return ["model", "stl"] + elif name.lower().endswith(".gco") or name.lower().endswith(".gcode") or name.lower.endswith(".g"): + return ["machinecode", "gcode"] + else: + return None + self.filemanager.get_file_type.side_effect = get_file_type + def tearDown(self): import shutil shutil.rmtree(self.basefolder) + self.filemanager_patcher.stop() + def test_add_file(self): self._add_file("bp_case.stl", "bp_case.stl", FILE_BP_CASE_STL)