Python Pydblibrary.set_current_database Examples

Example #1

File: unit_tests.py Project: F1ashhimself/robotframework-pydblibrary

    def testSetCurrentDatabase(self):
        dbDriver1 = DBDriverMock()
        dbDriver2 = DBDriverMock()

        sut = Pydblibrary()

        sut.connect_to_database(dbDriver1, 'someDbName1', 'someUsername',
                                'somePassword', 'someHost', '7777', 'db1')

        sut.connect_to_database(dbDriver2, 'someDbName2', 'someUsername',
                                'somePassword', 'someHost', '7777', 'db2')

        self.assertTrue(id(sut._connectionCache.current.connection) ==
                        id(dbDriver2.connection),
                        'Current database is db1 but should be db2.')

        sut.set_current_database('db1')

        self.assertTrue(id(sut._connectionCache.current.connection) ==
                        id(dbDriver1.connection),
                        'Current database is db2 but should be db1.')

Example #2

    def testSetCurrentDatabase(self):
        dbDriver1 = DBDriverMock()
        dbDriver2 = DBDriverMock()

        sut = Pydblibrary()

        sut.connect_to_database(dbDriver1, 'someDbName1', 'someUsername',
                                'somePassword', 'someHost', '7777', 'db1')

        sut.connect_to_database(dbDriver2, 'someDbName2', 'someUsername',
                                'somePassword', 'someHost', '7777', 'db2')

        self.assertTrue(
            id(sut._connectionCache.current.connection) == id(
                dbDriver2.connection),
            'Current database is db1 but should be db2.')

        sut.set_current_database('db1')

        self.assertTrue(
            id(sut._connectionCache.current.connection) == id(
                dbDriver1.connection),
            'Current database is db2 but should be db1.')