class TestPartnersDataSource(MyJobsBase):
def setUp(self):
super(TestPartnersDataSource, self).setUp()
# A company to work with
self.company = CompanyFactory(name='right')
self.company.save()
# A separate company that should not show up in results.
self.other_company = CompanyFactory(name='wrong')
self.other_company.save()
self.partner_a = PartnerFactory(
owner=self.company,
last_action_time='2015-09-30 13:23',
uri='http://www.example.com/',
data_source="zap",
name="aaa")
self.partner_b = PartnerFactory(
owner=self.company,
last_action_time='2015-10-03',
uri='http://www.asdf.com/',
data_source="bcd",
name="bbb")
# An unapproved parther. Associated data should be filtered out.
self.partner_unapp = PartnerFactory(
owner=self.company,
name="unapproved",
approval_status__code=Status.UNPROCESSED)
# An archived parther. Associated data should be filtered out.
self.partner_archived = PartnerFactory(owner=self.company)
self.east_tag = TagFactory.create(name='east', hex_color="aaaaaa")
self.west_tag = TagFactory.create(name='west', hex_color="bbbbbb")
self.bad_tag = TagFactory.create(name='bad', hex_color="cccccc")
self.partner_a.tags.add(self.east_tag)
self.partner_b.tags.add(self.west_tag)
self.john_user = UserFactory(email="*****@*****.**")
self.john = ContactFactory(
partner=self.partner_a,
name='john adams',
user=self.john_user,
email="*****@*****.**",
last_action_time='2015-10-03')
self.john.locations.add(
LocationFactory.create(
city="Indianapolis",
state="IN"))
self.john.locations.add(
LocationFactory.create(
city="Chicago",
state="IL"))
self.sue_user = UserFactory(email="*****@*****.**")
self.sue = ContactFactory(
partner=self.partner_b,
name='Sue Baxter',
user=self.sue_user,
email="*****@*****.**",
last_action_time='2015-09-30 13:23')
self.sue.locations.add(
LocationFactory.create(
address_line_one="123",
city="Los Angeles",
state="CA"))
self.sue.locations.add(
LocationFactory.create(
address_line_one="234",
city="Los Angeles",
state="CA"))
self.partner_a.primary_contact = self.john
self.partner_b.primary_contact = self.sue
self.partner_a.save()
self.partner_b.save()
# Archive archived data here. Doing this earlier in the set up results
# in odd exceptions.
self.partner_archived.archive()
def test_run_unfiltered(self):
"""Make sure we only get data for this user."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(self.company, PartnersFilter(), [])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name, self.partner_b.name}
self.assertEqual(expected, names)
expected_tags = {
self.partner_a.pk: {'east'},
self.partner_b.pk: {'west'},
}
def tags_by_pk(rec):
return (rec['partner_id'], set(t['name'] for t in rec['tags']))
found_tags = dict(tags_by_pk(r) for r in recs)
self.assertEqual(expected_tags, found_tags)
def test_run_count_comm_rec_per_month(self):
"""Make sure we only get data for this user."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(
subject=subject,
date_time=datetime(2015, month, 1)))
ds = PartnersDataSource()
recs = ds.run_count_comm_rec_per_month(
self.company, PartnersFilter(), ['name', 'year', '-month'])
data = [
(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs
]
expected = [
(self.partner_a.name, 2015, 12, 0),
(self.partner_a.name, 2015, 11, 0),
(self.partner_a.name, 2015, 10, 0),
(self.partner_a.name, 2015, 9, 0),
(self.partner_a.name, 2015, 8, 0),
(self.partner_a.name, 2015, 7, 0),
(self.partner_a.name, 2015, 6, 0),
(self.partner_a.name, 2015, 5, 0),
(self.partner_a.name, 2015, 4, 0),
(self.partner_a.name, 2015, 3, 0),
(self.partner_a.name, 2015, 2, 0),
(self.partner_a.name, 2015, 1, 0),
(self.partner_b.name, 2015, 12, 0),
(self.partner_b.name, 2015, 11, 0),
(self.partner_b.name, 2015, 10, 0),
(self.partner_b.name, 2015, 9, 0),
(self.partner_b.name, 2015, 8, 0),
(self.partner_b.name, 2015, 7, 0),
(self.partner_b.name, 2015, 6, 0),
(self.partner_b.name, 2015, 5, 0),
(self.partner_b.name, 2015, 4, 3),
(self.partner_b.name, 2015, 3, 3),
(self.partner_b.name, 2015, 2, 3),
(self.partner_b.name, 2015, 1, 0),
]
self.assertEqual(expected, data)
def test_run_count_comm_rec_per_month_one_partner(self):
"""Trip over a bug in the mysql client driver."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for year in [2013, 2015, 2014]:
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(
subject=subject,
date_time=datetime(year, month, 1)))
ds = PartnersDataSource()
partners_filter = PartnersFilter(tags=[['west']])
recs = ds.run_count_comm_rec_per_month(
self.company, partners_filter, ['name', 'year', '-month'])
data = [
(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs
]
self.assertEqual(36, len(data))
def test_run_count_comm_rec_per_month_no_partners(self):
"""Trip over a bug in the mysql client driver."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(
subject=subject,
date_time=datetime(2015, month, 1)))
ds = PartnersDataSource()
partners_filter = PartnersFilter(tags=[['zzz']])
recs = ds.run_count_comm_rec_per_month(
self.company, partners_filter, ['name', 'year', '-month'])
self.assertEqual(0, len(recs))
def test_run_count_comm_rec_per_month_empty_partner(self):
"""One partner, no communication records."""
ds = PartnersDataSource()
partners_filter = PartnersFilter(tags=[['east']])
recs = ds.run_count_comm_rec_per_month(
self.company, partners_filter, ['name', 'year', '-month'])
data = [
(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs
]
self.assertEqual(12, len(data))
default_year = datetime.now().year
for item in data:
self.assertEqual(default_year, item[1])
self.assertEqual(0, item[3])
def test_filter_by_date_range(self):
"""Should show only partner with last_action_time in range."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
date=[datetime(2015, 9, 1), datetime(2015, 9, 30)]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_date_before(self):
"""Should show only partner with last_action_time before date."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
date=[None, datetime(2015, 9, 30)]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_date_after(self):
"""Should show only partner with last_action_time after date."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
date=[datetime(2015, 10, 1), None]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_b.name}
self.assertEqual(expected, names)
def test_filter_by_state(self):
"""Should show only partners with correct state."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
locations={
'state': 'CA'
}),
[])
names = [r['name'] for r in recs]
expected = [self.partner_b.name]
self.assertEqual(expected, names)
def test_filter_by_city(self):
"""Should show only partners with correct city."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
locations={
'city': 'Los Angeles'
}),
[])
names = [r['name'] for r in recs]
expected = [self.partner_b.name]
self.assertEqual(expected, names)
def test_filter_by_tags(self):
"""Show only partner with correct tags."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=[['EaSt']]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_tags_or(self):
"""Show only partner with correct tags in 'or' configuration."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=[['EaSt', 'wEsT']]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name, self.partner_b.name}
self.assertEqual(expected, names)
def test_filter_by_tags_and(self):
"""Show only partner with correct tags in 'and' configuration."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=[['EaSt'], ['wEsT']]),
[])
names = set([r['name'] for r in recs])
expected = set()
self.assertEqual(expected, names)
# Now try adding another tag.
self.partner_a.tags.add(self.west_tag)
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=[['EaSt'], ['wEsT']]),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_data_source(self):
"""Check datasource filter works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(data_source="zap"),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_uri(self):
"""Check uri filter works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(uri="http://www.asdf.com/"),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_b.name}
self.assertEqual(expected, names)
def test_filter_by_empty_things(self):
"""Empty filters should not filter, just like missing filters."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
locations={'city': '', 'state': ''}),
[])
names = set([r['name'] for r in recs])
expected = {self.partner_a.name, self.partner_b.name}
self.assertEqual(expected, names)
def test_help_city(self):
"""Check city help works and ignores current city filter."""
ds = PartnersDataSource()
recs = ds.help_city(
self.company,
PartnersFilter(locations={'city': "zz"}),
"angel")
actual = set([r['key'] for r in recs])
self.assertEqual({'Los Angeles'}, actual)
def test_help_state(self):
"""Check state help works and ignores current state filter."""
ds = PartnersDataSource()
recs = ds.help_state(
self.company,
PartnersFilter(locations={'state': "zz"}),
"i")
actual = set([r['key'] for r in recs])
self.assertEqual({'IL', 'IN'}, actual)
def test_help_tags(self):
"""Check tags help works at all."""
ds = PartnersDataSource()
recs = ds.help_tags(self.company, PartnersFilter(), "E")
actual = set([r['key'] for r in recs])
self.assertEqual({'east', 'west'}, actual)
def test_help_tags_colors(self):
"""Tags should have colors"""
ds = PartnersDataSource()
recs = ds.help_tags(self.company, PartnersFilter(), "east")
self.assertEqual("aaaaaa", recs[0]['hexColor'])
def test_help_uri(self):
"""Check uri help works at all."""
ds = PartnersDataSource()
recs = ds.help_uri(
self.company,
PartnersFilter(),
"ex")
actual = set([r['key'] for r in recs])
self.assertEqual({'http://www.example.com/'}, actual)
def test_help_data_source(self):
"""Check data_source help works at all."""
ds = PartnersDataSource()
recs = ds.help_data_source(
self.company,
PartnersFilter(),
"z")
actual = set([r['key'] for r in recs])
self.assertEqual({'zap'}, actual)
def test_order(self):
"""Check ordering results works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(),
["-name"])
names = [r['name'] for r in recs]
expected = [self.partner_b.name, self.partner_a.name]
self.assertEqual(expected, names)
class TestCommRecordsDataSource(MyJobsBase):
def setUp(self):
super(TestCommRecordsDataSource, self).setUp()
# A company to work with
self.company = CompanyFactory(name='right')
self.company.save()
# A separate company that should not show up in results.
self.other_company = CompanyFactory(name='wrong')
self.other_company.save()
self.partner_a = PartnerFactory(
owner=self.company,
uri='http://www.example.com/',
data_source="zap",
name="aaa")
self.partner_b = PartnerFactory(
owner=self.company,
uri='http://www.asdf.com/',
data_source="bcd",
name="bbb")
# An unapproved parther. Associated data should be filtered out.
self.partner_unapp = PartnerFactory(
owner=self.company,
name="unapproved",
approval_status__code=Status.UNPROCESSED)
# An archived parther. Associated data should be filtered out.
self.partner_archived = PartnerFactory(owner=self.company)
self.east_tag = TagFactory.create(name='east', hex_color="aaaaaa")
self.west_tag = TagFactory.create(name='west', hex_color="bbbbbb")
self.bad_tag = TagFactory.create(name='bad', hex_color="cccccc")
self.john_user = UserFactory(email="*****@*****.**")
self.john = ContactFactory(
partner=self.partner_a,
name='john adams',
user=self.john_user,
email="*****@*****.**",
last_action_time='2015-10-03')
self.john.locations.add(
LocationFactory.create(
city="Indianapolis",
state="IN"))
self.john.locations.add(
LocationFactory.create(
city="Chicago",
state="IL"))
self.sue_user = UserFactory(email="*****@*****.**")
self.sue = ContactFactory(
partner=self.partner_b,
name='Sue Baxter',
user=self.sue_user,
email="*****@*****.**",
last_action_time='2015-09-30 13:23')
self.sue.locations.add(
LocationFactory.create(
address_line_one="123",
city="Los Angeles",
state="CA"))
self.sue.locations.add(
LocationFactory.create(
address_line_one="234",
city="Los Angeles",
state="CA"))
self.partner_a.primary_contact = self.john
self.partner_b.primary_contact = self.sue
self.partner_a.save()
self.partner_b.save()
self.record_1 = ContactRecordFactory(
subject='record 1',
date_time='2015-09-30 13:23',
contact=self.john,
contact_type="Email",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_2 = ContactRecordFactory(
subject='record 2',
date_time='2015-01-01',
contact=self.john,
contact_type="Meeting Or Event",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_3 = ContactRecordFactory(
subject='record 3',
date_time='2015-10-03',
contact=self.sue,
contact_type="Phone",
partner=self.partner_b,
location='Los Angeles, CA',
tags=[self.west_tag])
# Archive archived data here. Doing this earlier in the set up results
# in odd exceptions.
self.partner_archived.archive()
def test_run_unfiltered(self):
"""Make sure we only get data for this user."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(self.company, CommRecordsFilter(), [])
subjects = set([r['subject'] for r in recs])
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_date_range(self):
"""Should show only commrec with last_action_time in range."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
date_time=[datetime(2015, 9, 1), datetime(2015, 9, 30)]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_before(self):
"""Should show only commrec with last_action_time before date."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
date_time=[None, datetime(2015, 9, 30)]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_after(self):
"""Should show only commrec with last_action_time after date."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
date_time=[datetime(2015, 10, 1), None]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_state(self):
"""Should show only commrecs with correct state."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations={
'state': 'CA'
}),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_city(self):
"""Should show only commrecs with correct city."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations={
'city': 'Los Angeles'
}),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags(self):
"""Show only commrec with correct tags."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=[['EaSt']]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags_or(self):
"""Show only commrec with correct tags in 'or' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=[['EaSt', 'wEsT']]),
[])
subjects = set([r['subject'] for r in recs])
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_tags_and(self):
"""Show only commrec with correct tags in 'and' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=[['EaSt'], ['wEsT']]),
[])
subjects = set([r['subject'] for r in recs])
expected = set()
self.assertEqual(expected, subjects)
# Now try adding another tag.
self.record_1.tags.add(self.west_tag)
self.record_1.save()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=[['EaSt'], ['wEsT']]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_empty_things(self):
"""Empty filters should not filter, just like missing filters."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations={'city': '', 'state': ''}),
[])
subjects = set([r['subject'] for r in recs])
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_communication_type(self):
"""Check communication_type filter works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(communication_type='Email'),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner(self):
"""Check partner filter works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(partner=[self.partner_a.pk]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact(self):
"""Check partner filter works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(contact=[self.sue.pk]),
[])
subjects = set([r['subject'] for r in recs])
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_help_city(self):
"""Check city help works and ignores current city filter."""
ds = CommRecordsDataSource()
recs = ds.help_city(
self.company,
CommRecordsFilter(locations={'city': "zz"}),
"angel")
actual = set([r['key'] for r in recs])
self.assertEqual({'Los Angeles'}, actual)
def test_help_state(self):
"""Check state help works and ignores current state filter."""
ds = CommRecordsDataSource()
recs = ds.help_state(
self.company,
CommRecordsFilter(locations={'state': "zz"}),
"i")
actual = set([r['key'] for r in recs])
self.assertEqual({'IL', 'IN'}, actual)
def test_help_tags(self):
"""Check tags help works at all."""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "E")
actual = set([r['key'] for r in recs])
self.assertEqual({'east', 'west'}, actual)
def test_help_tags_colors(self):
"""Tags should have colors"""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "east")
self.assertEqual("aaaaaa", recs[0]['hexColor'])
def test_help_communication_types(self):
"""Check communication_types help works at all."""
ds = CommRecordsDataSource()
recs = ds.help_communication_type(
self.company, CommRecordsFilter(), "ph")
actual = set([r['key'] for r in recs])
self.assertEqual({'Phone'}, actual)
def test_help_partner(self):
"""Check partner help works at all."""
ds = CommRecordsDataSource()
recs = ds.help_partner(self.company, CommRecordsFilter(), "A")
self.assertEqual(
[{'key': self.partner_a.pk, 'display': self.partner_a.name}],
recs)
def test_help_contact(self):
"""Check contact help works at all."""
ds = CommRecordsDataSource()
recs = ds.help_contact(self.company, CommRecordsFilter(), "U")
self.assertEqual(
[{'key': self.sue.pk, 'display': self.sue.name}],
recs)
def test_order(self):
"""Check ordering results works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(),
["-subject"])
subjects = [r['subject'] for r in recs]
expected = [
self.record_3.subject,
self.record_2.subject,
self.record_1.subject,
]
self.assertEqual(expected, subjects)
class MyPartnerTests(MyJobsBase):
def setUp(self):
super(MyPartnerTests, self).setUp()
self.company = CompanyFactory()
self.partner = PartnerFactory(owner=self.company)
self.contact = ContactFactory(partner=self.partner)
def test_contact_to_partner_relationship(self):
"""
Tests adding a contact to partner's contacts list and tests
primary_contact. Also tests if the contact gets deleted the partner
stays and turns primary_contact to None.
"""
self.assertEqual(
Contact.objects.filter(partner=self.partner).count(), 1)
self.partner.primary_contact = self.contact
self.partner.save()
self.assertIsNotNone(self.partner.primary_contact)
# making sure contact is the contact obj vs a factory object.
contact = Contact.objects.get(name=self.contact.name)
contact.delete()
partner = Partner.objects.get(name=self.partner.name)
self.assertFalse(
Contact.objects.filter(partner=partner, archived_on__isnull=True))
self.assertIsNone(partner.primary_contact)
def test_contact_user_relationship(self):
"""
Tests adding a User to Contact. Then tests to make sure User cascading
delete doesn't delete the Contact and instead turns Contact.user to
None.
"""
self.contact.user = UserFactory(email=self.contact.email)
self.contact.save()
self.assertIsNotNone(self.contact.user)
self.assertEqual(self.contact.name, self.contact.__unicode__())
user = User.objects.get(email=self.contact.email)
user.delete()
contact = Contact.objects.get(name=self.contact.name)
self.assertIsNone(contact.user)
def test_location_to_contact_relationship(self):
"""
Tests adding a Location to Contact.
"""
location = LocationFactory()
# make sure that we can add a location to a contact
self.contact.locations.add(location)
self.contact.save()
self.assertTrue(len(self.contact.locations.all()) > 0)
# ensure that we can remove a location
self.contact.locations.remove(location)
self.assertTrue(len(self.contact.locations.all()) == 0)
# make sure that removing a location from a contact doesn't delete that
# location entirely
self.assertIn(location, Location.objects.all())
def test_bad_filename(self):
"""
Confirms that non-alphanumeric or underscore characters are being
stripped from file names.
"""
actual_file = path.join(path.abspath(path.dirname(__file__)), 'data',
'test.txt')
f = File(open(actual_file))
filenames = [('zz\\x80\\xff*file(copy)na.me.htm)_-)l',
'zzx80xfffilecopyname.htm_l'), ('...', 'unnamed_file'),
('..', 'unnamed_file'), ('../../file.txt', 'file.txt'),
('../..', 'unnamed_file'), ('\.\./file.txt', 'file.txt'),
('fiяыle.txt', 'file.txt')]
for filename, expected_filename in filenames:
f.name = filename
prm_attachment = PRMAttachmentFactory(attachment=f)
result = PRMAttachment.objects.get(
attachment__contains=expected_filename)
result.delete()
def test_partner_saved_search_delete_contact(self):
"""
When a contact gets deleted, we should log it and disable any partner
saved searches for that contact
"""
user = UserFactory(email='*****@*****.**')
self.contact.user = user
self.contact.save()
self.contact = Contact.objects.get(pk=self.contact.pk)
owner = UserFactory(email='*****@*****.**')
partner_saved_search = PartnerSavedSearchFactory(created_by=owner,
provider=self.company,
partner=self.partner,
user=user,
notes='')
self.assertTrue(partner_saved_search.is_active)
self.contact.delete()
partner_saved_search = PartnerSavedSearch.objects.get(
pk=partner_saved_search.pk)
self.assertFalse(partner_saved_search.is_active)
self.assertTrue(self.contact.name in partner_saved_search.notes)
def test_tag_added_to_taggable_models(self):
tag = TagFactory(company=self.company)
tag.save()
tag2 = TagFactory(name="bar", company=self.company)
tag2.save()
cr = ContactRecordFactory(partner=self.partner)
# Add tag to models
cr.tags.add(tag)
self.partner.tags.add(tag)
self.partner.save()
self.contact.tags.add(tag)
self.contact.save()
# Check to make sure it was added
self.assertEquals(1, len(cr.tags.all()))
self.assertEquals(1, len(self.partner.tags.all()))
self.assertEquals(1, len(self.contact.tags.all()))
# Add a 2nd tag and check
self.partner.tags.add(tag2)
self.partner.save()
self.assertEquals(2, len(self.partner.tags.all()))
def test_contact_archived(self):
"""Test that attempting to delete a contact archives it instead."""
self.assertFalse(self.contact.archived_on)
self.contact.archive()
self.assertEqual(Contact.objects.count(), 0)
self.assertEqual(Contact.all_objects.count(), 1)
self.assertTrue(self.contact.archived_on)
def test_archived_manager_weirdness(self):
"""
Demonstrates that archived instances are returned by the
ArchivedModel.all_objects manager, not the standard
ArchivedModel.objects.
"""
self.partner.archive()
# Try retrieving the archived partner using both managers. The "objects"
# manager excludes archived instances so an exception is raised.
self.assertRaises(Partner.DoesNotExist,
lambda: Partner.objects.get(pk=self.partner.pk))
Partner.all_objects.get(pk=self.partner.pk)
# This contact has not been archived (as we demonstrate in a few lines)
# but is excluded by the "objects" manager as its partner has been
# archived.
self.assertRaises(Contact.DoesNotExist,
lambda: Contact.objects.get(pk=self.contact.pk))
self.contact = Contact.all_objects.get(pk=self.contact.pk)
self.assertFalse(self.contact.archived_on)
# The manager used by related objects in this instance excludes
# archived partners. As it's basically a Partner.objects.get(id=...),
# this fails.
self.assertRaises(Partner.DoesNotExist, lambda: self.contact.partner)
self.assertEqual(self.contact.partner_id, self.partner.pk)
def test_archive_primary_contacts(self):
"""
Archiving a primary contact should clear that contact's status as the
partner's primary contact. Doing otherwise raises exceptions.
"""
self.partner.primary_contact = self.contact
self.partner.save()
self.partner.primary_contact.archive()
self.partner = Partner.objects.get(pk=self.partner.pk)
self.partner.primary_contact
def test_models_approved(self):
"""
By default, new partners, contacts, and contactrecords should be
approved.
"""
contactrecord = ContactRecordFactory(partner=self.partner)
for instance in (self.contact, self.partner, contactrecord):
self.assertEqual(instance.approval_status.code, Status.APPROVED)
def test_contact_locations(self):
"""
Test that `get_contact_locations` returns a properly formatted string.
"""
ny = LocationFactory.create_batch(2, city="Albany", state="NY")
il = LocationFactory.create(city="Chicago", state="IL")
mo = LocationFactory.create(city="St. Louis", state="MO")
contacts = ContactFactory.create_batch(4, partner=self.partner)
for contact, location in zip(contacts, ny + [il, mo]):
contact.locations.add(location)
self.assertEqual("Chicago, IL; St. Louis, MO; Albany, NY",
"; ".join(self.partner.get_contact_locations()))
def test_uncommon_outreach_email_domain(self):
"""
Adding an uncommon email domain for outreach to a company should work.
"""
# data migrations aren't run during tests, so we populate manually
CommonEmailDomain.objects.create(domain="gmail.com")
with self.assertRaises(ValidationError):
OutreachEmailDomain.objects.create(company=self.company,
domain="gmail.com")
def test_outreach_domain_unique_to_company(self):
"""
Allowed domains should be unique within a company, but not necessarily
across PRM.
"""
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
# duplicate domains allowed between companies
company = CompanyFactory.create(name="A Whole New World")
OutreachEmailDomain.objects.create(company=company, domain="foo.bar")
# dupliate domains disallowed within the same company
with self.assertRaises(IntegrityError):
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
def test_contact_record_counts_vs_list(self):
"""
ContactRecord counts for Communication Records and Referals
should match summed counts from contacts.
"""
contacts = ContactFactory.create_batch(4)
contacts[0].name = 'Other name'
contacts[1].email = '*****@*****.**'
contacts[2].partner = PartnerFactory(name='Other Partner')
for contact in contacts:
ContactRecordFactory.create(contact_type="job",
contact=contact,
partner=contact.partner)
ContactRecordFactory.create(contact_type='email',
contact=contact,
partner=contact.partner)
contacts[0].email = '*****@*****.**'
ContactRecordFactory.create(contact_type='email',
contact=contacts[0],
partner=contact.partner)
queryset = ContactRecord.objects.all()
self.assertEqual(queryset.count(), 9)
contacts = list(queryset.contacts)
sum_referrals = sum([contact['referrals'] for contact in contacts])
sum_records = sum([contact['records'] for contact in contacts])
self.assertEqual(sum_referrals, queryset.referrals)
self.assertEqual(sum_records, queryset.communication_activity.count())
def test_contact_record_report_numbers(self):
"""
Contact records have properties which represent various aggregated
values. This test ensures that given a number of contact records, those
aggregated numbers are correct.
"""
email_record = ContactRecordFactory(contact_type="email",
partner=self.partner,
contact=self.contact)
job_record = ContactRecordFactory(contact_type="job",
partner=self.partner,
contact=self.contact,
job_applications=10,
job_interviews=6,
job_hires=5)
phone_record = ContactRecordFactory(contact_type="phone",
partner=self.partner,
contact=ContactFactory(name="Joe"))
records = ContactRecord.objects.all()
self.assertEqual(len(records), 3)
self.assertEqual(len(records.contacts), 2)
# job follow ups don't count as comm activity
self.assertEqual(len(records.communication_activity), 2)
# only job follow ups count as referrals
self.assertEqual(records.referrals, 1)
self.assertEqual(records.applications, 10)
self.assertEqual(records.interviews, 6)
self.assertEqual(records.hires, 5)
self.assertEqual(records.emails, 1)
self.assertEqual(records.calls, 1)
class TestCommRecordsDataSource(MyJobsBase):
def setUp(self):
super(TestCommRecordsDataSource, self).setUp()
# A company to work with
self.company = CompanyFactory(name='right')
self.company.save()
# A separate company that should not show up in results.
self.other_company = CompanyFactory(name='wrong')
self.other_company.save()
self.partner_a = PartnerFactory(owner=self.company,
uri='http://www.example.com/',
data_source="zap",
name="aaa")
self.partner_b = PartnerFactory(owner=self.company,
uri='http://www.asdf.com/',
data_source="bcd",
name="bbb")
# An unapproved parther. Associated data should be filtered out.
self.partner_unapp = PartnerFactory(
owner=self.company,
name="unapproved",
approval_status__code=Status.UNPROCESSED)
# An archived parther. Associated data should be filtered out.
self.partner_archived = PartnerFactory(owner=self.company)
self.east_tag = TagFactory.create(company=self.company,
name='east',
hex_color="aaaaaa")
self.west_tag = TagFactory.create(company=self.company,
name='west',
hex_color="bbbbbb")
self.north_tag = TagFactory.create(company=self.company,
name='north',
hex_color="cccccc")
self.south_tag = TagFactory.create(company=self.company,
name='south',
hex_color="dddddd")
self.left_tag = TagFactory.create(company=self.company,
name='left',
hex_color="eeeeee")
self.right_tag = TagFactory.create(company=self.company,
name='right',
hex_color="ffffff")
self.bad_tag = TagFactory.create(company=self.company,
name='bad',
hex_color="cccccc")
self.partner_a.tags.add(self.left_tag)
self.partner_b.tags.add(self.right_tag)
self.john_user = UserFactory(email="*****@*****.**")
self.john = ContactFactory(partner=self.partner_a,
name='john adams',
user=self.john_user,
email="*****@*****.**",
last_action_time='2015-10-03')
self.john.locations.add(
LocationFactory.create(city="Indianapolis", state="IN"))
self.john.locations.add(
LocationFactory.create(city="Chicago", state="IL"))
self.john.tags.add(self.north_tag)
self.sue_user = UserFactory(email="*****@*****.**")
self.sue = ContactFactory(partner=self.partner_b,
name='Sue Baxter',
user=self.sue_user,
email="*****@*****.**",
last_action_time='2015-09-30 13:23')
self.sue.locations.add(
LocationFactory.create(address_line_one="123",
city="Los Angeles",
state="CA"))
self.sue.locations.add(
LocationFactory.create(address_line_one="234",
city="Los Angeles",
state="CA"))
self.sue.tags.add(self.south_tag)
self.partner_a.primary_contact = self.john
self.partner_b.primary_contact = self.sue
self.partner_a.save()
self.partner_b.save()
self.record_1 = ContactRecordFactory(subject='record 1',
date_time='2015-09-30 13:23',
contact=self.john,
contact_type="Email",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_2 = ContactRecordFactory(subject='record 2',
date_time='2015-01-01',
contact=self.john,
contact_type="Meeting Or Event",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_3 = ContactRecordFactory(subject='record 3',
date_time='2015-10-03',
contact=self.sue,
contact_type="Phone",
partner=self.partner_b,
location='Los Angeles, CA',
tags=[self.west_tag])
# Archive archived data here. Doing this earlier in the set up results
# in odd exceptions.
self.partner_archived.archive()
def test_run_unfiltered(self):
"""Make sure we only get data for this user."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(self.company, CommRecordsFilter(), [])
subjects = {r['subject'] for r in recs}
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_date_range(self):
"""Should show only commrec with last_action_time in range."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter(
[datetime(2015, 9, 1), datetime(2015, 9, 30)])
recs = ds.run_unaggregated(self.company,
CommRecordsFilter(date_time=date_range), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_before(self):
"""Should show only commrec with last_action_time before date."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter([None, datetime(2015, 9, 30)])
recs = ds.run_unaggregated(self.company,
CommRecordsFilter(date_time=date_range), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_after(self):
"""Should show only commrec with last_action_time after date."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter([datetime(2015, 10, 1), None])
recs = ds.run_unaggregated(self.company,
CommRecordsFilter(date_time=date_range), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_state(self):
"""Should show only commrecs with correct state."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({'state': MatchFilter('CA')})),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_city(self):
"""Should show only commrecs with correct city."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(locations=CompositeAndFilter(
{'city': MatchFilter('Los Angeles')})), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags(self):
"""
Show only commrec with correct tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
tags=AndGroupFilter([OrGroupFilter([MatchFilter('EaSt')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags_unlinked(self):
"""
Only return untagged commrecs.
"""
self.record_1.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(self.company,
CommRecordsFilter(tags=UnlinkedFilter()),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags_or(self):
"""Show only commrec with correct tags in 'or' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter(
[OrGroupFilter([MatchFilter('EaSt'),
MatchFilter('wEsT')])])), [])
subjects = {r['subject'] for r in recs}
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_tags_and(self):
"""Show only commrec with correct tags in 'and' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])
])), [])
subjects = {r['subject'] for r in recs}
expected = set()
self.assertEqual(expected, subjects)
# Now try adding another tag.
self.record_1.tags.add(self.west_tag)
self.record_1.save()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])
])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_communication_type(self):
"""Check communication_type filter works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
communication_type=OrGroupFilter([MatchFilter('email')])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner(self):
"""
Check partner filter works at all.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
partner=OrGroupFilter([MatchFilter(self.partner_a.pk)])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact(self):
"""
Check partner filter works at all.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
contact=OrGroupFilter([MatchFilter(self.sue.pk)])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact_tags(self):
"""
Test that we can filter by contact tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(contact_tags=AndGroupFilter(
[OrGroupFilter([MatchFilter('sOuTh')])])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact_tags_untagged(self):
"""
Check that we can find a record attached to an untagged contact.
"""
self.sue.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company, CommRecordsFilter(contact_tags=UnlinkedFilter()), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner_tags(self):
"""
Test that we can filter by partner tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(partner_tags=AndGroupFilter(
[OrGroupFilter([MatchFilter('rigHt')])])), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner_tags_untagged(self):
"""
Check that we can find a record attached to an untagged partner.
"""
self.partner_b.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company, CommRecordsFilter(partner_tags=UnlinkedFilter()), [])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_help_city(self):
"""
Check city help works and ignores current city filter.
"""
ds = CommRecordsDataSource()
recs = ds.help_city(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({'city': MatchFilter("zz")})),
"angel")
actual = {r['value'] for r in recs}
self.assertEqual({'Los Angeles'}, actual)
def test_help_state(self):
"""
Check state help works and ignores current state filter.
"""
ds = CommRecordsDataSource()
recs = ds.help_state(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({'state': MatchFilter("zz")})),
"i")
actual = {r['value'] for r in recs}
self.assertEqual({'IL', 'IN'}, actual)
def test_help_tags(self):
"""
Check tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "E")
actual = {r['value'] for r in recs}
self.assertEqual({'east', 'west'}, actual)
def test_help_contact_tags(self):
"""
Check contact tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_contact_tags(self.company, CommRecordsFilter(), "O")
actual = {r['value'] for r in recs}
self.assertEqual({'north', 'south'}, actual)
def test_help_partner_tags(self):
"""
Check partner tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_partner_tags(self.company, CommRecordsFilter(), "t")
actual = {r['value'] for r in recs}
self.assertEqual({'left', 'right'}, actual)
def test_help_tags_colors(self):
"""
Tags should have colors
"""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "east")
self.assertEqual("aaaaaa", recs[0]['hexColor'])
def test_help_communication_types(self):
"""
Check communication_types help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_communication_type(self.company, CommRecordsFilter(),
"ph")
actual = {r['value'] for r in recs}
expected = {'phone', 'job', 'meetingorevent', 'email', 'pssemail'}
self.assertEqual(expected, actual)
def test_help_partner(self):
"""
Check partner help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_partner(self.company, CommRecordsFilter(), "A")
self.assertEqual([{
'value': self.partner_a.pk,
'display': self.partner_a.name
}], recs)
def test_help_contact(self):
"""
Check contact help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_contact(self.company, CommRecordsFilter(), "U")
self.assertEqual([{
'value': self.sue.pk,
'display': self.sue.name
}], recs)
def test_values(self):
"""
Check limiting values works at all
"""
self.maxDiff = 10000
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(self.company, CommRecordsFilter(),
["partner", "subject"])
expected = [
{
'partner': u'aaa',
'subject': u'record 1'
},
{
'partner': u'aaa',
'subject': u'record 2'
},
{
'partner': u'bbb',
'subject': u'record 3'
},
]
self.assertEqual(expected, recs)
def test_adorn_filter(self):
self.maxDiff = 10000
found_filter_items = {
'tags': ['east', 'west'],
'communication_type': ['Email'],
'partner': [str(self.partner_a.pk)],
'contact': [str(self.sue.pk)],
'contact_tags': ['nOrth', 'south'],
'partner_tags': ['lEft', 'riGht'],
}
expected = {
u'partner': {
self.partner_a.pk: {
'value': self.partner_a.pk,
'display': u'aaa'
},
},
u'contact': {
self.sue.pk: {
'value': self.sue.pk,
'display': u'Sue Baxter'
},
},
u'tags': {
u'east': {
'value': u'east',
'display': u'east',
'hexColor': u'aaaaaa',
},
u'west': {
'value': u'west',
'display': u'west',
'hexColor': u'bbbbbb',
},
},
u'communication_type': {
'email': {
'value': 'email',
'display': 'Email'
}
},
u'contact_tags': {
u'north': {
'value': u'north',
'display': u'north',
'hexColor': u'cccccc',
},
u'south': {
'value': u'south',
'display': u'south',
'hexColor': u'dddddd',
},
},
u'partner_tags': {
u'left': {
'value': u'left',
'display': u'left',
'hexColor': u'eeeeee',
},
u'right': {
'value': u'right',
'display': u'right',
'hexColor': u'ffffff',
},
},
}
ds = CommRecordsDataSource()
result = ds.adorn_filter_items(self.company, found_filter_items)
self.assertEqual(expected, result)
def test_default_filter(self):
"""
should produce a populated filter object.
"""
ds = CommRecordsDataSource()
default_filter = ds.get_default_filter(None, self.company)
self.assertEquals(datetime.now().year,
default_filter.date_time.dates[1].year)
# Take out value dated today. Too hard to run through assertEquals.
default_filter.date_time.dates[1] = None
expected = CommRecordsFilter(
date_time=DateRangeFilter([datetime(2014, 1, 1), None]))
self.assertEquals(expected, default_filter)
class TestCommRecordsDataSource(MyJobsBase):
def setUp(self):
super(TestCommRecordsDataSource, self).setUp()
# A company to work with
self.company = CompanyFactory(name='right')
self.company.save()
# A separate company that should not show up in results.
self.other_company = CompanyFactory(name='wrong')
self.other_company.save()
self.partner_a = PartnerFactory(
owner=self.company,
uri='http://www.example.com/',
data_source="zap",
name="aaa")
self.partner_b = PartnerFactory(
owner=self.company,
uri='http://www.asdf.com/',
data_source="bcd",
name="bbb")
# An unapproved parther. Associated data should be filtered out.
self.partner_unapp = PartnerFactory(
owner=self.company,
name="unapproved",
approval_status__code=Status.UNPROCESSED)
# An archived parther. Associated data should be filtered out.
self.partner_archived = PartnerFactory(owner=self.company)
self.east_tag = TagFactory.create(
company=self.company, name='east', hex_color="aaaaaa")
self.west_tag = TagFactory.create(
company=self.company, name='west', hex_color="bbbbbb")
self.north_tag = TagFactory.create(
company=self.company, name='north', hex_color="cccccc")
self.south_tag = TagFactory.create(
company=self.company, name='south', hex_color="dddddd")
self.left_tag = TagFactory.create(
company=self.company, name='left', hex_color="eeeeee")
self.right_tag = TagFactory.create(
company=self.company, name='right', hex_color="ffffff")
self.bad_tag = TagFactory.create(
company=self.company, name='bad', hex_color="cccccc")
self.partner_a.tags.add(self.left_tag)
self.partner_b.tags.add(self.right_tag)
self.john_user = UserFactory(email="*****@*****.**")
self.john = ContactFactory(
partner=self.partner_a,
name='john adams',
user=self.john_user,
email="*****@*****.**",
last_action_time='2015-10-03')
self.john.locations.add(
LocationFactory.create(
city="Indianapolis",
state="IN"))
self.john.locations.add(
LocationFactory.create(
city="Chicago",
state="IL"))
self.john.tags.add(self.north_tag)
self.sue_user = UserFactory(email="*****@*****.**")
self.sue = ContactFactory(
partner=self.partner_b,
name='Sue Baxter',
user=self.sue_user,
email="*****@*****.**",
last_action_time='2015-09-30 13:23')
self.sue.locations.add(
LocationFactory.create(
address_line_one="123",
city="Los Angeles",
state="CA"))
self.sue.locations.add(
LocationFactory.create(
address_line_one="234",
city="Los Angeles",
state="CA"))
self.sue.tags.add(self.south_tag)
self.partner_a.primary_contact = self.john
self.partner_b.primary_contact = self.sue
self.partner_a.save()
self.partner_b.save()
self.record_1 = ContactRecordFactory(
subject='record 1',
date_time='2015-09-30 13:23',
contact=self.john,
contact_type="Email",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_2 = ContactRecordFactory(
subject='record 2',
date_time='2015-01-01',
contact=self.john,
contact_type="Meeting Or Event",
partner=self.partner_a,
location='Indianapolis, IN',
tags=[self.east_tag])
self.record_3 = ContactRecordFactory(
subject='record 3',
date_time='2015-10-03',
contact=self.sue,
contact_type="Phone",
partner=self.partner_b,
location='Los Angeles, CA',
tags=[self.west_tag])
# Archive archived data here. Doing this earlier in the set up results
# in odd exceptions.
self.partner_archived.archive()
def test_run_unfiltered(self):
"""Make sure we only get data for this user."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(self.company, CommRecordsFilter(), [])
subjects = {r['subject'] for r in recs}
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_date_range(self):
"""Should show only commrec with last_action_time in range."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter([
datetime(2015, 9, 1),
datetime(2015, 9, 30)])
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(date_time=date_range),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_before(self):
"""Should show only commrec with last_action_time before date."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter(
[None, datetime(2015, 9, 30)])
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(date_time=date_range),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_date_after(self):
"""Should show only commrec with last_action_time after date."""
ds = CommRecordsDataSource()
date_range = DateRangeFilter(
[datetime(2015, 10, 1), None])
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(date_time=date_range),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_state(self):
"""Should show only commrecs with correct state."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({'state': MatchFilter('CA')})),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_city(self):
"""Should show only commrecs with correct city."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({
'city': MatchFilter('Los Angeles')})),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags(self):
"""
Show only commrec with correct tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags_unlinked(self):
"""
Only return untagged commrecs.
"""
self.record_1.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=UnlinkedFilter()),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_tags_or(self):
"""Show only commrec with correct tags in 'or' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt'), MatchFilter('wEsT')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {
self.record_1.subject,
self.record_2.subject,
self.record_3.subject,
}
self.assertEqual(expected, subjects)
def test_filter_by_tags_and(self):
"""Show only commrec with correct tags in 'and' configuration."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])])),
[])
subjects = {r['subject'] for r in recs}
expected = set()
self.assertEqual(expected, subjects)
# Now try adding another tag.
self.record_1.tags.add(self.west_tag)
self.record_1.save()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_communication_type(self):
"""Check communication_type filter works at all."""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(communication_type=OrGroupFilter([
MatchFilter('email')])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner(self):
"""
Check partner filter works at all.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
partner=OrGroupFilter([MatchFilter(self.partner_a.pk)])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_1.subject, self.record_2.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact(self):
"""
Check partner filter works at all.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
contact=OrGroupFilter([MatchFilter(self.sue.pk)])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact_tags(self):
"""
Test that we can filter by contact tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
contact_tags=AndGroupFilter([
OrGroupFilter([MatchFilter('sOuTh')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_contact_tags_untagged(self):
"""
Check that we can find a record attached to an untagged contact.
"""
self.sue.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(contact_tags=UnlinkedFilter()),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner_tags(self):
"""
Test that we can filter by partner tags.
"""
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(
partner_tags=AndGroupFilter([
OrGroupFilter([MatchFilter('rigHt')])])),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_filter_by_partner_tags_untagged(self):
"""
Check that we can find a record attached to an untagged partner.
"""
self.partner_b.tags.clear()
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(partner_tags=UnlinkedFilter()),
[])
subjects = {r['subject'] for r in recs}
expected = {self.record_3.subject}
self.assertEqual(expected, subjects)
def test_help_city(self):
"""
Check city help works and ignores current city filter.
"""
ds = CommRecordsDataSource()
recs = ds.help_city(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({
'city': MatchFilter("zz")})),
"angel")
actual = {r['value'] for r in recs}
self.assertEqual({'Los Angeles'}, actual)
def test_help_state(self):
"""
Check state help works and ignores current state filter.
"""
ds = CommRecordsDataSource()
recs = ds.help_state(
self.company,
CommRecordsFilter(
locations=CompositeAndFilter({
'state': MatchFilter("zz")})),
"i")
actual = {r['value'] for r in recs}
self.assertEqual({'IL', 'IN'}, actual)
def test_help_tags(self):
"""
Check tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "E")
actual = {r['value'] for r in recs}
self.assertEqual({'east', 'west'}, actual)
def test_help_contact_tags(self):
"""
Check contact tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_contact_tags(self.company, CommRecordsFilter(), "O")
actual = {r['value'] for r in recs}
self.assertEqual({'north', 'south'}, actual)
def test_help_partner_tags(self):
"""
Check partner tags help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_partner_tags(self.company, CommRecordsFilter(), "t")
actual = {r['value'] for r in recs}
self.assertEqual({'left', 'right'}, actual)
def test_help_tags_colors(self):
"""
Tags should have colors
"""
ds = CommRecordsDataSource()
recs = ds.help_tags(self.company, CommRecordsFilter(), "east")
self.assertEqual("aaaaaa", recs[0]['hexColor'])
def test_help_communication_types(self):
"""
Check communication_types help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_communication_type(
self.company, CommRecordsFilter(), "ph")
actual = {r['value'] for r in recs}
expected = {'phone', 'job', 'meetingorevent', 'email', 'pssemail'}
self.assertEqual(expected, actual)
def test_help_partner(self):
"""
Check partner help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_partner(self.company, CommRecordsFilter(), "A")
self.assertEqual(
[{'value': self.partner_a.pk, 'display': self.partner_a.name}],
recs)
def test_help_contact(self):
"""
Check contact help works at all.
"""
ds = CommRecordsDataSource()
recs = ds.help_contact(self.company, CommRecordsFilter(), "U")
self.assertEqual(
[{'value': self.sue.pk, 'display': self.sue.name}],
recs)
def test_values(self):
"""
Check limiting values works at all
"""
self.maxDiff = 10000
ds = CommRecordsDataSource()
recs = ds.run_unaggregated(
self.company,
CommRecordsFilter(),
["partner", "subject"])
expected = [
{'partner': u'aaa', 'subject': u'record 1'},
{'partner': u'aaa', 'subject': u'record 2'},
{'partner': u'bbb', 'subject': u'record 3'},
]
self.assertEqual(expected, recs)
def test_adorn_filter(self):
self.maxDiff = 10000
found_filter_items = {
'tags': ['east', 'west'],
'communication_type': ['Email'],
'partner': [str(self.partner_a.pk)],
'contact': [str(self.sue.pk)],
'contact_tags': ['nOrth', 'south'],
'partner_tags': ['lEft', 'riGht'],
}
expected = {
u'partner': {
self.partner_a.pk:
{'value': self.partner_a.pk, 'display': u'aaa'},
},
u'contact': {
self.sue.pk:
{'value': self.sue.pk, 'display': u'Sue Baxter'},
},
u'tags': {
u'east': {
'value': u'east',
'display': u'east',
'hexColor': u'aaaaaa',
},
u'west': {
'value': u'west',
'display': u'west',
'hexColor': u'bbbbbb',
},
},
u'communication_type': {
'email': {'value': 'email', 'display': 'Email'}
},
u'contact_tags': {
u'north': {
'value': u'north',
'display': u'north',
'hexColor': u'cccccc',
},
u'south': {
'value': u'south',
'display': u'south',
'hexColor': u'dddddd',
},
},
u'partner_tags': {
u'left': {
'value': u'left',
'display': u'left',
'hexColor': u'eeeeee',
},
u'right': {
'value': u'right',
'display': u'right',
'hexColor': u'ffffff',
},
},
}
ds = CommRecordsDataSource()
result = ds.adorn_filter_items(self.company, found_filter_items)
self.assertEqual(expected, result)
def test_default_filter(self):
"""
should produce a populated filter object.
"""
ds = CommRecordsDataSource()
default_filter = ds.get_default_filter(None, self.company)
self.assertEquals(
datetime.now().year,
default_filter.date_time.dates[1].year)
# Take out value dated today. Too hard to run through assertEquals.
default_filter.date_time.dates[1] = None
expected = CommRecordsFilter(
date_time=DateRangeFilter([datetime(2014, 1, 1), None]))
self.assertEquals(expected, default_filter)
class TestPartnersDataSource(MyJobsBase):
def setUp(self):
super(TestPartnersDataSource, self).setUp()
# A company to work with
self.company = CompanyFactory(name='right')
self.company.save()
# A separate company that should not show up in results.
self.other_company = CompanyFactory(name='wrong')
self.other_company.save()
self.partner_a = PartnerFactory(owner=self.company,
last_action_time='2015-09-30 13:23',
uri='http://www.example.com/',
data_source="zap",
name="aaa")
self.partner_b = PartnerFactory(owner=self.company,
last_action_time='2015-10-03',
uri='http://www.asdf.com/',
data_source="bcd",
name="bbb")
# An unapproved parther. Associated data should be filtered out.
self.partner_unapp = PartnerFactory(
owner=self.company,
name="unapproved",
approval_status__code=Status.UNPROCESSED)
# An archived parther. Associated data should be filtered out.
self.partner_archived = PartnerFactory(owner=self.company)
self.east_tag = TagFactory.create(company=self.company,
name='east',
hex_color="aaaaaa")
self.west_tag = TagFactory.create(company=self.company,
name='west',
hex_color="bbbbbb")
self.bad_tag = TagFactory.create(company=self.company,
name='bad',
hex_color="cccccc")
self.partner_a.tags.add(self.east_tag)
self.partner_b.tags.add(self.west_tag)
self.john_user = UserFactory(email="*****@*****.**")
self.john = ContactFactory(partner=self.partner_a,
name='john adams',
user=self.john_user,
email="*****@*****.**",
last_action_time='2015-10-03')
self.john.locations.add(
LocationFactory.create(city="Indianapolis", state="IN"))
self.john.locations.add(
LocationFactory.create(city="Chicago", state="IL"))
self.sue_user = UserFactory(email="*****@*****.**")
self.sue = ContactFactory(partner=self.partner_b,
name='Sue Baxter',
user=self.sue_user,
email="*****@*****.**",
last_action_time='2015-09-30 13:23')
self.sue.locations.add(
LocationFactory.create(address_line_one="123",
city="Los Angeles",
state="CA"))
self.sue.locations.add(
LocationFactory.create(address_line_one="234",
city="Los Angeles",
state="CA"))
self.partner_a.primary_contact = self.john
self.partner_b.primary_contact = self.sue
self.partner_a.save()
self.partner_b.save()
# Archive archived data here. Doing this earlier in the set up results
# in odd exceptions.
self.partner_archived.archive()
def test_run_unfiltered(self):
"""Make sure we only get data for this user."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(self.company, PartnersFilter(), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name, self.partner_b.name}
self.assertEqual(expected, names)
expected_tags = {
self.partner_a.pk: {'east'},
self.partner_b.pk: {'west'},
}
def tags_by_pk(rec):
return (rec['partner_id'], set(t['name'] for t in rec['tags']))
found_tags = dict(tags_by_pk(r) for r in recs)
self.assertEqual(expected_tags, found_tags)
def test_run_count_comm_rec_per_month(self):
"""Make sure we only get data for this user."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(subject=subject,
date_time=datetime(2015, month, 1)))
ds = PartnersDataSource()
recs = ds.run_count_comm_rec_per_month(self.company, PartnersFilter(),
[])
data = [(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs]
expected = [
(self.partner_a.name, 2015, 1, 0),
(self.partner_a.name, 2015, 2, 0),
(self.partner_a.name, 2015, 3, 0),
(self.partner_a.name, 2015, 4, 0),
(self.partner_a.name, 2015, 5, 0),
(self.partner_a.name, 2015, 6, 0),
(self.partner_a.name, 2015, 7, 0),
(self.partner_a.name, 2015, 8, 0),
(self.partner_a.name, 2015, 9, 0),
(self.partner_a.name, 2015, 10, 0),
(self.partner_a.name, 2015, 11, 0),
(self.partner_a.name, 2015, 12, 0),
(self.partner_b.name, 2015, 1, 0),
(self.partner_b.name, 2015, 2, 3),
(self.partner_b.name, 2015, 3, 3),
(self.partner_b.name, 2015, 4, 3),
(self.partner_b.name, 2015, 5, 0),
(self.partner_b.name, 2015, 6, 0),
(self.partner_b.name, 2015, 7, 0),
(self.partner_b.name, 2015, 8, 0),
(self.partner_b.name, 2015, 9, 0),
(self.partner_b.name, 2015, 10, 0),
(self.partner_b.name, 2015, 11, 0),
(self.partner_b.name, 2015, 12, 0),
]
self.assertEqual(expected, data)
def test_run_count_comm_rec_per_month_one_partner(self):
"""Trip over a bug in the mysql client driver."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for year in [2013, 2015, 2014]:
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(subject=subject,
date_time=datetime(
year, month, 1)))
ds = PartnersDataSource()
partners_filter = PartnersFilter(
tags=AndGroupFilter([OrGroupFilter([MatchFilter('west')])]))
recs = ds.run_count_comm_rec_per_month(self.company, partners_filter,
['name', 'year', '-month'])
data = [(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs]
self.assertEqual(36, len(data))
def test_run_count_comm_rec_per_month_no_partners(self):
"""Trip over a bug in the mysql client driver."""
for (subject, itx) in enumerate(['a', 'b', 'c']):
for month in [2, 3, 4]:
self.sue.contactrecord_set.add(
ContactRecordFactory(subject=subject,
date_time=datetime(2015, month, 1)))
ds = PartnersDataSource()
partners_filter = PartnersFilter(
tags=AndGroupFilter([OrGroupFilter([MatchFilter('zzz')])]))
recs = ds.run_count_comm_rec_per_month(self.company, partners_filter,
['name', 'year', '-month'])
self.assertEqual(0, len(recs))
def test_run_count_comm_rec_per_month_empty_partner(self):
"""One partner, no communication records."""
ds = PartnersDataSource()
partners_filter = PartnersFilter(
tags=AndGroupFilter([OrGroupFilter([MatchFilter('east')])]))
recs = ds.run_count_comm_rec_per_month(self.company, partners_filter,
['name', 'year', '-month'])
data = [(r['name'], r['year'], r['month'], r['comm_rec_count'])
for r in recs]
self.assertEqual(12, len(data))
default_year = datetime.now().year
for item in data:
self.assertEqual(default_year, item[1])
self.assertEqual(0, item[3])
def test_filter_by_date_range(self):
"""Should show only partner with last_action_time in range."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(date=DateRangeFilter(
[datetime(2015, 9, 1),
datetime(2015, 9, 30)])), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_date_before(self):
"""Should show only partner with last_action_time before date."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
date=DateRangeFilter([None, datetime(2015, 9, 30)])), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_date_after(self):
"""Should show only partner with last_action_time after date."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
date=DateRangeFilter([datetime(2015, 10, 1), None])), [])
names = {r['name'] for r in recs}
expected = {self.partner_b.name}
self.assertEqual(expected, names)
def test_filter_by_state(self):
"""Should show only partners with correct state."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
locations=CompositeAndFilter({
'state': MatchFilter('CA'),
})), [])
names = [r['name'] for r in recs]
expected = [self.partner_b.name]
self.assertEqual(expected, names)
def test_filter_by_city(self):
"""Should show only partners with correct city."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(locations=CompositeAndFilter({
'city':
MatchFilter('Los Angeles'),
})), [])
names = [r['name'] for r in recs]
expected = [self.partner_b.name]
self.assertEqual(expected, names)
def test_filter_by_tags(self):
"""Show only partner with correct tags."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(
tags=AndGroupFilter([OrGroupFilter([MatchFilter('EaSt')])])),
[])
names = {r['name'] for r in recs}
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_tags_or(self):
"""Show only partner with correct tags in 'or' configuration."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=AndGroupFilter(
[OrGroupFilter([MatchFilter('EaSt'),
MatchFilter('wEsT')])])), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name, self.partner_b.name}
self.assertEqual(expected, names)
def test_filter_by_tags_and(self):
"""Show only partner with correct tags in 'and' configuration."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])
])), [])
names = {r['name'] for r in recs}
expected = set()
self.assertEqual(expected, names)
# Now try adding another tag.
self.partner_a.tags.add(self.west_tag)
recs = ds.run_unaggregated(
self.company,
PartnersFilter(tags=AndGroupFilter([
OrGroupFilter([MatchFilter('EaSt')]),
OrGroupFilter([MatchFilter('wEsT')])
])), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_data_source(self):
"""Check datasource filter works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company, PartnersFilter(data_source=MatchFilter("zap")), [])
names = {r['name'] for r in recs}
expected = {self.partner_a.name}
self.assertEqual(expected, names)
def test_filter_by_uri(self):
"""Check uri filter works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(
self.company,
PartnersFilter(uri=MatchFilter("http://www.asdf.com/")), [])
names = {r['name'] for r in recs}
expected = {self.partner_b.name}
self.assertEqual(expected, names)
def test_help_city(self):
"""Check city help works and ignores current city filter."""
ds = PartnersDataSource()
recs = ds.help_city(
self.company,
PartnersFilter(
locations=CompositeAndFilter({'city': MatchFilter("zz")})),
"angel")
actual = {r['value'] for r in recs}
self.assertEqual({'Los Angeles'}, actual)
def test_help_state(self):
"""Check state help works and ignores current state filter."""
ds = PartnersDataSource()
recs = ds.help_state(
self.company,
PartnersFilter(
locations=CompositeAndFilter({'state': MatchFilter("zz")})),
"i")
actual = {r['value'] for r in recs}
self.assertEqual({'IL', 'IN'}, actual)
def test_help_tags(self):
"""Check tags help works at all."""
ds = PartnersDataSource()
recs = ds.help_tags(self.company, PartnersFilter(), "E")
actual = {r['value'] for r in recs}
self.assertEqual({'east', 'west'}, actual)
def test_help_tags_colors(self):
"""Tags should have colors"""
ds = PartnersDataSource()
recs = ds.help_tags(self.company, PartnersFilter(), "east")
self.assertEqual("aaaaaa", recs[0]['hexColor'])
def test_help_uri(self):
"""Check uri help works at all."""
ds = PartnersDataSource()
recs = ds.help_uri(self.company, PartnersFilter(), "ex")
actual = {r['value'] for r in recs}
self.assertEqual({'http://www.example.com/'}, actual)
def test_help_data_source(self):
"""Check data_source help works at all."""
ds = PartnersDataSource()
recs = ds.help_data_source(self.company, PartnersFilter(), "z")
actual = {r['value'] for r in recs}
self.assertEqual({'zap'}, actual)
def test_values(self):
"""Check limiting values works at all."""
ds = PartnersDataSource()
recs = ds.run_unaggregated(self.company, PartnersFilter(),
["name", "uri"])
expected = [
{
'name': self.partner_a.name,
'uri': 'http://www.example.com/'
},
{
'name': self.partner_b.name,
'uri': 'http://www.asdf.com/'
},
]
self.assertEqual(expected, recs)
def test_adorn_filter_items(self):
found_filter_items = {
'tags': ['east', 'west'],
'uri': 'http://www.example.com/',
}
expected = {
u'tags': {
'east': {
'value': u'east',
'display': u'east',
'hexColor': u'aaaaaa',
},
'west': {
'value': u'west',
'display': u'west',
'hexColor': u'bbbbbb',
},
},
}
ds = PartnersDataSource()
adorned_filter = ds.adorn_filter_items(self.company,
found_filter_items)
self.assertEqual(expected, adorned_filter)
def test_default_filter(self):
"""should produce a populated filter object."""
ds = PartnersDataSource()
default_filter = ds.get_default_filter(None, self.company)
self.assertEquals(datetime.now().year,
default_filter.date.dates[1].year)
# Take out value dated today. Too hard to run through assertEquals.
default_filter.date.dates[1] = None
expected = PartnersFilter(
date=DateRangeFilter([datetime(2014, 1, 1), None]))
self.assertEquals(expected, default_filter)
class MyPartnerTests(MyJobsBase):
def setUp(self):
super(MyPartnerTests, self).setUp()
self.company = CompanyFactory()
self.partner = PartnerFactory(owner=self.company)
self.contact = ContactFactory(partner=self.partner)
def test_contact_to_partner_relationship(self):
"""
Tests adding a contact to partner's contacts list and tests
primary_contact. Also tests if the contact gets deleted the partner
stays and turns primary_contact to None.
"""
self.assertEqual(Contact.objects.filter(partner=self.partner).count(),
1)
self.partner.primary_contact = self.contact
self.partner.save()
self.assertIsNotNone(self.partner.primary_contact)
# making sure contact is the contact obj vs a factory object.
contact = Contact.objects.get(name=self.contact.name)
contact.delete()
partner = Partner.objects.get(name=self.partner.name)
self.assertFalse(Contact.objects.filter(
partner=partner, archived_on__isnull=True))
self.assertIsNone(partner.primary_contact)
def test_contact_user_relationship(self):
"""
Tests adding a User to Contact. Then tests to make sure User cascading
delete doesn't delete the Contact and instead turns Contact.user to
None.
"""
self.contact.user = UserFactory(email=self.contact.email)
self.contact.save()
self.assertIsNotNone(self.contact.user)
self.assertEqual(self.contact.name, self.contact.__unicode__())
user = User.objects.get(email=self.contact.email)
user.delete()
contact = Contact.objects.get(name=self.contact.name)
self.assertIsNone(contact.user)
def test_location_to_contact_relationship(self):
"""
Tests adding a Location to Contact.
"""
location = LocationFactory()
# make sure that we can add a location to a contact
self.contact.locations.add(location)
self.contact.save()
self.assertTrue(len(self.contact.locations.all()) > 0)
# ensure that we can remove a location
self.contact.locations.remove(location)
self.assertTrue(len(self.contact.locations.all()) == 0)
# make sure that removing a location from a contact doesn't delete that
# location entirely
self.assertIn(location, Location.objects.all())
def test_bad_filename(self):
"""
Confirms that non-alphanumeric or underscore characters are being
stripped from file names.
"""
actual_file = path.join(path.abspath(path.dirname(__file__)), 'data',
'test.txt')
f = File(open(actual_file))
filenames = [
('zz\\x80\\xff*file(copy)na.me.htm)_-)l',
'zzx80xfffilecopyname.htm_l'),
('...', 'unnamed_file'),
('..', 'unnamed_file'),
('../../file.txt', 'file.txt'),
('../..', 'unnamed_file'),
('\.\./file.txt', 'file.txt'),
('fiяыle.txt', 'file.txt')
]
for filename, expected_filename in filenames:
f.name = filename
prm_attachment = PRMAttachmentFactory(attachment=f)
result = PRMAttachment.objects.get(
attachment__contains=expected_filename)
result.delete()
def test_partner_saved_search_delete_contact(self):
"""
When a contact gets deleted, we should log it and disable any partner
saved searches for that contact
"""
user = UserFactory(email='*****@*****.**')
self.contact.user = user
self.contact.save()
self.contact = Contact.objects.get(pk=self.contact.pk)
owner = UserFactory(email='*****@*****.**')
partner_saved_search = PartnerSavedSearchFactory(created_by=owner,
provider=self.company,
partner=self.partner,
user=user,
notes='')
self.assertTrue(partner_saved_search.is_active)
self.contact.delete()
partner_saved_search = PartnerSavedSearch.objects.get(
pk=partner_saved_search.pk)
self.assertFalse(partner_saved_search.is_active)
self.assertTrue(self.contact.name in partner_saved_search.notes)
def test_tag_added_to_taggable_models(self):
tag = TagFactory(company=self.company)
tag.save()
tag2 = TagFactory(name="bar", company=self.company)
tag2.save()
cr = ContactRecordFactory(partner=self.partner)
# Add tag to models
cr.tags.add(tag)
self.partner.tags.add(tag)
self.partner.save()
self.contact.tags.add(tag)
self.contact.save()
# Check to make sure it was added
self.assertEquals(1, len(cr.tags.all()))
self.assertEquals(1, len(self.partner.tags.all()))
self.assertEquals(1, len(self.contact.tags.all()))
# Add a 2nd tag and check
self.partner.tags.add(tag2)
self.partner.save()
self.assertEquals(2, len(self.partner.tags.all()))
def test_contact_archived(self):
"""Test that attempting to delete a contact archives it instead."""
self.assertFalse(self.contact.archived_on)
self.contact.archive()
self.assertEqual(Contact.objects.count(), 0)
self.assertEqual(Contact.all_objects.count(), 1)
self.assertTrue(self.contact.archived_on)
def test_archived_manager_weirdness(self):
"""
Demonstrates that archived instances are returned by the
ArchivedModel.all_objects manager, not the standard
ArchivedModel.objects.
"""
self.partner.archive()
# Try retrieving the archived partner using both managers. The "objects"
# manager excludes archived instances so an exception is raised.
self.assertRaises(Partner.DoesNotExist,
lambda: Partner.objects.get(pk=self.partner.pk))
Partner.all_objects.get(pk=self.partner.pk)
# This contact has not been archived (as we demonstrate in a few lines)
# but is excluded by the "objects" manager as its partner has been
# archived.
self.assertRaises(Contact.DoesNotExist,
lambda: Contact.objects.get(pk=self.contact.pk))
self.contact = Contact.all_objects.get(pk=self.contact.pk)
self.assertFalse(self.contact.archived_on)
# The manager used by related objects in this instance excludes
# archived partners. As it's basically a Partner.objects.get(id=...),
# this fails.
self.assertRaises(Partner.DoesNotExist, lambda: self.contact.partner)
self.assertEqual(self.contact.partner_id, self.partner.pk)
def test_archive_primary_contacts(self):
"""
Archiving a primary contact should clear that contact's status as the
partner's primary contact. Doing otherwise raises exceptions.
"""
self.partner.primary_contact = self.contact
self.partner.save()
self.partner.primary_contact.archive()
self.partner = Partner.objects.get(pk=self.partner.pk)
self.partner.primary_contact
def test_models_approved(self):
"""
By default, new partners, contacts, and contactrecords should be
approved.
"""
contactrecord = ContactRecordFactory(partner=self.partner)
for instance in (self.contact, self.partner, contactrecord):
self.assertEqual(instance.approval_status.code, Status.APPROVED)
def test_contact_locations(self):
"""
Test that `get_contact_locations` returns a properly formatted string.
"""
ny = LocationFactory.create_batch(2, city="Albany", state="NY")
il = LocationFactory.create(city="Chicago", state="IL")
mo = LocationFactory.create(city="St. Louis", state="MO")
contacts = ContactFactory.create_batch(4, partner=self.partner)
for contact, location in zip(contacts, ny + [il, mo]):
contact.locations.add(location)
self.assertEqual("Chicago, IL; St. Louis, MO; Albany, NY",
"; ".join(self.partner.get_contact_locations()))
def test_uncommon_outreach_email_domain(self):
"""
Adding an uncommon email domain for outreach to a company should work.
"""
# data migrations aren't run during tests, so we populate manually
CommonEmailDomain.objects.create(domain="gmail.com")
with self.assertRaises(ValidationError):
OutreachEmailDomain.objects.create(company=self.company,
domain="gmail.com")
def test_outreach_domain_unique_to_company(self):
"""
Allowed domains should be unique within a company, but not necessarily
across PRM.
"""
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
# duplicate domains allowed between companies
company = CompanyFactory.create(name="A Whole New World")
OutreachEmailDomain.objects.create(company=company,
domain="foo.bar")
# dupliate domains disallowed within the same company
with self.assertRaises(IntegrityError):
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
def test_contact_record_report_numbers(self):
"""
Contact records have properties which represent various aggregated
values. This test ensures that given a number of contact records, those
aggregated numbers are correct.
"""
email_record = ContactRecordFactory(contact_type="email",
partner=self.partner,
contact=self.contact)
job_record = ContactRecordFactory(contact_type="job",
partner=self.partner,
contact=self.contact,
job_applications=10,
job_interviews=6,
job_hires=5)
phone_record = ContactRecordFactory(contact_type="phone",
partner=self.partner,
contact=ContactFactory(name="Joe"))
records = ContactRecord.objects.all()
self.assertEqual(len(records), 3)
self.assertEqual(len(records.contacts), 2)
# job follow ups don't count as comm activity
self.assertEqual(len(records.communication_activity), 2)
# only job follow ups count as referrals
self.assertEqual(records.referrals, 1)
self.assertEqual(records.applications, 10)
self.assertEqual(records.interviews, 6)
self.assertEqual(records.hires, 5)
self.assertEqual(records.emails, 1)
self.assertEqual(records.calls, 1)
class MyPartnerTests(MyJobsBase):
def setUp(self):
super(MyPartnerTests, self).setUp()
self.company = CompanyFactory()
self.partner = PartnerFactory(owner=self.company)
self.contact = ContactFactory(partner=self.partner)
def test_contact_to_partner_relationship(self):
"""
Tests adding a contact to partner's contacts list and tests
primary_contact. Also tests if the contact gets deleted the partner
stays and turns primary_contact to None.
"""
self.assertEqual(Contact.objects.filter(partner=self.partner).count(),
1)
self.partner.primary_contact = self.contact
self.partner.save()
self.assertIsNotNone(self.partner.primary_contact)
# making sure contact is the contact obj vs a factory object.
contact = Contact.objects.get(name=self.contact.name)
contact.delete()
partner = Partner.objects.get(name=self.partner.name)
self.assertFalse(Contact.objects.filter(
partner=partner, archived_on__isnull=True))
self.assertIsNone(partner.primary_contact)
def test_contact_user_relationship(self):
"""
Tests adding a User to Contact. Then tests to make sure User cascading
delete doesn't delete the Contact and instead turns Contact.user to
None.
"""
self.contact.user = UserFactory(email=self.contact.email)
self.contact.save()
self.assertIsNotNone(self.contact.user)
self.assertEqual(self.contact.name, self.contact.__unicode__())
user = User.objects.get(email=self.contact.email)
user.delete()
contact = Contact.objects.get(name=self.contact.name)
self.assertIsNone(contact.user)
def test_location_to_contact_relationship(self):
"""
Tests adding a Location to Contact.
"""
location = LocationFactory()
# make sure that we can add a location to a contact
self.contact.locations.add(location)
self.contact.save()
self.assertTrue(len(self.contact.locations.all()) > 0)
# ensure that we can remove a location
self.contact.locations.remove(location)
self.assertTrue(len(self.contact.locations.all()) == 0)
# make sure that removing a location from a contact doesn't delete that
# location entirely
self.assertIn(location, Location.objects.all())
def test_bad_filename(self):
"""
Confirms that non-alphanumeric or underscore characters are being
stripped from file names.
"""
actual_file = path.join(path.abspath(path.dirname(__file__)), 'data',
'test.txt')
f = File(open(actual_file))
filenames = [
('zz\\x80\\xff*file(copy)na.me.htm)_-)l',
'zzx80xfffilecopyname.htm_l'),
('...', 'unnamed_file'),
('..', 'unnamed_file'),
('../../file.txt', 'file.txt'),
('../..', 'unnamed_file'),
('\.\./file.txt', 'file.txt'),
('fiяыle.txt', 'file.txt')
]
for filename, expected_filename in filenames:
f.name = filename
prm_attachment = PRMAttachment(attachment=f)
setattr(prm_attachment, 'partner', self.partner)
prm_attachment.save()
result = PRMAttachment.objects.get(
attachment__contains=expected_filename)
result.delete()
def test_partner_saved_search_delete_contact(self):
"""
When a contact gets deleted, we should log it and disable any partner
saved searches for that contact
"""
user = UserFactory(email='*****@*****.**')
self.contact.user = user
self.contact.save()
self.contact = Contact.objects.get(pk=self.contact.pk)
owner = UserFactory(email='*****@*****.**')
partner_saved_search = PartnerSavedSearchFactory(created_by=owner,
provider=self.company,
partner=self.partner,
user=user,
notes='')
self.assertTrue(partner_saved_search.is_active)
self.contact.delete()
partner_saved_search = PartnerSavedSearch.objects.get(
pk=partner_saved_search.pk)
self.assertFalse(partner_saved_search.is_active)
self.assertTrue(self.contact.name in partner_saved_search.notes)
def test_tag_added_to_taggable_models(self):
tag = TagFactory(company=self.company)
tag.save()
tag2 = TagFactory(name="bar", company=self.company)
tag2.save()
cr = ContactRecordFactory(partner=self.partner)
# Add tag to models
cr.tags.add(tag)
self.partner.tags.add(tag)
self.partner.save()
self.contact.tags.add(tag)
self.contact.save()
# Check to make sure it was added
self.assertEquals(1, len(cr.tags.all()))
self.assertEquals(1, len(self.partner.tags.all()))
self.assertEquals(1, len(self.contact.tags.all()))
# Add a 2nd tag and check
self.partner.tags.add(tag2)
self.partner.save()
self.assertEquals(2, len(self.partner.tags.all()))
def test_contact_archived(self):
"""Test that attempting to delete a contact archives it instead."""
self.assertFalse(self.contact.archived_on)
self.contact.delete()
self.assertEqual(len(Contact.objects.all()), 1)
self.assertTrue(self.contact.archived_on)
class MyPartnerTests(MyJobsBase):
def setUp(self):
super(MyPartnerTests, self).setUp()
self.company = CompanyFactory()
self.partner = PartnerFactory(owner=self.company)
self.contact = ContactFactory(partner=self.partner)
def test_contact_to_partner_relationship(self):
"""
Tests adding a contact to partner's contacts list and tests
primary_contact. Also tests if the contact gets deleted the partner
stays and turns primary_contact to None.
"""
self.assertEqual(Contact.objects.filter(partner=self.partner).count(),
1)
self.partner.primary_contact = self.contact
self.partner.save()
self.assertIsNotNone(self.partner.primary_contact)
# making sure contact is the contact obj vs a factory object.
contact = Contact.objects.get(name=self.contact.name)
contact.delete()
partner = Partner.objects.get(name=self.partner.name)
self.assertFalse(Contact.objects.filter(
partner=partner, archived_on__isnull=True))
self.assertIsNone(partner.primary_contact)
def test_contact_user_relationship(self):
"""
Tests adding a User to Contact. Then tests to make sure User cascading
delete doesn't delete the Contact and instead turns Contact.user to
None.
"""
self.contact.user = UserFactory(email=self.contact.email)
self.contact.save()
self.assertIsNotNone(self.contact.user)
self.assertEqual(self.contact.name, self.contact.__unicode__())
user = User.objects.get(email=self.contact.email)
user.delete()
contact = Contact.objects.get(name=self.contact.name)
self.assertIsNone(contact.user)
def test_location_to_contact_relationship(self):
"""
Tests adding a Location to Contact.
"""
location = LocationFactory()
# make sure that we can add a location to a contact
self.contact.locations.add(location)
self.contact.save()
self.assertTrue(len(self.contact.locations.all()) > 0)
# ensure that we can remove a location
self.contact.locations.remove(location)
self.assertTrue(len(self.contact.locations.all()) == 0)
# make sure that removing a location from a contact doesn't delete that
# location entirely
self.assertIn(location, Location.objects.all())
def test_bad_filename(self):
"""
Confirms that non-alphanumeric or underscore characters are being
stripped from file names.
"""
actual_file = path.join(path.abspath(path.dirname(__file__)), 'data',
'test.txt')
f = File(open(actual_file))
filenames = [
('zz\\x80\\xff*file(copy)na.me.htm)_-)l',
'zzx80xfffilecopyname.htm_l'),
('...', 'unnamed_file'),
('..', 'unnamed_file'),
('../../file.txt', 'file.txt'),
('../..', 'unnamed_file'),
('\.\./file.txt', 'file.txt'),
('fiяыle.txt', 'file.txt')
]
for filename, expected_filename in filenames:
f.name = filename
prm_attachment = PRMAttachmentFactory(attachment=f)
result = PRMAttachment.objects.get(
attachment__contains=expected_filename)
result.delete()
def test_partner_saved_search_delete_contact(self):
"""
When a contact gets deleted, we should log it and disable any partner
saved searches for that contact
"""
user = UserFactory(email='*****@*****.**')
self.contact.user = user
self.contact.save()
self.contact = Contact.objects.get(pk=self.contact.pk)
owner = UserFactory(email='*****@*****.**')
partner_saved_search = PartnerSavedSearchFactory(created_by=owner,
provider=self.company,
partner=self.partner,
user=user,
notes='')
self.assertTrue(partner_saved_search.is_active)
self.contact.delete()
partner_saved_search = PartnerSavedSearch.objects.get(
pk=partner_saved_search.pk)
self.assertFalse(partner_saved_search.is_active)
self.assertTrue(self.contact.name in partner_saved_search.notes)
def test_tag_added_to_taggable_models(self):
tag = TagFactory(company=self.company)
tag.save()
tag2 = TagFactory(name="bar", company=self.company)
tag2.save()
cr = ContactRecordFactory(partner=self.partner)
# Add tag to models
cr.tags.add(tag)
self.partner.tags.add(tag)
self.partner.save()
self.contact.tags.add(tag)
self.contact.save()
# Check to make sure it was added
self.assertEquals(1, len(cr.tags.all()))
self.assertEquals(1, len(self.partner.tags.all()))
self.assertEquals(1, len(self.contact.tags.all()))
# Add a 2nd tag and check
self.partner.tags.add(tag2)
self.partner.save()
self.assertEquals(2, len(self.partner.tags.all()))
def test_contact_archived(self):
"""Test that attempting to delete a contact archives it instead."""
self.assertFalse(self.contact.archived_on)
self.contact.delete()
self.assertEqual(Contact.objects.count(), 0)
self.assertEqual(Contact.all_objects.count(), 1)
self.assertTrue(self.contact.archived_on)
def test_models_approved(self):
"""
By default, new partners, contacts, and contactrecords should be
approved.
"""
contactrecord = ContactRecordFactory(partner=self.partner)
for instance in (self.contact, self.partner, contactrecord):
self.assertEqual(instance.approval_status.code, Status.APPROVED)
def test_contact_locations(self):
"""
Test that `get_contact_locations` returns a properly formatted string.
"""
ny = LocationFactory.create_batch(2, city="Albany", state="NY")
il = LocationFactory.create(city="Chicago", state="IL")
mo = LocationFactory.create(city="St. Louis", state="MO")
contacts = ContactFactory.create_batch(4, partner=self.partner)
for contact, location in zip(contacts, ny + [il, mo]):
contact.locations.add(location)
self.assertEqual("Chicago, IL; St. Louis, MO; Albany, NY",
"; ".join(self.partner.get_contact_locations()))
def test_uncommon_outreach_email_domain(self):
"""
Adding an uncommon email domain for outreach to a company should work.
"""
# data migrations aren't run during tests, so we populate manually
CommonEmailDomain.objects.create(domain="gmail.com")
with self.assertRaises(ValidationError):
OutreachEmailDomain.objects.create(company=self.company,
domain="gmail.com")
def test_outreach_domain_unique_to_company(self):
"""
Allowed domains should be unique within a company, but not necessarily
across PRM.
"""
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
# duplicate domains allowed between companies
company = CompanyFactory.create(name="A Whole New World")
OutreachEmailDomain.objects.create(company=company,
domain="foo.bar")
# dupliate domains disallowed within the same company
with self.assertRaises(IntegrityError):
OutreachEmailDomain.objects.create(company=self.company,
domain="foo.bar")
