def setUp(self): self.channel = channels.get_channel('polls') self.storage = self.channel.storage try: shutil.rmtree(self.storage.location) except OSError: pass
def inline(name, mimetype, offset, limit): datas = [] channel = get_channel(name) for obj in channel.get_query(offset, limit): datas.append([channel.format(column, obj) for column in channel.columns]) channel.write(datas, mimetype, offset, limit) return offset, limit
def test_inline_task(self): tasks.inline('polls', 'csv', 0, 100) # NOQA file_root = self.channel.get_file_root('csv', 0, 100) file = self.storage.open(file_root) lines = file.readlines() self.assertEqual(len(lines), 101) self.assertEqual(len(lines[0].split(',')), len(channels.get_channel('polls').columns))
def test_inline_task(self): with patch('__builtin__.open', create=True) as mock_open: mock_open.return_value = MagicMock(spec=file) tasks.inline('polls', 'csv', 0, 100) # NOQA file_handle = mock_open.return_value.__enter__.return_value datas = file_handle.write._mock_call_args[0][0].split('\n') self.assertEqual(len(datas), 102) self.assertEqual(len(datas[0].split(',')), len(channels.get_channel('polls').columns))
def setUp(self): self.channel = channels.get_channel('polls')
def test_get_channel(self): channel = channels.get_channel('polls') self.assertTrue(isinstance(channel, PollExport))
def generate_subtasks_builder(name, mimetype, chunks): return [inline.subtask((name, mimetype, i, chunks)) for i in xrange(0, get_channel(name).get_count(), chunks)]
def compute(offsets, **kwargs): channel = get_channel(kwargs.get('name')) channel.combine(offsets, kwargs.get('mimetype'))