Census definitions

1 files changed, 20 insertions, 0 deletions

diff --git a/tests/app/Census/CensusOfDenmark1921Test.php b/tests/app/Census/CensusOfDenmark1921Test.php
index 3f3e82dc0b..99a418a7ad 100644
--- a/tests/app/Census/CensusOfDenmark1921Test.php
+++ b/tests/app/Census/CensusOfDenmark1921Test.php
@@ -23,6 +23,8 @@ namespace Fisharebest\Webtrees\Census;
 class CensusOfDenmark1921Test extends \PHPUnit_Framework_TestCase {
 	/**
 	 * Test the census place and date
+	 *
+	 * @covers Fisharebest\Webtrees\Census\CensusOfDenmark1921
 	 */
 	public function testPlaceAndDate() {
 		$census = new CensusOfDenmark1921;
@@ -30,4 +32,22 @@ class CensusOfDenmark1921Test extends \PHPUnit_Framework_TestCase {
 		$this->assertSame('Danmark', $census->censusPlace());
 		$this->assertSame('01 FEB 1921', $census->censusDate());
 	}
+
+	/**
+	 * Test the census columns
+	 *
+	 * @covers Fisharebest\Webtrees\Census\CensusOfDenmark1921
+	 */
+	public function testColumns() {
+		$census  = new CensusOfDenmark1921;
+		$columns = $census->columns();
+
+		$this->assertCount(6, $columns);
+		$this->assertInstanceOf(CensusColumnFullName::class, $columns[0]);
+		$this->assertInstanceOf(CensusColumnAge::class, $columns[1]);
+		$this->assertInstanceOf(CensusColumnCondition::class, $columns[2]);
+		$this->assertInstanceOf(CensusColumnRelationToHead::class, $columns[3]);
+		$this->assertInstanceOf(CensusColumnOccupation::class, $columns[4]);
+		$this->assertInstanceOf(CensusColumnBirthPlace::class, $columns[5]);
+	}
 }