add test for ten hours

diff --git a/src/worktime.c b/src/worktime.c
index e54e5a2bd92f8e6b997f922050aa4d5b27633735..1a92d649da8b16be8fe83ef6d2d4e4565e60f810 100644 (file)
--- a/src/worktime.c
+++ b/src/worktime.c
@@ -30,10 +30,10 @@ time_t get_ten_hour_end_worktime(time_t begin)
 
 time_t get_current_worktime_diff_to_end_eight_hour(time_t begin, time_t now)
 {
-       return EIGHT_HOURS + get_break_time(EIGHT_HOURS) - get_current_worktime(begin, now);
+       return EIGHT_HOURS - get_current_worktime(begin, now);
 }
 
 time_t get_current_worktime_diff_to_end_ten_hour(time_t begin, time_t now)
 {
-       return 0;
+       return TEN_HOURS - get_current_worktime(begin, now);
 }
\ No newline at end of file

diff --git a/tests/worktime_tests.c b/tests/worktime_tests.c
index 7318dddb1bff2fb8fab3e1748bb8589fb66955ce..2c01cb3a94ea6d8e0406c86c2fb4dea9d148ca65 100644 (file)
--- a/tests/worktime_tests.c
+++ b/tests/worktime_tests.c
@@ -41,7 +41,17 @@ void get_diff_to_end_eight_hour_end_test()
 
        time_t result = get_current_worktime_diff_to_end_eight_hour(begin, now);
 
-       assert_int_equal(result, 17220);
+       assert_int_equal(result, 14520);
+}
+
+void get_diff_to_end_ten_hour_end_test()
+{
+       time_t begin = 23400;
+       time_t now = 38580;
+
+       time_t result = get_current_worktime_diff_to_end_ten_hour(begin, now);
+
+       assert_int_equal(result, 21720);
 }
 
 int main()
@@ -51,6 +61,7 @@ int main()
                cmocka_unit_test(get_ten_hour_end_time_test),
                cmocka_unit_test(get_current_worktime_four_hours_thirteen_test),
                cmocka_unit_test(get_diff_to_end_eight_hour_end_test),
+               cmocka_unit_test(get_diff_to_end_ten_hour_end_test),
        };
 
        return cmocka_run_group_tests(tests, NULL, NULL);