def test_get_company_by_email_with_long_suffix_mapped_2(self):
record_processor_inst = self.make_record_processor(
companies=[{
'company_name': 'NEC',
'domains': ['nec.co.jp', 'nec.com']
}])
email = '*****@*****.**'
res = get_company_by_email(record_processor_inst.domains_index, email)
self.assertEqual('NEC', res)
def test_get_company_by_email_mapped(self):
record_processor_inst = self.make_record_processor(
companies=[{
'company_name': 'IBM',
'domains': ['ibm.com']
}])
email = '*****@*****.**'
res = get_company_by_email(record_processor_inst.domains_index, email)
self.assertEqual('IBM', res)
def _update_record_and_user(self, record):
user = self._update_user(record)
record['user_id'] = user['user_id']
if user.get('user_name'):
record['author_name'] = user['user_name']
company, policy = user_processor.get_company_for_date(
user['companies'], record['date'])
if not user.get('static'):
# for auto-generated profiles affiliation may be overridden
if company != '*robots' and policy == 'open':
company = (user_processor.get_company_by_email(
self.domains_index, record.get('author_email')) or company)
record['company_name'] = company
def test_get_company_by_email_not_mapped(self):
record_processor_inst = self.make_record_processor()
email = '*****@*****.**'
res = get_company_by_email(record_processor_inst.domains_index, email)
self.assertIsNone(res)