WHAT
This commit is contained in:
0
code/tests/__init__.py
Normal file
0
code/tests/__init__.py
Normal file
0
code/tests/fixtures/__init__.py
vendored
Normal file
0
code/tests/fixtures/__init__.py
vendored
Normal file
15
code/tests/fixtures/shared_fixtures.py
vendored
Normal file
15
code/tests/fixtures/shared_fixtures.py
vendored
Normal file
@@ -0,0 +1,15 @@
|
||||
import pytest
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
@pytest.fixture(scope='session')
|
||||
def tmp_dir():
|
||||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||||
yield Path(tmp_dir)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_device_metadata_json_(tmp_dir):
|
||||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||||
pass
|
||||
0
code/tests/models/test_capture_metadata_model.py
Normal file
0
code/tests/models/test_capture_metadata_model.py
Normal file
0
code/tests/models/test_device_metadata_model.py
Normal file
0
code/tests/models/test_device_metadata_model.py
Normal file
47
code/tests/subcommands/test_add_device.py
Normal file
47
code/tests/subcommands/test_add_device.py
Normal file
@@ -0,0 +1,47 @@
|
||||
import sys
|
||||
import unittest
|
||||
from io import StringIO
|
||||
from unittest.mock import patch, MagicMock
|
||||
from pathlib import Path
|
||||
from iottb.definitions import DEVICE_METADATA_FILE
|
||||
import shutil
|
||||
from iottb.__main__ import main
|
||||
|
||||
|
||||
class TestDeviceMetadataFileCreation(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.test_dir = Path('/tmp/iottbtest/test_add_device')
|
||||
self.test_dir.mkdir(parents=True, exist_ok=True)
|
||||
# self.captured_output = StringIO()
|
||||
# sys.stdout = self.captured_output
|
||||
|
||||
def tearDown(self):
|
||||
# shutil.rmtree(str(self.test_dir))
|
||||
for item in self.test_dir.iterdir():
|
||||
if item.is_dir():
|
||||
item.rmdir()
|
||||
else:
|
||||
item.unlink()
|
||||
self.test_dir.rmdir()
|
||||
# sys.stdout = sys.__stdout__
|
||||
|
||||
@patch('builtins.input', side_effect=['iPhone 14', 'y', 'y'])
|
||||
def test_guided_device_setup(self, mock_input):
|
||||
sys.argv = ['__main__.py', 'add', '--root_dir', str(self.test_dir), '--guided']
|
||||
main()
|
||||
expected_file = self.test_dir / DEVICE_METADATA_FILE
|
||||
self.assertTrue(expected_file.exists()), f'Expected file not created: {expected_file}'
|
||||
|
||||
@patch('builtins.input', side_effect=['y']) # need mock_input else wont work
|
||||
def test_device_setup(self, mock_input):
|
||||
sys.argv = ['__main__.py', 'add', '--root_dir', str(self.test_dir), '--name', 'iPhone 14']
|
||||
main()
|
||||
expected_file = self.test_dir / DEVICE_METADATA_FILE
|
||||
self.assertTrue(expected_file.exists()), f'Expected file not created: {expected_file}'
|
||||
|
||||
def test_add_when_file_exists(self):
|
||||
# TODO
|
||||
pass
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
2
code/tests/test_capture_metadata_model.py
Normal file
2
code/tests/test_capture_metadata_model.py
Normal file
@@ -0,0 +1,2 @@
|
||||
def test_save_to_json():
|
||||
assert False
|
||||
0
code/tests/test_main.py
Normal file
0
code/tests/test_main.py
Normal file
2
code/tests/utils/test_capture_metadata_utils.py
Normal file
2
code/tests/utils/test_capture_metadata_utils.py
Normal file
@@ -0,0 +1,2 @@
|
||||
|
||||
|
||||
0
code/tests/utils/test_device_metadata_utils.py
Normal file
0
code/tests/utils/test_device_metadata_utils.py
Normal file
0
code/tests/utils/test_tcpdump_utils.py
Normal file
0
code/tests/utils/test_tcpdump_utils.py
Normal file
Reference in New Issue
Block a user