如何在django中將特定數據庫用於單個測試？

Question

有誰知道如何在Django中對單個測試使用特定的數據庫？

編輯：目前，我正在使用sqlite進行測試，但是，我想使用mysql數據庫進行特定測試（def test_benchmark ...），它將對一些sql語句進行基准測試。

謝謝

Answer 1

在Django中，您可以在settings.py文件中設置數據庫。 配置應如下所示：

DATABASES = {
    'default': {
        'NAME': 'db_name',
        'ENGINE': 'sql_server.pyodbc',
        'HOST': 'server',
        'USER': 'django',
        'PASSWORD': 'django',
        'OPTIONS' : { "use_mars" : False,
                      "provider" : "SQLNCLI10",
                      "extra_params" : "MARS Connection=True"},

    }
}

Answer 2

我不知道有任何內置方法可以執行此操作-但我也沒有誠實地尋找它。 一種可能的（骯臟的）破解方法是檢查您的設置（這只是另一個Python模塊）中sys.argv中的內容，並為您的特定測試提供備用數據庫設置，即：

# settings.py
import os, sys


DATABASES = {    
   'default': {
       'NAME': 'db_name',
        'ENGINE': 'your.engine',
        'HOST': 'server',
        'USER': 'django',
        'PASSWORD': 'django',
    }
}

if "test" in sys.argv and "your-test-name" in sys.argv:
    DATABASES = {    
         'default': {
            'NAME': 'other_db_name',
            'ENGINE': 'your.engine',
            'HOST': 'server',
            'USER': 'django',
            'PASSWORD': 'django',
        }  
    }