From 4a89017868958b545d33bf43eb7f60d74b626335 Mon Sep 17 00:00:00 2001
From: ph <petr.hudik@seznam.cz>
Date: Sun, 6 Sep 2015 09:33:45 +0200
Subject: [PATCH] tests: related query count

---
 tests/Database/Table/bugs/query.count.phpt | 36 ++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 tests/Database/Table/bugs/query.count.phpt

diff --git a/tests/Database/Table/bugs/query.count.phpt b/tests/Database/Table/bugs/query.count.phpt
new file mode 100644
index 000000000..4f281d8ab
--- /dev/null
+++ b/tests/Database/Table/bugs/query.count.phpt
@@ -0,0 +1,36 @@
+<?php
+
+/**
+ * @dataProvider? ../../databases.ini
+ */
+
+use Tester\Assert;
+
+require __DIR__ . '/../../connect.inc.php'; // create $connection
+
+Nette\Database\Helpers::loadFromFile($connection, __DIR__ . "/../../files/{$driverName}-nette_test1.sql");
+
+// add additional tags (not relevant to other tests)
+$context->query("INSERT INTO book_tag_alt (book_id, tag_id, state) VALUES (1, 24, 'private');");
+$context->query("INSERT INTO book_tag_alt (book_id, tag_id, state) VALUES (2, 24, 'private');");
+$context->query("INSERT INTO book_tag_alt (book_id, tag_id, state) VALUES (2, 22, 'private');");
+
+test(function () use ($connection, $context) {
+
+	$context->table('author')->get(11); // have to build cache first
+
+	$count = 0;
+	$connection->onQuery[] = function() use (&$count) {
+		$count ++;
+	};
+
+	foreach ($context->table('book') as $book) {
+		foreach ($book->related('book_tag_alt')->where('state', 'private') as $bookTag) {
+			$tag = $bookTag->tag;
+		}
+	}
+
+	Assert::same(3, $count);
+
+
+});