def setUp(self): self.dsn = 'user={} password={} dbname={} host=localhost port={}'\ .format(self.DB_USER, self.DB_PASSWORD, self.DB_NAME, self.DB_PORT) try: self.pg_client = PostgresClient(dsn=self.dsn, pool_size=self.POOL_SIZE) except psycopg2.OperationalError as err: print('Check that postgres docker container is started. ' 'Check README for more information') raise psycopg2.OperationalError(err.message) try: self._create_table() except (psycopg2.Error, pg_client_exc.PgClientError): self._drop_table() self._create_table() # Insert 100 entries with self.pg_client.get_cursor() as cursor: for _ in range(100): insert_str = "INSERT INTO {} (username) VALUES (%s)".format( self.TABLE_NAME) cursor.execute(insert_str, (random.choice(NAMES), ))
def test_create_with_wrong_pool_value(self): with self.assertRaises(ValueError) as err: pg_client = PostgresClient(dsn=self.dsn, pool_size=0) self.assertIsNone(pg_client) self.assertIn('Wrong pool_size value', err)