marinamele_nose_factory_test.py

# -*- coding: utf-8 -*-
import nose.tools as nt
import factory
from django.test import TestCase
from myapp.models import House

class HouseFactory(factory.DjangoModelFactory):
  FATORY_FOR = House
  location = "Barcelona"
  rooms = 3
  
class TestHouseCreation(TestCase):
  
  def setUp(self):
    self.house1 = HouseFactory.create(balkon=True)

  def tearDown(self):
    self.house1.delete()
  
  def test_house_created(self):
    nt.eq_(self.house1.location, "Barcelona")
    nt.eq_(self.house1.rooms, 3)
    nt.eq_(self.house1.balkon, True)
    
  def test_second_house_created(self):
    house2 = HouseFactory.create(location="Hannover", balkon=False)
    self.assertNotEqual(self.house1.location, house2.location)
    self.assertEqual(self.house1.rooms, house2.rooms)
    self.assertNotEqual(self.house1.balkon, house2.balkon)
    house2.delete()