1 /* 2 * Copyright 2008 Google Inc. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #include <stdarg.h> 17 #include <stddef.h> 18 #include <setjmp.h> 19 #include <cmockery.h> 20 #include <database.h> 21 22 extern DatabaseConnection* connect_to_customer_database(); 23 extern unsigned int get_customer_id_by_name( 24 DatabaseConnection * const connection, const char * const customer_name); 25 26 // Mock query database function. 27 unsigned int mock_query_database( 28 DatabaseConnection* const connection, const char * const query_string, 29 void *** const results) { 30 *results = (void**)mock(); 31 return (unsigned int)mock(); 32 } 33 34 // Mock of the connect to database function. 35 DatabaseConnection* connect_to_database(const char * const database_url, 36 const unsigned int port) { 37 return (DatabaseConnection*)mock(); 38 } 39 40 void test_connect_to_customer_database(void **state) { 41 will_return(connect_to_database, 0x0DA7ABA53); 42 assert_int_equal((int)connect_to_customer_database(), 0x0DA7ABA53); 43 } 44 45 /* This test fails as the mock function connect_to_database() will have no 46 * value to return. */ 47 void fail_connect_to_customer_database(void **state) { 48 assert_true(connect_to_customer_database() == 49 (DatabaseConnection*)0x0DA7ABA53); 50 } 51 52 void test_get_customer_id_by_name(void **state) { 53 DatabaseConnection connection = { 54 "somedatabase.somewhere.com", 12345678, mock_query_database 55 }; 56 // Return a single customer ID when mock_query_database() is called. 57 int customer_ids = 543; 58 will_return(mock_query_database, &customer_ids); 59 will_return(mock_query_database, 1); 60 assert_int_equal(get_customer_id_by_name(&connection, "john doe"), 543); 61 } 62 63 int main(int argc, char* argv[]) { 64 const UnitTest tests[] = { 65 unit_test(test_connect_to_customer_database), 66 unit_test(test_get_customer_id_by_name), 67 }; 68 return run_tests(tests); 69 } 70