class StudentTest(TestCase):
def setUp(self):
self.school1 = School(state = "AK", name = "A Fine University")
self.school2 = School(state = "OH", name = "The Finest University")
self.school1.save()
self.school2.save()
self.user1 = User(username = "******")
self.user2 = User(username = "******")
self.user3 = User(username = "******")
self.user1.save()
self.user2.save()
self.user3.save()
self.student1 = Student(user = self.user1, school = self.school1)
self.student2 = Student(user = self.user2, school = self.school1, referred_by = self.user1)
self.student3 = Student(user = self.user3, school = self.school2)
self.student1.save()
self.student2.save()
self.student3.save()
unittest.TestCase.setUp(self)
def test_students_link_to_schools(self):
self.assertEqual(self.school1, self.student1.school)
self.assertEqual(self.school2, self.student3.school, "Student.school does not correctly point to school")
def test_referrer_rep_link(self):
old_rep = self.student1.rep
points = random.randint(0, 100)
self.student2.add_rep(points)
rep_change = self.student1.rep - old_rep
print rep_change
self.assertEqual(rep_change, points, "Referrer does not get correct points added when referree gains points.")
def import_schools():
# First, remove all existing schools:
School.objects.all().delete()
path_to_file = os.path.join(settings.PROJECT_ROOT, 'spudmart', 'CERN', 'schools.csv')
with open(path_to_file, 'rb') as csvfile:
schools = csv.reader(csvfile, delimiter='|')
x = 0
for school in schools:
if x > 1:
break
x += 1
try:
school_obj = School(name=school[0], state=school[1])
school_obj.save()
except Exception as e:
print school
raise e
def import_schools():
# First, remove all existing schools:
School.objects.all().delete()
path_to_file = os.path.join(settings.PROJECT_ROOT, 'spudmart', 'CERN',
'schools.csv')
with open(path_to_file, 'rb') as csvfile:
schools = csv.reader(csvfile, delimiter='|')
x = 0
for school in schools:
if x > 1:
break
x += 1
try:
school_obj = School(name=school[0], state=school[1])
school_obj.save()
except Exception as e:
print school
raise e
def setUp(self):
self.school1 = School(state = "AK", name = "A Fine University")
self.school2 = School(state = "OH", name = "The Finest University")
self.school1.save()
self.school2.save()
self.user1 = User(username = "******")
self.user2 = User(username = "******")
self.user3 = User(username = "******")
self.user1.save()
self.user2.save()
self.user3.save()
self.student1 = Student(user = self.user1, school = self.school1)
self.student2 = Student(user = self.user2, school = self.school1, referred_by = self.user1)
self.student3 = Student(user = self.user3, school = self.school2)
self.student1.save()
self.student2.save()
self.student3.save()
unittest.TestCase.setUp(self)
class StudentTest(TestCase):
def setUp(self):
self.school1 = School(state="AK", name="A Fine University")
self.school2 = School(state="OH", name="The Finest University")
self.school1.save()
self.school2.save()
self.user1 = User(username="******")
self.user2 = User(username="******")
self.user3 = User(username="******")
self.user1.save()
self.user2.save()
self.user3.save()
self.student1 = Student(user=self.user1, school=self.school1)
self.student2 = Student(user=self.user2,
school=self.school1,
referred_by=self.user1)
self.student3 = Student(user=self.user3, school=self.school2)
self.student1.save()
self.student2.save()
self.student3.save()
unittest.TestCase.setUp(self)
def test_students_link_to_schools(self):
self.assertEqual(self.school1, self.student1.school)
self.assertEqual(self.school2, self.student3.school,
"Student.school does not correctly point to school")
def test_referrer_rep_link(self):
old_rep = self.student1.rep
points = random.randint(0, 100)
self.student2.add_rep(points)
rep_change = self.student1.rep - old_rep
print rep_change
self.assertEqual(
rep_change, points,
"Referrer does not get correct points added when referree gains points."
)
def setUp(self):
self.school1 = School(state="AK", name="A Fine University")
self.school2 = School(state="OH", name="The Finest University")
self.school1.save()
self.school2.save()
self.user1 = User(username="******")
self.user2 = User(username="******")
self.user3 = User(username="******")
self.user1.save()
self.user2.save()
self.user3.save()
self.student1 = Student(user=self.user1, school=self.school1)
self.student2 = Student(user=self.user2,
school=self.school1,
referred_by=self.user1)
self.student3 = Student(user=self.user3, school=self.school2)
self.student1.save()
self.student2.save()
self.student3.save()
unittest.TestCase.setUp(self)
