def produce(self):
if self.scheme == 'sqlite':
return SQLite(self.model)
elif self.scheme == 'mysql':
return MySQL(self.model)
elif self.scheme == 'postgres':
return PostgreSQL(self.model)
else:
return CommonSQL(self.model)
def test_sqlite_raises_missing_primary_key_exception(self):
dummy = NotPrimaryModel()
sqlite = SQLite(dummy)
self.assertRaises(SQLiteMissingPrimaryKey, sqlite.detect_primary_key)
