def testFile(self):
explorer = FileExplorer(
Namespace(path="tests/samples/sample-data.csv",
catalog={"format": "ascii_table"}))
explorer.scan()
result = explorer.get_dict()
self._check(result)
def testFile(self):
explorer = FileExplorer(
Namespace(path="tests/samples/sample-data.csv",
catalog={"format": "ascii_table"}))
explorer.scan()
result = explorer.get_dict()
self.assertEqual(len(result["files"]), 1)
self.assertEqual(result["files"][0]["Mime/Type"], "text/plain")
self.assertEqual(result["files"][0]["path"],
"tests/samples/sample-data.csv")
self.assertEqual(len(result["files"][0]["pii"]), 5)
def test_file_dispatch(self):
with mock.patch('piicatcher.explorer.files.FileExplorer.scan',
autospec=True) as mock_scan_method:
with mock.patch(
'piicatcher.explorer.files.FileExplorer.get_tabular',
autospec=True) as mock_tabular_method:
with mock.patch('piicatcher.explorer.files.tableprint',
autospec=True) as MockTablePrint:
FileExplorer.dispatch(
Namespace(path='/a/b/c', output_format='ascii_table'))
mock_scan_method.assert_called_once()
mock_tabular_method.assert_called_once()
MockTablePrint.table.assert_called_once()
def test_file_dispatch(self):
with mock.patch("piicatcher.explorer.files.FileExplorer.scan",
autospec=True) as mock_scan_method:
with mock.patch(
"piicatcher.explorer.files.FileExplorer.get_tabular",
autospec=True) as mock_tabular_method:
with mock.patch("piicatcher.explorer.files.tableprint",
autospec=True) as MockTablePrint:
FileExplorer.dispatch(
Namespace(path="/a/b/c",
catalog={"format": "ascii_table"}))
mock_scan_method.assert_called_once()
mock_tabular_method.assert_called_once()
MockTablePrint.table.assert_called_once()
def testDirectory(self):
explorer = FileExplorer(
Namespace(path="tests/samples", catalog={"format": "ascii_table"}))
explorer.scan()
result = explorer.get_dict()
self.assertEqual(len(result["files"]), 2)
self.assertEqual(result["files"][0]["Mime/Type"], "application/gzip")
self.assertEqual(result["files"][0]["path"],
"tests/samples/sample-data-2.csv.gz")
self.assertEqual(len(result["files"][1]["pii"]), 5)
self.assertEqual(result["files"][1]["Mime/Type"], "text/plain")
self.assertEqual(result["files"][1]["path"],
"tests/samples/sample-data.csv")
self.assertEqual(len(result["files"][1]["pii"]), 5)
def setUpClass(cls):
cls.explorer = FileExplorer("/tmp")
f1 = File("/tmp/1", "text/plain")
f1._pii.add(PiiTypes.BIRTH_DATE)
f2 = File("/tmp/2", "application/pdf")
f2._pii.add(PiiTypes.UNSUPPORTED)
cls.explorer._files.append(f1)
cls.explorer._files.append(f2)
def testFile(self):
explorer = FileExplorer("tests/samples/sample-data.csv")
explorer.scan()
result = explorer.get_dict()
self._check(result)
def testDirectory(self):
explorer = FileExplorer("tests/samples")
explorer.scan()
result = explorer.get_dict()
self._check(result)
def testDirectory(self):
explorer = FileExplorer(
Namespace(path="tests/samples", catalog={'format': 'ascii_table'}))
explorer.scan()
result = explorer.get_dict()
self._check(result)