|
| 1 | +<?php |
| 2 | +declare(strict_types=1); |
| 3 | + |
| 4 | +namespace Tasuku43\Tests\MermaidClassDiagram\ClassDiagramRenderer\Node\Relationship; |
| 5 | + |
| 6 | +use PHPUnit\Framework\TestCase; |
| 7 | +use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\Node; |
| 8 | +use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\Relationship\Dependency; |
| 9 | +use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\Relationship\Inheritance; |
| 10 | +use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\Relationship\Realization; |
| 11 | +use Tasuku43\MermaidClassDiagram\ClassDiagramRenderer\Node\Relationship\Relationships; |
| 12 | + |
| 13 | +class RelationshipsTest extends TestCase |
| 14 | +{ |
| 15 | + public function testSort(): void |
| 16 | + { |
| 17 | + $a = $this->mockNode('A'); |
| 18 | + $b = $this->mockNode('B'); |
| 19 | + $c = $this->mockNode('C'); |
| 20 | + |
| 21 | + $relationships = Relationships::empty(); |
| 22 | + |
| 23 | + // Intentionally add in non-sorted order by (from, to) |
| 24 | + $relationships->add(new Inheritance($c, $b)); // key: "C B" |
| 25 | + $relationships->add(new Dependency($a, $c)); // key: "A C" |
| 26 | + $relationships->add(new Realization($b, $a)); // key: "B A" |
| 27 | + |
| 28 | + $relationships->sort(); |
| 29 | + $sorted = $relationships->getAll(); |
| 30 | + |
| 31 | + $this->assertCount(3, $sorted); |
| 32 | + $this->assertInstanceOf(Dependency::class, $sorted[0]); // A C |
| 33 | + $this->assertSame('A', $sorted[0]->from->nodeName()); |
| 34 | + $this->assertSame('C', $sorted[0]->to->nodeName()); |
| 35 | + |
| 36 | + $this->assertInstanceOf(Realization::class, $sorted[1]); // B A |
| 37 | + $this->assertSame('B', $sorted[1]->from->nodeName()); |
| 38 | + $this->assertSame('A', $sorted[1]->to->nodeName()); |
| 39 | + |
| 40 | + $this->assertInstanceOf(Inheritance::class, $sorted[2]); // C B |
| 41 | + $this->assertSame('C', $sorted[2]->from->nodeName()); |
| 42 | + $this->assertSame('B', $sorted[2]->to->nodeName()); |
| 43 | + } |
| 44 | + |
| 45 | + private function mockNode(string $name): Node |
| 46 | + { |
| 47 | + return new class($name) extends Node { |
| 48 | + public function render(): string |
| 49 | + { |
| 50 | + return "class {$this->name} {}"; |
| 51 | + } |
| 52 | + }; |
| 53 | + } |
| 54 | +} |
| 55 | + |
0 commit comments