def test_key_mappings_are_applied_when_building_documents(): gapy_response = { "metrics": { "visits": "12345" }, "dimensions": { "date": "2013-04-02" } } data = build_document(gapy_response, "weeklyvisits", date(2013, 4, 1), {"date": "mydate"}) assert_that( data, has_entry( "_id", "d2Vla2x5dmlzaXRzXzIwMTMwMzMxMjMwMDAwX3dlZWtfMjAxMy0wNC0wMg==")) assert_that(data, has_entry("dataType", "weeklyvisits")) assert_that( data, has_entry("_timestamp", dt(2013, 4, 1, 0, 0, 0, "Europe/London"))) assert_that(data, has_entry("timeSpan", "week")) assert_that(data, has_entry("mydate", "2013-04-02")) assert_that(data, has_entry("visits", 12345))
def test_build_document_no_dimensions(): gapy_response = {"metrics": {"visits": "12345", "visitors": "5376"}} data = build_document(gapy_response, "foo", date(2013, 4, 1)) assert_that( data, has_entry("_timestamp", dt(2013, 4, 1, 0, 0, 0, "Europe/London"))) assert_that(data, has_entry("timeSpan", "week")) assert_that(data, has_entry("visits", 12345)) assert_that(data, has_entry("visitors", 5376))
def test_build_document_no_dimensions(): gapy_response = { "metrics": {"visits": "12345", "visitors": "5376"} } data = build_document(gapy_response, "foo", date(2013, 4, 1)) assert_that(data, has_entry("_timestamp", dt(2013, 4, 1, 0, 0, 0, "Europe/London"))) assert_that(data, has_entry("timeSpan", "week")) assert_that(data, has_entry("visits", 12345)) assert_that(data, has_entry("visitors", 5376))
def test_build_document(): gapy_response = { "metrics": {"visits": "12345"}, "dimensions": {"date": "2013-04-02"} } data = build_document(gapy_response, "weeklyvisits", date(2013, 4, 1)) assert_that(data, has_entry("_id", "d2Vla2x5dmlzaXRzXzIwMTMwMzMxMjMwMDAwX3dlZWtfMjAxMy0wNC0wMg==")) assert_that(data, has_entry("dataType", "weeklyvisits")) assert_that(data, has_entry("_timestamp", dt(2013, 4, 1, 0, 0, 0, "Europe/London"))) assert_that(data, has_entry("timeSpan", "week")) assert_that(data, has_entry("date", "2013-04-02")) assert_that(data, has_entry("visits", 12345))
def test_build_document_fails_with_no_data_type(): build_document({}, None, date(2012, 12, 12))