def test_create_metadata_object_cv():
input_file_name = "tests/data/head_object_input_cv.json"
metadata_output_file_name = "tests/data/metadata_output_cv.json"
s3_key = "cv/thea/Bluetooth/2010/06/28/tmp-Bluetooth_RAW_11_26_18.csv_0"
data = open(input_file_name, 'r').read()
head_object = json.loads(data)
head_object["LastModified"] = datetime.datetime(2010,
6,
28,
5,
00,
52,
tzinfo=tzutc())
metadata = open(metadata_output_file_name, 'r').read()
metadata_output = json.loads(metadata)
metadata_obj = HandleBucketEvent()
metadata_response = metadata_obj.create_metadata_object(
head_object, s3_key)
assert metadata_response == metadata_output
def test_create_metadata_object_waze_raw():
input_file_name = "tests/data/head_object_input_raw.json"
metadata_output_file_name = "tests/data/metadata_output_raw.json"
s3_key = "waze/state=ME/type=jam/year=2010/month=06/day=28/hour=07/minute=54/foo5.json"
data = open(input_file_name, 'r').read()
head_object = json.loads(data)
head_object["LastModified"] = datetime.datetime(2010,
6,
28,
7,
55,
52,
tzinfo=tzutc())
metadata = open(metadata_output_file_name, 'r').read()
metadata_output = json.loads(metadata)
metadata_obj = HandleBucketEvent()
metadata_response = metadata_obj.create_metadata_object(
head_object, s3_key)
assert metadata_response == metadata_output
def test_create_metadata_object_waze_curated():
input_file_name = "tests/data/head_object_input_curated.json"
metadata_output_file_name = "tests/data/metadata_output_curated.json"
s3_key = "waze/version=20180720/content/state=IA/table=alert/projection=redshift/year=2010/month=06/day=28/hour=07/minute=55/foo4.csv.gz"
data = open(input_file_name, 'r').read()
head_object = json.loads(data)
head_object["LastModified"] = datetime.datetime(2010,
6,
28,
7,
55,
50,
tzinfo=tzutc())
metadata = open(metadata_output_file_name, 'r').read()
metadata_output = json.loads(metadata)
metadata_obj = HandleBucketEvent()
metadata_response = metadata_obj.create_metadata_object(
head_object, s3_key)
assert metadata_response == metadata_output