silverstripe / silverstripe-dms

    /**

     * Test that documents can be returned by a given page

     */

    public function testGetByPageWithoutEmbargoes()

    {

        $pageWithEmbargoes = $this->objFromFixture('SiteTree', 's3');

        $documents = $this->dms->getByPage($pageWithEmbargoes);

        // Fixture: 6 documents in set, 1 is embargoed

        $this->assertCount(5, $documents, 'Embargoed documents are excluded by default');

        $this->assertContainsOnlyInstancesOf('DMSDocument', $documents);

    }

    /**

     * Test that embargoed documents are excluded from getByPage

    /**

     * Test that embargoed documents are excluded from getByPage

     */

    public function testGetByPageWithEmbargoedDocuments()

    {

        $pageWithEmbargoes = $this->objFromFixture('SiteTree', 's3');

        $documents = $this->dms->getByPage($pageWithEmbargoes, true);

        // Fixture: 6 documents in set, 1 is embargoed

        $this->assertCount(6, $documents, 'Embargoed documents can be included');

        $this->assertContainsOnlyInstancesOf('DMSDocument', $documents);

    }

    /**

     * Ensure the shortcode handler key is configurable