Example #1
0
    def test_exception_if_table_does_not_exist(self):
        table = mock.Mock(spec=Table)
        table.name = 'users'

        db = Db([table])
        with self.assertRaises(Exception):
            db.insert('despacito', id=2)
Example #2
0
    def test_insert(self):
        table = mock.Mock(spec=Table)
        table.name = 'users'

        db = Db([table])
        db.insert('users', id=2, name='Roger')
        table.insert.assert_called_once_with(id=2, name='Roger')