def test_extract_model_data_should_process_data_and_return_it(self):
model_ids = {"name": ["message", "subject"]}
processor = Mock()
processor.process.return_value = "subject is something"
data_extractor = DataExtractor(model_id_mapping=model_ids, using_processors={"name": processor})
model_ids, model_data = data_extractor.extract_model_data_from(discussion_data)
self.assertEquals({"name": "subject is something"}, model_ids)
def test_extract_model_data_from_should_get_data_extracted_from_a_key_defined_by_list(
self):
model_ids = {'name': ['message', 'subject']}
data_extractor = DataExtractor(model_id_mapping=model_ids)
model_ids, model_data = data_extractor.extract_model_data_from(
discussion_data)
self.assertEquals({'name': 'subject is nothing'}, model_ids)
def test_extract_model_data_from_should_get_data_extracted_from_what_is_passed_in(self):
model_data = {"name": "name", "mytw_user_id": "id", "email": "email"}
model_ids = {"username": "******"}
data_extractor = DataExtractor(model_id_mapping=model_ids, model_data_mapping=model_data)
model_ids, model_data = data_extractor.extract_model_data_from(user_test_data[0])
self.assertEquals({"username": "******"}, model_ids)
self.assertEquals({"email": "*****@*****.**", "name": "John Doe", "mytw_user_id": 1278237}, model_data)
def test_extract_model_data_from_should_merge_default_into_model_id_if_specefied(self):
model_data = {"name": "name", "mytw_user_id": "id", "email": "email"}
model_ids = {"username": "******"}
data_extractor = DataExtractor(
model_id_mapping=model_ids, model_data_mapping=model_data, default={"type": "user"}
)
model_ids, model_data = data_extractor.extract_model_data_from(user_test_data[0])
self.assertEquals({"username": "******", "type": "user"}, model_ids)
def test_extract_model_data_should_process_data_and_return_it(self):
model_ids = {'name': ['message', 'subject']}
processor = Mock()
processor.process.return_value = "subject is something"
data_extractor = DataExtractor(model_id_mapping=model_ids,
using_processors={'name': processor})
model_ids, model_data = data_extractor.extract_model_data_from(
discussion_data)
self.assertEquals({'name': 'subject is something'}, model_ids)
def test_extract_model_data_from_should_merge_default_into_model_id_if_specefied(
self):
model_data = {"name": "name", "mytw_user_id": "id", "email": "email"}
model_ids = {"username": "******"}
data_extractor = DataExtractor(model_id_mapping=model_ids,
model_data_mapping=model_data,
default={'type': 'user'})
model_ids, model_data = data_extractor.extract_model_data_from(
user_test_data[0])
self.assertEquals({'username': '******', 'type': 'user'}, model_ids)
def test_extract_model_data_from_should_get_data_extracted_from_what_is_passed_in(
self):
model_data = {"name": "name", "mytw_user_id": "id", "email": "email"}
model_ids = {"username": "******"}
data_extractor = DataExtractor(model_id_mapping=model_ids,
model_data_mapping=model_data)
model_ids, model_data = data_extractor.extract_model_data_from(
user_test_data[0])
self.assertEquals({'username': '******'}, model_ids)
self.assertEquals(
{
'email': '*****@*****.**',
'name': 'John Doe',
'mytw_user_id': 1278237
}, model_data)
def test_extract_model_data_from_should_get_data_extracted_from_a_key_defined_by_list(self):
model_ids = {"name": ["message", "subject"]}
data_extractor = DataExtractor(model_id_mapping=model_ids)
model_ids, model_data = data_extractor.extract_model_data_from(discussion_data)
self.assertEquals({"name": "subject is nothing"}, model_ids)
