def test_date_filter_bson_parse_args(self): args = date_filter_bson.parse_args([ '-i', '~/data1.bson', '-d1', '2016-01-19 09:47:00', '-d2', '2016-01-20 09:47:00', '-o', '~/data2.bson' ]) self.assertEqual(args.input, '~/data1.bson') self.assertEqual(args.output, '~/data2.bson') self.assertEqual(args.dateone, '2016-01-19 09:47:00') self.assertEqual(args.datetwo, '2016-01-20 09:47:00')
def test_date_filter_bson_date_filter_until(self): self.setUp() args = date_filter_bson.parse_args(['-i', os.path.dirname(os.path.abspath(__file__))+'/../test/test.bson', '-d2', '2016-01-18 00:00:00', '-o', os.path.dirname(os.path.abspath(__file__))+'/../test/output.bson']) date_filter_bson.date_filter(args.output, args.input, args.dateone, args.datetwo) count = 0 with open(os.path.dirname(os.path.abspath(__file__))+'/../test/output.bson', 'rb') as bsonfile_handle: iterator = decode_file_iter(bsonfile_handle) for line in iterator: count+=1 self.assertEqual(count, 21) #remove output self.tearDown self.tearDown()
def test_date_filter_bson_date_filter_until(self): self.setUp() args = date_filter_bson.parse_args([ '-i', os.path.dirname(os.path.abspath(__file__)) + '/../test/test.bson', '-d2', '2016-01-18 00:00:00', '-o', os.path.dirname(os.path.abspath(__file__)) + '/../test/output.bson' ]) date_filter_bson.date_filter(args.output, args.input, args.dateone, args.datetwo) count = 0 with open( os.path.dirname(os.path.abspath(__file__)) + '/../test/output.bson', 'rb') as bsonfile_handle: iterator = decode_file_iter(bsonfile_handle) for line in iterator: count += 1 self.assertEqual(count, 21) #remove output self.tearDown self.tearDown()
def test_date_filter_bson_parse_args(self): args = date_filter_bson.parse_args(['-i', '~/data1.bson', '-d1', '2016-01-19 09:47:00', '-d2', '2016-01-20 09:47:00', '-o', '~/data2.bson']) self.assertEqual(args.input,'~/data1.bson') self.assertEqual(args.output,'~/data2.bson') self.assertEqual(args.dateone,'2016-01-19 09:47:00') self.assertEqual(args.datetwo,'2016-01-20 09:47:00')