tests/tests/compute_size.py

   1 import unittest
   2 from pysize.core import compute_size
   3 from pysize.core.compute_size import size_observable
   4 from pysize.core.pysize_global_fs_cache import drop_caches
   5
   6 class TestComputeSize(unittest.TestCase):
   7     def testSizeDir(self):
   8         drop_caches()
   9         self.assertEqual(compute_size.slow('/tmp/pysize_example_dir'), 4608000)
  10         # Retry using cached results
  11         self.assertEqual(compute_size.slow('/tmp/pysize_example_dir'), 4608000)
  12
  13     def testSizeFile(self):
  14         drop_caches()
  15         size = compute_size.slow('/tmp/pysize_example_dir/unreadable_file')
  16         self.assertEqual(size, 12288)
  17
  18     def testSizeNoSuchPath(self):
  19         drop_caches()
  20         self.assertEqual(compute_size.slow('/no such/ path'), 0)
  21
  22     def testSizeCache(self):
  23         def inc_nr_calls():
  24             self.nr_calls += 1
  25         size_observable.add_observer(inc_nr_calls)
  26         drop_caches()
  27         self.nr_calls = 0
  28         self.assertEqual(compute_size.slow('/tmp/pysize_example_dir'), 4608000)
  29         self.assertEqual(self.nr_calls, 10132)
  30         self.assertEqual(compute_size.slow('/tmp/pysize_example_dir'), 4608000)
  31         self.assertEqual(self.nr_calls, 10133)
  32         size_observable.del_observer(inc_nr_calls)
  33         self.assertEqual(compute_size.slow('/tmp/pysize_example_dir'), 4608000)
  34         self.assertEqual(self.nr_calls, 10133)
  35
  36 TESTS = (TestComputeSize,)