Add cleanup tests

This commit is contained in:
Delirious Lettuce 2017-07-07 23:14:47 -06:00
parent 17f301022b
commit cf05204e65

View file

@ -16,6 +16,7 @@ GNU General Public License for more details.
You should have received a copy of the GNU General Public License You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>. along with this program. If not, see <http://www.gnu.org/licenses/>.
""" """
import os
import pytest import pytest
@ -32,7 +33,6 @@ class MyOnion:
return 'test_service_id.onion' return 'test_service_id.onion'
# pytest > 2.9 only needs pytest.fixture
@pytest.fixture() @pytest.fixture()
def onionshare_obj(): def onionshare_obj():
return onionshare.OnionShare(MyOnion()) return onionshare.OnionShare(MyOnion())
@ -40,7 +40,6 @@ def onionshare_obj():
class TestOnionShare: class TestOnionShare:
def test_init(self, onionshare_obj): def test_init(self, onionshare_obj):
# test onion? own fixture?
assert onionshare_obj.hidserv_dir is None assert onionshare_obj.hidserv_dir is None
assert onionshare_obj.onion_host is None assert onionshare_obj.onion_host is None
assert onionshare_obj.stealth is None assert onionshare_obj.stealth is None
@ -75,9 +74,10 @@ class TestOnionShare:
assert onionshare_obj.onion_host == '127.0.0.1:{}'.format( assert onionshare_obj.onion_host == '127.0.0.1:{}'.format(
onionshare_obj.port) onionshare_obj.port)
def test_cleanup(self, onionshare_obj): def test_cleanup(self, onionshare_obj, temp_dir_1024, temp_file_1024):
# create temporary files/directories and then remove them onionshare_obj.cleanup_filenames = [temp_dir_1024, temp_file_1024]
# use helper function from other file (use conftest.py???) onionshare_obj.cleanup()
onionshare_obj.cleanup_filenames = [] # add temporary filenames & dirs
# remove the files & dirs assert os.path.exists(temp_dir_1024) is False
assert os.path.exists(temp_dir_1024) is False
assert onionshare_obj.cleanup_filenames == [] assert onionshare_obj.cleanup_filenames == []