timetracker/tests/Unit/Db/ReportItemMapperTest.php at master · 200529-Hugo/timetracker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
<?php

namespace OCA\TimeTracker\Tests\Unit\Db;

use PHPUnit\Framework\TestCase;
use OCA\TimeTracker\Db\ReportItemMapper;
use OCP\IDBConnection;
use OCP\DB\IPreparedStatement;
use Doctrine\DBAL\Platforms\AbstractPlatform;
use Doctrine\DBAL\Platforms\MySqlPlatform;
use Doctrine\DBAL\Platforms\PostgreSqlPlatform;
use Doctrine\DBAL\Platforms\SqlitePlatform;

class ReportItemMapperTest extends TestCase {
    private $db;

    protected function setUp(): void {
        $this->db = $this->createMock(IDBConnection::class);
    }

    private function getMapperWithPlatform($platform) {
        $this->db->method('getDatabasePlatform')->willReturn($platform);
        return new ReportItemMapper($this->db);
    }

    public function testReportMySql() {
        $platform = $this->createMock(MySqlPlatform::class);
        $mapper = $this->getMapperWithPlatform($platform);

        $stmt = $this->createMock(IPreparedStatement::class);
        $this->db->expects($this->once())
            ->method('prepare')
            ->with($this->callback(function($sql) {
                return strpos($sql, 'DATE(FROM_UNIXTIME(start))') !== false;
            }))
            ->willReturn($stmt);

        $mapper->report('user', 0, 0, [], [], [], 'day', '', '', false, 0, 100);
    }

    public function testReportPostgres() {
        $platform = $this->createMock(PostgreSqlPlatform::class);
        $mapper = $this->getMapperWithPlatform($platform);

        $stmt = $this->createMock(IPreparedStatement::class);
        $this->db->expects($this->once())
            ->method('prepare')
            ->with($this->callback(function($sql) {
                return strpos($sql, "to_char(to_timestamp(start), 'YYYY-MM-DD')") !== false;
            }))
            ->willReturn($stmt);

        $mapper->report('user', 0, 0, [], [], [], 'day', '', '', false, 0, 100);
    }

    public function testReportSqlite() {
        $platform = $this->createMock(SqlitePlatform::class);
        $mapper = $this->getMapperWithPlatform($platform);

        $stmt = $this->createMock(IPreparedStatement::class);
        $this->db->expects($this->once())
            ->method('prepare')
            ->with($this->callback(function($sql) {
                return strpos($sql, "strftime('%Y-%m-%d', datetime(start, 'unixepoch'))") !== false;
            }))
            ->willReturn($stmt);

        $mapper->report('user', 0, 0, [], [], [], 'day', '', '', false, 0, 100);
    }

    public function testReportWithComplexFilters() {
        $platform = $this->createMock(MySqlPlatform::class);
        $mapper = $this->getMapperWithPlatform($platform);

        $stmt = $this->createMock(IPreparedStatement::class);
        $this->db->expects($this->once())
            ->method('prepare')
            ->with($this->callback(function($sql) {
                // Verify that BOTH project_id and client id filters are present in the SQL
                return strpos($sql, 'wi.project_id in (?,?)') !== false &&
                       strpos($sql, 'c.id in (?,?)') !== false;
            }))
            ->willReturn($stmt);

        $mapper->report('user', 0, 0, [1, 2], [10, 11], [], 'day', '', '', false, 0, 100);
    }

    public function testReportReturnsBothEntries() {
        $platform = $this->createMock(MySqlPlatform::class);
        $mapper = $this->getMapperWithPlatform($platform);

        $stmt = $this->createMock(IPreparedStatement::class);
        $this->db->method('prepare')->willReturn($stmt);

        // Simulate two rows returned by the database
        // One for Project 1 (Client 10) and one for Project 2 (Client 11)
        $stmt->expects($this->exactly(3))
            ->method('fetch')
            ->willReturnOnConsecutiveCalls(
                [
                    'id' => 101,
                    'name' => 'Work 1',
                    'projectId' => 1,
                    'project' => 'Project X',
                    'clientId' => 10,
                    'client' => 'Client 10',
                    'totalDuration' => 3600,
                    'time' => 1600000000,
                    'ftime' => '2020-09-13',
                    'userUid' => 'user'
                ],
                [
                    'id' => 102,
                    'name' => 'Work 2',
                    'projectId' => 2,
                    'project' => 'Project X',
                    'clientId' => 11,
                    'client' => 'Client 11',
                    'totalDuration' => 1800,
                    'time' => 1600000000,
                    'ftime' => '2020-09-13',
                    'userUid' => 'user'
                ],
                false // End of results
            );

        $results = $mapper->report('user', 0, 0, [1, 2], [10, 11], [], 'day', 'project', 'client', false, 0, 100);

        $this->assertCount(2, $results);
        $this->assertEquals('Client 10', $results[0]->client);
        $this->assertEquals('Client 11', $results[1]->client);
        $this->assertEquals('Project X', $results[0]->project);
        $this->assertEquals('Project X', $results[1]->project);
    }
}