t/unit_fixture/CXGN/Trial/AddSubplotsToTrial.t

   1 use strict;
   2
   3 use Test::More;
   4 use lib 't/lib';
   5 use SGN::Test::Fixture;
   6 use JSON::Any;
   7 use Data::Dumper;
   8 use Test::WWW::Mechanize;
   9 use JSON;
  10
  11 use SGN::Model::Cvterm;
  12 use CXGN::Trial::TrialDesign;
  13 use CXGN::Trial::TrialCreate;
  14
  15
  16 my $fix = SGN::Test::Fixture->new();
  17 my $chado_schema = $fix->bcs_schema;
  18 my $metadata_schema = $fix->metadata_schema;
  19 my $phenome_schema = $fix->phenome_schema;
  20 my $dbh = $fix->dbh;
  21
  22 #
  23 # CREATE A TEST TRIAL
  24 #
  25
  26 my $ayt_cvterm_id = SGN::Model::Cvterm->get_cvterm_row($chado_schema, 'Advanced Yield Trial', 'project_type')->cvterm_id();
  27
  28 my @stock_names = ('test_accession1', 'test_accession2', 'test_accession3', 'test_accession4', 'test_accession5');
  29
  30 my $trial_design = CXGN::Trial::TrialDesign->new();
  31 $trial_design->set_trial_name("test_trial");
  32 $trial_design->set_stock_list(\@stock_names);
  33 $trial_design->set_plot_start_number(1);
  34 $trial_design->set_plot_number_increment(1);
  35 $trial_design->set_plot_layout_format("zigzag");
  36 $trial_design->set_number_of_blocks(2);
  37 $trial_design->set_design_type("RCBD");
  38 $trial_design->calculate_design();
  39 ok(
  40     my $design = $trial_design->get_design(),
  41     "create trial design"
  42 );
  43
  44 ok(
  45     my $trial_create = CXGN::Trial::TrialCreate->new({
  46         chado_schema => $chado_schema,
  47         dbh => $dbh,
  48         owner_id => 41,
  49         design => $design,
  50         program => "test",
  51         trial_year => "2015",
  52         trial_description => "test description",
  53         trial_location => "test_location",
  54         trial_name => "subplot_test_trial",
  55         trial_type=>$ayt_cvterm_id,
  56         design_type => "RCBD",
  57         operator => "janedoe"
  58     }),
  59     "create trial object"
  60 );
  61
  62 my $save = $trial_create->save_trial();
  63 ok(my $trial_id = $save->{'trial_id'}, "save trial");
  64
  65
  66 #
  67 # ADD SUBPLOTS TO PLOTS
  68 #
  69
  70 my $trial = CXGN::Trial->new({
  71     bcs_schema => $chado_schema,
  72     metadata_schema => $metadata_schema,
  73     phenome_schema => $phenome_schema,
  74     trial_id => $trial_id
  75 });
  76 $trial->create_subplot_entities('2');
  77
  78 my $tl = CXGN::Trial::TrialLayout->new({ schema => $chado_schema, trial_id => $trial_id, experiment_type => 'field_layout' });
  79 my $d = $tl->get_design();
  80
  81 my @subplot_names = ();
  82 my @subplot_names_flat = ();
  83 foreach my $plot_num (keys %$d) {
  84     push @subplot_names, $d->{$plot_num}->{'subplot_names'}
  85 }
  86 foreach my $subplot_name_arr_ref (@subplot_names) {
  87     foreach (@$subplot_name_arr_ref) {
  88         push @subplot_names_flat, $_;
  89     }
  90 }
  91
  92 #print STDERR Dumper \@subplot_names_flat;
  93 # 5 plots/block * 2 blocks * 2 subplots/plot = 20 subplots
  94 is(scalar(@subplot_names_flat), 20);
  95
  96
  97 #
  98 # ADD PLANTS TO SUBPLOTS
  99 #
 100
 101 $trial->create_plant_subplot_entities(4);
 102 $tl = CXGN::Trial::TrialLayout->new({ schema => $chado_schema, trial_id => $trial_id, experiment_type => 'field_layout' });
 103 $d = $tl->get_design();
 104
 105 my @plant_names = ();
 106 my @plant_names_flat = ();
 107 foreach my $plot_num (keys %$d) {
 108     push @plant_names, $d->{$plot_num}->{'plant_names'}
 109 }
 110 foreach my $plant_name_arr_ref (@plant_names) {
 111     foreach (@$plant_name_arr_ref) {
 112         push @plant_names_flat, $_;
 113     }
 114 }
 115
 116 # print STDERR Dumper \@plant_names_flat;
 117 # 5 plots/block * 2 blocks * 2 subplots/plot * 4 plants/subplot = 80 plants
 118 is(scalar(@plant_names_flat), 80);
 119
 120
 121 #
 122 # Remove Trial
 123 #
 124 $trial->delete_metadata();
 125 $trial->delete_field_layout();
 126 $trial->delete_project_entry();
 127
 128
 129 done_testing();