def test_highlighted_subjects(self):
subj_a = SubjectFactory(provider=self.provider, text='A')
subj_b = SubjectFactory(provider=self.provider, text='B')
subj_aa = SubjectFactory(provider=self.provider,
text='AA',
parent=subj_a)
subj_ab = SubjectFactory(provider=self.provider,
text='AB',
parent=subj_a)
subj_ba = SubjectFactory(provider=self.provider,
text='BA',
parent=subj_b)
subj_bb = SubjectFactory(provider=self.provider,
text='BB',
parent=subj_b)
subj_aaa = SubjectFactory(provider=self.provider,
text='AAA',
parent=subj_aa)
assert self.provider.has_highlighted_subjects is False
assert set(self.provider.highlighted_subjects) == set([subj_a, subj_b])
subj_aaa.highlighted = True
subj_aaa.save()
assert self.provider.has_highlighted_subjects is True
assert set(self.provider.highlighted_subjects) == set([subj_aaa])
def test_highlighted_subjects(self):
subj_a = SubjectFactory(provider=self.provider, text='A')
subj_b = SubjectFactory(provider=self.provider, text='B')
subj_aa = SubjectFactory(provider=self.provider, text='AA', parent=subj_a)
subj_ab = SubjectFactory(provider=self.provider, text='AB', parent=subj_a)
subj_ba = SubjectFactory(provider=self.provider, text='BA', parent=subj_b)
subj_bb = SubjectFactory(provider=self.provider, text='BB', parent=subj_b)
subj_aaa = SubjectFactory(provider=self.provider, text='AAA', parent=subj_aa)
assert set(self.provider.highlighted_subjects) == set([subj_a, subj_b])
subj_aaa.highlighted = True
subj_aaa.save()
assert set(self.provider.highlighted_subjects) == set([subj_aaa])
