Example #1
0
    def test_get_database(self):
        from shortcut import get_database, connection
        db = get_database()
        self.assertEqual(db.connection, connection)

        db = get_database(connection)
        self.assertEqual(db.connection, connection)
Example #2
0
    def test_get_database(self):
        from shortcut import get_database, connection
        db = get_database()
        self.assertEqual(db.connection, connection)

        db = get_database(connection)
        self.assertEqual(db.connection, connection)
Example #3
0
 def get_database(cls):
     return get_database(connection)