"SELECT ID FROM CUSTOMERS WHERE NAME = %s", customer_name);
number_of_results = connection->query_database(connection, query_string,
&results);
+
if (number_of_results != 1) {
return -1;
}
- return (unsigned int)(size_t) results[0];
+
+ return (unsigned int)*((int *)results);
}
(void) connection; /* unused */
(void) query_string; /* unused */
- *results = (void**)((size_t)mock());
- return (unsigned int)mock();
+ *results = (void **)mock_ptr_type(int *);
+ return mock_ptr_type(int);
}
/* Mock of the connect to database function. */
};
/* Return a single customer ID when mock_query_database() is called. */
int customer_ids = 543;
+ int rc;
(void) state; /* unused */
will_return(mock_query_database, &customer_ids);
will_return(mock_query_database, 1);
- assert_int_equal(get_customer_id_by_name(&connection, "john doe"), 543);
+
+ rc = get_customer_id_by_name(&connection, "john doe");
+ assert_int_equal(rc, 543);
}
int main(void) {