def setup_mock(self, return_value): conn = Mock(Connection) conn.execute_sql_fetch_all_with_params = Mock(return_value=return_value) return conn
def test_book_exists_returns_book(self): conn = Mock(Connection) conn.execute_sql_fetch_all_with_params = lambda x, y: [[0, "Title", "Author"]] o = GetBookByTitleAndAuthor(conn) book = o.execute(Book()) self.assertEquals(0, book.id)