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_mysql_raises_missing_primary_key_exception(self):
dummy = NotPrimaryModel()
mysql = MySQL(dummy)
self.assertRaises(MySQLMissingPrimaryKey, mysql.detect_primary_key)
