add ten hour end test

diff --git a/src/worktime.c b/src/worktime.c
index d8f1b33c739228a202f2ecd437a6cb1d5689a256..20b37175cbe8127e9582c92b96fb49e2d61f0d43 100644 (file)
--- a/src/worktime.c
+++ b/src/worktime.c
@@ -2,6 +2,7 @@
 #include "break.h"
 
 #define EIGHT_HOURS 28800
+#define TEN_HOURS 36000
 
 time_t get_current_worktime(time_t begin, time_t now)
 {
@@ -28,5 +29,8 @@ time_t get_eight_hour_end_worktime(time_t begin)
 
 time_t get_ten_hour_end_worktime(time_t begin)
 {
-       return 0;
+       time_t worktime = begin + TEN_HOURS;
+       worktime += get_break_time(worktime);
+
+       return worktime;
 }
\ No newline at end of file

diff --git a/tests/worktime_tests.c b/tests/worktime_tests.c
index 721db488421f4e823aa6c07bb94936268b9077e2..4446aae75d5291bf9953b59f3ce0b2bd78199c50 100644 (file)
--- a/tests/worktime_tests.c
+++ b/tests/worktime_tests.c
@@ -15,10 +15,20 @@ void get_eigth_hour_end_time_test()
        assert_int_equal(result, 54900);
 }
 
+void get_ten_hour_end_time_test()
+{
+       time_t begin = 23400;
+
+       time_t result = get_ten_hour_end_worktime(begin);
+
+       assert_int_equal(result, 62100);
+}
+
 int main()
 {
        const struct CMUnitTest tests[] = {
-               cmocka_unit_test(get_eigth_hour_end_time_test)
+               cmocka_unit_test(get_eigth_hour_end_time_test),
+               cmocka_unit_test(get_ten_hour_end_time_test)
        };
 
        return cmocka_run_group_tests(tests, NULL, NULL);