|
16 | 16 | * See the License for the specific language governing permissions and
|
17 | 17 | * limitations under the License.
|
18 | 18 | */
|
19 |
| -import { ConnectionProvider, newError, Session } from '../src' |
20 |
| -import Driver from '../src/driver' |
| 19 | +import { ConnectionProvider, newError, ServerInfo, Session } from '../src' |
| 20 | +import Driver, { READ } from '../src/driver' |
21 | 21 | import { Bookmarks } from '../src/internal/bookmarks'
|
22 | 22 | import { Logger } from '../src/internal/logger'
|
23 | 23 | import { ConfiguredCustomResolver } from '../src/internal/resolver'
|
@@ -155,6 +155,48 @@ describe('Driver', () => {
|
155 | 155 | expect(driver.isEncrypted()).toEqual(expectedValue)
|
156 | 156 | })
|
157 | 157 |
|
| 158 | + it.each([ |
| 159 | + [undefined, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 160 | + [undefined, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 161 | + [{}, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 162 | + [{}, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 163 | + [{ database: undefined }, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 164 | + [{ database: undefined }, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 165 | + [{ database: 'db' }, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 166 | + [{ database: 'db' }, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 167 | + ])('.verifyConnectivity(%o) => %s', (input: { database?: string } | undefined, _, expectedPromise) => { |
| 168 | + connectionProvider.verifyConnectivityAndGetServerInfo = jest.fn(() => expectedPromise) |
| 169 | + |
| 170 | + const promise: Promise<ServerInfo> = driver!.verifyConnectivity(input) |
| 171 | + |
| 172 | + expect(promise).toBe(expectedPromise) |
| 173 | + expect(connectionProvider.verifyConnectivityAndGetServerInfo) |
| 174 | + .toBeCalledWith({ database: input && input.database ? input.database : '', accessMode: READ }) |
| 175 | + |
| 176 | + promise.catch(_ => 'Do nothing').finally(() => { }) |
| 177 | + }) |
| 178 | + |
| 179 | + it.each([ |
| 180 | + [undefined, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 181 | + [undefined, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 182 | + [{}, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 183 | + [{}, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 184 | + [{ database: undefined }, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 185 | + [{ database: undefined }, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 186 | + [{ database: 'db' }, 'Promise.resolve(ServerInfo>)', Promise.resolve(new ServerInfo())], |
| 187 | + [{ database: 'db' }, 'Promise.reject(Error)', Promise.reject(newError('something went wrong'))], |
| 188 | + ])('.getServerInfo(%o) => %s', (input: { database?: string } | undefined, _, expectedPromise) => { |
| 189 | + connectionProvider.verifyConnectivityAndGetServerInfo = jest.fn(() => expectedPromise) |
| 190 | + |
| 191 | + const promise: Promise<ServerInfo> = driver!.getServerInfo(input) |
| 192 | + |
| 193 | + expect(promise).toBe(expectedPromise) |
| 194 | + expect(connectionProvider.verifyConnectivityAndGetServerInfo) |
| 195 | + .toBeCalledWith({ database: input && input.database ? input.database : '', accessMode: READ }) |
| 196 | + |
| 197 | + promise.catch(_ => 'Do nothing').finally(() => { }) |
| 198 | + }) |
| 199 | + |
158 | 200 | function mockCreateConnectonProvider(connectionProvider: ConnectionProvider) {
|
159 | 201 | return (
|
160 | 202 | id: number,
|
|
0 commit comments