def setUp(self): """ Instantiate the adapter. """ from random import randint # Generate a random name for the database database_name = str(randint(0, 9000)) self.adapter = JsonDatabaseAdapter(database=database_name)
def setUp(self): from chatterbot.adapters import Adaptation from chatterbot.adapters.storage import JsonDatabaseAdapter self.mixin = KnownResponseMixin() context = Adaptation() adapter = JsonDatabaseAdapter() adapter.set_context(context) # Simulate a storage adapter setattr(self.mixin, "context", context) setattr(self.mixin.context, "storage", adapter)
class BaseJsonDatabaseAdapterTestCase(TestCase): def setUp(self): """ Instantiate the adapter. """ from random import randint # Generate a random name for the database database_name = str(randint(0, 9000)) self.adapter = JsonDatabaseAdapter(database=database_name) def tearDown(self): """ Remove the test database. """ self.adapter.drop()
def setUp(self): super(JsonStorageIntegrationTests, self).setUp() self.chatbot.storage_adapters = [] self.chatbot.storage = JsonDatabaseAdapter()