|
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