Filter out non public treebanks from list

ufal · Aug 24, 2015 · a9a854b · a9a854b
1 parent 7e9a407
commit a9a854b
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/lib/PMLTQ/Server/Controller/Treebank.pm b/lib/PMLTQ/Server/Controller/Treebank.pm
@@ -19,7 +19,7 @@ use PMLTQ::Server::Validation;
 sub list {
   my $c = shift;
 
-  $c->mandel->collection('treebank')->all(sub {
+  $c->mandel->collection('treebank')->search({ public => Mojo::JSON->true })->all(sub {
     my($collection, $err, $treebanks) = @_;
 
     $c->render(json => [ map {

diff --git a/t/api_treebank.t b/t/api_treebank.t
@@ -131,8 +131,17 @@ ok ($treebanks_url, 'Treebanks url');
 $t->get_ok($treebanks_url)
   ->status_is(200);
 
+is(scalar(@{$t->tx->res->json}), 1, 'Returned one treebank');
 $t->json_is("/0/$_", $tb->$_) for (qw/id name title description homepage/);
-#$t->json_is('/0/url', $t->app->url_for('treebank', treebank => $tb->name));
+
+# Set public to false
+$tb->public(Mojo::JSON->false);
+$tb->save();
+
+$t->get_ok($treebanks_url)
+  ->status_is(200);
+
+is(scalar(@{$t->tx->res->json}), 0, 'Returned no treebank');
 
 ok $t->app->routes->find('treebank'), 'Single treebank route exists';
 

diff --git a/t/bootstrap.pl b/t/bootstrap.pl
@@ -121,8 +121,8 @@ sub test_treebank {
     database => 'test',
     username => 'postgres',
     password => '',
-    public => 1,
-    anonaccess => 1,
+    public => Mojo::JSON->true,
+    anonaccess => Mojo::JSON->true,
     data_sources => {
       adata => File::Spec->catdir('pdt20_mini', 'data'),
       tdata => File::Spec->catdir('pdt20_mini', 'data')