class KeyFixTestCase(unittest.TestCase): def setUp(self): self.t = self.assertTrue self.inst = self.assertIsInstance self.inserter = BulkInserter(host="localhost",port=27017,database="reddit_stream_test",collection="users") def tearDown(self): self.inserter.source.drop() def test_keyfix_characters(self): count = 0 for ind, user in enumerate(annotation.user_reader("data/test_key.json")): self.inserter.insert(item=user) count = ind self.inserter.send() count_2 = 0 for ind, user in enumerate(self.inserter.source.find()): count_2 = ind self.t(count == count_2)
def main(argv): inserter = BulkInserter(source=Source(host="localhost", port=27017, database=argv[1], collection=argv[2])) for doc in annotation.sbr_reader(argv[0]): inserter.insert(item=doc) inserter.send()