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')
