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 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
