Python FileExplorer Examples

Example #1

File: test_file_explorer.py Project: SRS-Business-Solutions/tokern_piicatcher

 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)

Example #2

 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)

Example #3

 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()

Example #4

File: test_file_explorer.py Project: SRS-Business-Solutions/tokern_piicatcher

 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()

Example #5

 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)

Example #6

    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)

Example #7

 def testFile(self):
     explorer = FileExplorer("tests/samples/sample-data.csv")
     explorer.scan()
     result = explorer.get_dict()
     self._check(result)

Example #8

 def testDirectory(self):
     explorer = FileExplorer("tests/samples")
     explorer.scan()
     result = explorer.get_dict()
     self._check(result)

Example #9

 def testDirectory(self):
     explorer = FileExplorer(
         Namespace(path="tests/samples", catalog={'format': 'ascii_table'}))
     explorer.scan()
     result = explorer.get_dict()
     self._check(result)