python-模拟django的数据库保存

我正在尝试模拟django models.Model上的save方法调用.
我正在使用Mock作为模拟库.

我正在测试位于houses.deps.house_factory中的文件house_factory.py中的功能.

house_factory.py：
    从apps.market.models导入House

def create_house(location, date, price):
    house = House(id=None, date, price)
    house.save()

    # calculate some stuff and further expand the house instance
    # for example house.tag.add("some-tag")

    # save after calculations
    house.save()

我想模拟房子模型.

class HouseModelMock(mock.Mock):
    def save(self):
        pass

测试方法,是unittest.TestCase类的一部分

  @patch('apps.deps.house_factory.House', new_callable=HouseModelMock)
  def create_house_test(self, MockedHouse):

      """ Constants """
      DAYS_FROM_TODAY = 55
      DATE = datetime.date.today() + datetime.timedelta(days=DAYS_FROM_TODAY)
      PRICE = 250000

    # A location is also a django module , I'm using factory_boy here for building a 'mocked' location
      location = LocationFactory.build()

      create_house(DATE, PRICE)
      MockedHouse.assert_called_with(None, DATE, PRICE)   
      MockedHouse.save.assert_called_with()

如果我运行此测试,则会得到：

呼叫__
    返回self.call(* arg,** kw)
MemoryError

这是我认真对待django和测试的第一步.所以也许我把事情弄错了,以模拟出数据库调用.

感谢您的帮助,

乔纳斯

解决方法:

“这是我认真对待django和进行测试的第一个尝试”……您无需模拟数据库保存,因为Django在运行python manage.py test时会自动针对creates a test DB运行测试套件.然后只需声明存储在数据库中的值即可.

理想情况下,mock用于修补自己的测试(和逻辑),而不是默认的Django测试.

提示：使用内存数据库进行单元测试,例如sqlite将以下内容放入settings.py文件中：

if 'test' in sys.argv:
    DATABASES['default']['ENGINE'] = 'sqlite3'

这将大大加快您的测试运行速度.

标签：django-unittest,unit-testing,mocking,python,django
来源： https://codeday.me/bug/20191201/2081017.html