public function setPrimaryKey($primary_key)
{
$this->primary_key = $primary_key;
if ($this->dataSource) {
$this->dataSource->setPrimaryKey($primary_key);
}
return $this;
}
private function assertLimit(IDataSource $source)
{
$all = $source->fetchAll();
Assert::count(self::LIMIT, $all);
Assert::count(self::FULL_USER_COUNT, $source->fetchFullData());
Assert::count(self::FULL_USER_COUNT, $source->fetchAllForExport());
Assert::same(self::FULL_USER_COUNT, $source->getTotalCount());
Assert::same(self::LIMIT, $source->count());
return $all;
}