def test_writes_stat_to_writer(self, fake_init, fake_run):
fake_run.return_value = {"stat": 1}
fake_init.return_value = None
temp_fs = fsopendir("temp://")
hdf_partition = HDFPartition(temp_fs, path="temp.h5")
fake_reader = MagicMock()
fake_writer = MagicMock(spec=HDFWriter)
fake_set_stats = MagicMock()
fake_writer.__enter__ = lambda x: fake_set_stats
# FIXME: So complicated. Refactor.
with patch.object(hdf_partition, "_reader", fake_reader):
with patch.object(hdf_partition, "_writer", fake_writer):
hdf_partition.run_stats()
self.assertEqual(fake_set_stats.mock_calls, [call.set_stats({"stat": 1})])
def test_writes_stat_to_writer(self, fake_init, fake_run):
fake_run.return_value = {'stat': 1}
fake_init.return_value = None
temp_fs = fsopendir('temp://')
hdf_partition = HDFPartition(temp_fs, path='temp.h5')
fake_reader = MagicMock()
fake_writer = MagicMock(spec=HDFWriter)
fake_set_stats = MagicMock()
fake_writer.__enter__ = lambda x: fake_set_stats
# FIXME: So complicated. Refactor.
with patch.object(hdf_partition, '_reader', fake_reader):
with patch.object(hdf_partition, '_writer', fake_writer):
hdf_partition.run_stats()
self.assertEqual(fake_set_stats.mock_calls,
[call.set_stats({'stat': 1})])
def test_creates_stat_from_reader(self, fake_init, fake_run):
fake_init.return_value = None
fake_run.return_value = {"a": 1}
temp_fs = fsopendir("temp://")
hdf_partition = HDFPartition(temp_fs, path="temp.h5")
with patch.object(hdf_partition, "_reader", MagicMock()):
with patch.object(hdf_partition, "_writer", MagicMock()):
ret = hdf_partition.run_stats()
self.assertEqual(ret, {"a": 1})
def test_creates_stat_from_reader(self, fake_init, fake_run):
fake_init.return_value = None
fake_run.return_value = {'a': 1}
temp_fs = fsopendir('temp://')
hdf_partition = HDFPartition(temp_fs, path='temp.h5')
with patch.object(hdf_partition, '_reader', MagicMock()):
with patch.object(hdf_partition, '_writer', MagicMock()):
ret = hdf_partition.run_stats()
self.assertEqual(ret, {'a': 1})