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
22extern DatabaseConnection* connect_to_product_database();
23
24/* Mock connect to database function.
25 * NOTE: This mock function is very general could be shared between tests
26 * that use the imaginary database.h module. */
27DatabaseConnection* connect_to_database(const char * const url,
28                                        const unsigned int port) {
29    check_expected(url);
30    check_expected(port);
31    return (DatabaseConnection*)mock();
32}
33
34void test_connect_to_product_database(void **state) {
35    expect_string(connect_to_database, url, "products.abcd.org");
36    expect_value(connect_to_database, port, 322);
37    will_return(connect_to_database, 0xDA7ABA53);
38    assert_int_equal((int)connect_to_product_database(), 0xDA7ABA53);
39}
40
41/* This test will fail since the expected URL is different to the URL that is
42 * passed to connect_to_database() by connect_to_product_database(). */
43void test_connect_to_product_database_bad_url(void **state) {
44    expect_string(connect_to_database, url, "products.abcd.com");
45    expect_value(connect_to_database, port, 322);
46    will_return(connect_to_database, 0xDA7ABA53);
47    assert_int_equal((int)connect_to_product_database(), 0xDA7ABA53);
48}
49
50/* This test will fail since the mock connect_to_database() will attempt to
51 * retrieve a value for the parameter port which isn't specified by this
52 * test function. */
53void test_connect_to_product_database_missing_parameter(void **state) {
54    expect_string(connect_to_database, url, "products.abcd.org");
55    will_return(connect_to_database, 0xDA7ABA53);
56    assert_int_equal((int)connect_to_product_database(), 0xDA7ABA53);
57}
58
59int main(int argc, char* argv[]) {
60    const UnitTest tests[] = {
61        unit_test(test_connect_to_product_database),
62        unit_test(test_connect_to_product_database_bad_url),
63        unit_test(test_connect_to_product_database_missing_parameter),
64    };
65    return run_tests(tests);
66}
67