FilesExpand file tree

master

/

EnumSQLTest.php

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

136 lines (118 loc) · 6.34 KB

master

/

EnumSQLTest.php

Copy path

Top

File metadata and controls

• Code
• Blame

136 lines (118 loc) · 6.34 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

136

<?php

namespace DBDiff\Tests\Unit;

use PHPUnit\Framework\TestCase;

use DBDiff\Diff\CreateEnum;

use DBDiff\Diff\DropEnum;

use DBDiff\Diff\AlterEnum;

use DBDiff\SQLGen\DiffToSQL\CreateEnumSQL;

use DBDiff\SQLGen\DiffToSQL\DropEnumSQL;

use DBDiff\SQLGen\DiffToSQL\AlterEnumSQL;

use DBDiff\SQLGen\Dialect\PostgresDialect;

class EnumSQLTest extends TestCase

{

// ── CreateEnum ─────────────────────────────────────────────────────────

public function testCreateEnumUp(): void

{

$def = 'CREATE TYPE "status" AS ENUM (\'active\', \'inactive\', \'pending\')';

$diff = new CreateEnum('status', $def);

$sql = new CreateEnumSQL($diff, new PostgresDialect());

$this->assertSame($def . ';', $sql->getUp());

}

public function testCreateEnumDown(): void

{

$def = 'CREATE TYPE "status" AS ENUM (\'active\', \'inactive\', \'pending\')';

$diff = new CreateEnum('status', $def);

$sql = new CreateEnumSQL($diff, new PostgresDialect());

$this->assertSame('DROP TYPE IF EXISTS "status";', $sql->getDown());

}

// ── DropEnum ──────────────────────────────────────────────────────────

public function testDropEnumUp(): void

{

$def = 'CREATE TYPE "old_status" AS ENUM (\'a\', \'b\')';

$diff = new DropEnum('old_status', $def);

$sql = new DropEnumSQL($diff, new PostgresDialect());

$this->assertSame('DROP TYPE IF EXISTS "old_status";', $sql->getUp());

}

public function testDropEnumDown(): void

{

$def = 'CREATE TYPE "old_status" AS ENUM (\'a\', \'b\')';

$diff = new DropEnum('old_status', $def);

$sql = new DropEnumSQL($diff, new PostgresDialect());

$this->assertSame($def . ';', $sql->getDown());

}

// ── AlterEnum ─────────────────────────────────────────────────────────

public function testAlterEnumUpAddsTheLabelRatherThanReplacingTheType(): void

{

// The target is one label behind, so the UP appends. Replacing the

// type here produced a migration PostgreSQL refuses whenever a column

// uses it, which is the normal case (issue #228).

$srcDef = 'CREATE TYPE "priority" AS ENUM (\'low\', \'medium\', \'high\', \'critical\')';

$tgtDef = 'CREATE TYPE "priority" AS ENUM (\'low\', \'medium\', \'high\')';

$diff = new AlterEnum('priority', $srcDef, $tgtDef);

$sql = new AlterEnumSQL($diff, new PostgresDialect());

$up = $sql->getUp();

$this->assertStringContainsString('ADD VALUE IF NOT EXISTS \'critical\' AFTER \'high\';', $up);

$this->assertStringNotContainsString('DROP TYPE', $up);

}

public function testAlterEnumDown(): void

{

$srcDef = 'CREATE TYPE "priority" AS ENUM (\'low\', \'medium\', \'high\', \'critical\')';

$tgtDef = 'CREATE TYPE "priority" AS ENUM (\'low\', \'medium\', \'high\')';

$diff = new AlterEnum('priority', $srcDef, $tgtDef);

$sql = new AlterEnumSQL($diff, new PostgresDialect());

$down = $sql->getDown();

// Reverting an addition means removing a label, which ADD VALUE cannot

// do — so this direction still replaces the type.

$this->assertStringContainsString('DROP TYPE IF EXISTS "priority";', $down);

$this->assertStringNotContainsString('critical', $down);

}

// ── Quoting ───────────────────────────────────────────────────────────

public function testEnumNameWithSpecialCharsIsQuotedProperly(): void

{

$def = 'CREATE TYPE "my-type" AS ENUM (\'x\')';

$diff = new CreateEnum('my-type', $def);

$sql = new CreateEnumSQL($diff, new PostgresDialect());

$this->assertSame('DROP TYPE IF EXISTS "my-type";', $sql->getDown());

}

// ── Edge cases ────────────────────────────────────────────────────────

public function testSingleValueEnum(): void

{

$def = 'CREATE TYPE "boolean_flag" AS ENUM (\'yes\')';

$diff = new CreateEnum('boolean_flag', $def);

$sql = new CreateEnumSQL($diff, new PostgresDialect());

$this->assertSame($def . ';', $sql->getUp());

$this->assertSame('DROP TYPE IF EXISTS "boolean_flag";', $sql->getDown());

}

public function testEnumLabelWithEscapedQuotes(): void

{

$def = 'CREATE TYPE "tricky" AS ENUM (\'it\'\'s\', \'O\'\'Brien\')';

$diff = new CreateEnum('tricky', $def);

$sql = new CreateEnumSQL($diff, new PostgresDialect());

$this->assertSame($def . ';', $sql->getUp());

}

public function testAReplacementIsDropThenCreate(): void

{

// A *removed* label still needs the type replaced — there is no DROP

// VALUE — and the two halves must come out in that order.

$srcDef = 'CREATE TYPE "status" AS ENUM (\'a\', \'b\')';

$tgtDef = 'CREATE TYPE "status" AS ENUM (\'a\', \'b\', \'c\')';

$diff = new AlterEnum('status', $srcDef, $tgtDef);

$sql = new AlterEnumSQL($diff, new PostgresDialect());

$up = $sql->getUp();

$dropPos = strpos($up, 'DROP TYPE IF EXISTS "status"');

$createPos = strpos($up, 'CREATE TYPE "status" AS ENUM');

$this->assertNotFalse($dropPos);

$this->assertNotFalse($createPos);

$this->assertLessThan($createPos, $dropPos, 'DROP must precede CREATE in AlterEnum UP');

}

public function testDropEnumUpAndCreateEnumDownAreInverse(): void

{

$def = 'CREATE TYPE "role" AS ENUM (\'admin\', \'user\', \'guest\')';

$diff = new DropEnum('role', $def);

$sql = new DropEnumSQL($diff, new PostgresDialect());

// DropEnum.getUp() == CreateEnum.getDown()

$this->assertSame('DROP TYPE IF EXISTS "role";', $sql->getUp());

// DropEnum.getDown() == CreateEnum.getUp()

$this->assertSame($def . ';', $sql->getDown());

}

}