Esempio n. 1
0
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import os

import unittest
from unittest.mock import patch

import asyncio
import collections
import gzip
import json

from settings import Config
Config.test_config()

from amzn_review.data.uploader import (FileItem, load_json_content,
                                       MetadataUploader, FileExtractor,
                                       MultipleCategoryFileExtractor,
                                       ReviewUploader, _run,
                                       _create_upload_coro)


class _CollectData(object):
    """Counter items to be sent."""
    def __init__(self, key=None, index=0):
        """Target item to store.
        If `key` is given, store by kwargs by the `key`,
        otherwise, store item from args by `index`."""
        self.key = key
        self.index = 0