From 5e06f0a69bbb02ab699df25e5bc1469108decb82 Mon Sep 17 00:00:00 2001 From: Tom Bloor Date: Mon, 5 Jun 2017 22:34:48 +0100 Subject: [PATCH] Added resultset for leaderboard for more consice code --- .../LocalLoop/Schema/ResultSet/Leaderboard.pm | 42 +++ t/schema/leaderboard.t | 2 + t/schema/resultset_leaderboard.t | 279 ++++++++++++++++++ 3 files changed, 323 insertions(+) create mode 100644 lib/Pear/LocalLoop/Schema/ResultSet/Leaderboard.pm create mode 100644 t/schema/resultset_leaderboard.t diff --git a/lib/Pear/LocalLoop/Schema/ResultSet/Leaderboard.pm b/lib/Pear/LocalLoop/Schema/ResultSet/Leaderboard.pm new file mode 100644 index 0000000..d8fd9c6 --- /dev/null +++ b/lib/Pear/LocalLoop/Schema/ResultSet/Leaderboard.pm @@ -0,0 +1,42 @@ +package Pear::LocalLoop::Schema::ResultSet::Leaderboard; + +use strict; +use warnings; + +use base 'DBIx::Class::ResultSet'; + +sub get_latest { + my $self = shift; + my $type = shift; + + my $type_result = $self->find_by_type( $type ); + + return undef unless defined $type_result; + + my $latest = $type_result->search_related('sets', {}, { + order_by => { -desc => 'date' }, + })->first; + + return $latest; +} + +sub create_new { + my $self = shift; + my $type = shift; + my $date = shift; + + my $type_result = $self->find_by_type($type); + + return undef unless $type_result; + + return $type_result->create_new($date); +} + +sub find_by_type { + my $self = shift; + my $type = shift; + + return $self->find({ type => $type }); +} + +1; diff --git a/t/schema/leaderboard.t b/t/schema/leaderboard.t index 13e8a82..e0c3fb8 100644 --- a/t/schema/leaderboard.t +++ b/t/schema/leaderboard.t @@ -270,6 +270,8 @@ subtest 'get_latest' => sub { ]; is_deeply [ $today_values->all ], $expected, 'array as expected'; + + is $leaderboard_rs->find({ type => 'daily_total' })->sets->count, 4, 'correct leaderboard count'; }; done_testing; diff --git a/t/schema/resultset_leaderboard.t b/t/schema/resultset_leaderboard.t new file mode 100644 index 0000000..f14912f --- /dev/null +++ b/t/schema/resultset_leaderboard.t @@ -0,0 +1,279 @@ +use Mojo::Base -strict; + +use Test::More; +use Mojo::JSON; +use Test::Pear::LocalLoop; +use DateTime; + +my $framework = Test::Pear::LocalLoop->new; +my $t = $framework->framework; +my $schema = $t->app->schema; +my $dtf = $schema->storage->datetime_parser; + +my $user1 = { + token => 'a', + full_name => 'Test User1', + display_name => 'Test User1', + email => 'test1@example.com', + postcode => 'LA1 1AA', + password => 'abc123', + age_range => 1, +}; + +my $user2 = { + token => 'b', + full_name => 'Test User2', + display_name => 'Test User2', + email => 'test2@example.com', + postcode => 'LA1 1AA', + password => 'abc123', + age_range => 1, +}; + +my $user3 = { + token => 'c', + full_name => 'Test User3', + display_name => 'Test User3', + email => 'test3@example.com', + postcode => 'LA1 1AA', + password => 'abc123', + age_range => 1, +}; + +my $user4 = { + token => 'd', + full_name => 'Test User4', + display_name => 'Test User4', + email => 'test4@example.com', + postcode => 'LA1 1AA', + password => 'abc123', + age_range => 1, +}; + +my $org = { + token => 'e', + email => 'test5@example.com', + name => 'Test Org', + street_name => 'Test Street', + town => 'Lancaster', + postcode => 'LA1 1AA', + password => 'abc123', +}; + +$schema->resultset('AccountToken')->create({ name => $_->{token} }) + for ( $user1, $user2, $user3, $user4, $org ); + +$framework->register_customer($_) + for ( $user1, $user2, $user3, $user4 ); + +$framework->register_organisation($org); + +my $org_result = $schema->resultset('Organisation')->find({ name => $org->{name} }); + +my $tweak = 0; + +my $now = DateTime->today(); + +for my $user ( $user1, $user2, $user3, $user4 ) { + $tweak ++; + my $user_result = $schema->resultset('User')->find({ email => $user->{email} }); + for ( 1 .. 10 ) { + $user_result->create_related( 'transactions', { + seller_id => $org_result->id, + value => $_ + $tweak, + proof_image => 'a', + }); + } + + for ( 11 .. 20 ) { + $user_result->create_related( 'transactions', { + seller_id => $org_result->id, + value => $_ + $tweak, + proof_image => 'a', + submitted_at => $dtf->format_datetime($now->clone->subtract( days => 5 )), + }); + } + + for ( 21 .. 30 ) { + $user_result->create_related( 'transactions', { + seller_id => $org_result->id, + value => $_ + $tweak, + proof_image => 'a', + submitted_at => $dtf->format_datetime($now->clone->subtract( days => 25 )), + }); + } + + for ( 31 .. 40 ) { + $user_result->create_related( 'transactions', { + seller_id => $org_result->id, + value => $_ + $tweak, + proof_image => 'a', + submitted_at => $dtf->format_datetime($now->clone->subtract( days => 50 )), + }); + } + + is $user_result->transactions->count, 40, 'correct count for user' . $tweak; +} + +sub test_leaderboard { + my ( $title, $name, $date, $expected ) = @_; + + subtest $title => sub { + my $leaderboard_rs = $schema->resultset('Leaderboard'); + + $leaderboard_rs->create_new( $name, $date ); + + my $today_board = $leaderboard_rs->get_latest( $name ); + + is $today_board->values->count, 5, 'correct value count'; + + my $today_values = $today_board->values->search( + {}, + { + order_by => { -desc => 'value' }, + columns => [ qw/ user_id value / ], + }, + ); + $today_values->result_class( 'DBIx::Class::ResultClass::HashRefInflator' ); + + is_deeply [ $today_values->all ], $expected, 'array as expected'; + }; +} + +test_leaderboard( + 'Daily Total', + 'daily_total', + $now, + [ + { user_id => 4, value => 95 }, + { user_id => 3, value => 85 }, + { user_id => 2, value => 75 }, + { user_id => 1, value => 65 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'Daily Count', + 'daily_count', + $now, + [ + { user_id => 1, value => 10 }, + { user_id => 2, value => 10 }, + { user_id => 3, value => 10 }, + { user_id => 4, value => 10 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'Weekly Total', + 'weekly_total', + $now->clone->subtract( days => 7 ), + [ + { user_id => 4, value => 195 }, + { user_id => 3, value => 185 }, + { user_id => 2, value => 175 }, + { user_id => 1, value => 165 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'Weekly Count', + 'weekly_count', + $now->clone->subtract( days => 7 ), + [ + { user_id => 1, value => 10 }, + { user_id => 2, value => 10 }, + { user_id => 3, value => 10 }, + { user_id => 4, value => 10 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'Monthly Total', + 'monthly_total', + $now->clone->subtract( months => 1 ), + [ + { user_id => 4, value => 490 }, + { user_id => 3, value => 470 }, + { user_id => 2, value => 450 }, + { user_id => 1, value => 430 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'Monthly Count', + 'monthly_count', + $now->clone->subtract( months => 1 ), + [ + { user_id => 1, value => 20 }, + { user_id => 2, value => 20 }, + { user_id => 3, value => 20 }, + { user_id => 4, value => 20 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'All Time Total', + 'all_time_total', + $now, + [ + { user_id => 4, value => 980 }, + { user_id => 3, value => 940 }, + { user_id => 2, value => 900 }, + { user_id => 1, value => 860 }, + { user_id => 5, value => 0 }, + ] +); + +test_leaderboard( + 'All Time Count', + 'all_time_count', + $now, + [ + { user_id => 1, value => 40 }, + { user_id => 2, value => 40 }, + { user_id => 3, value => 40 }, + { user_id => 4, value => 40 }, + { user_id => 5, value => 0 }, + ] +); + +subtest 'get_latest' => sub { + my $leaderboard_rs = $schema->resultset('Leaderboard'); + $leaderboard_rs->create_new( 'daily_total', $now->clone->subtract(days => 5)); + $leaderboard_rs->create_new( 'daily_total', $now->clone->subtract(days => 25)); + $leaderboard_rs->create_new( 'daily_total', $now->clone->subtract(days => 50)); + + my $today_board = $leaderboard_rs->get_latest( 'daily_total' ); + + is $today_board->values->count, 5, 'correct value count'; + + my $today_values = $today_board->values->search( + {}, + { + order_by => { -desc => 'value' }, + columns => [ qw/ user_id value / ], + }, + ); + $today_values->result_class( 'DBIx::Class::ResultClass::HashRefInflator' ); + + my $expected = [ + { user_id => 4, value => 95 }, + { user_id => 3, value => 85 }, + { user_id => 2, value => 75 }, + { user_id => 1, value => 65 }, + { user_id => 5, value => 0 }, + ]; + + is_deeply [ $today_values->all ], $expected, 'array as expected'; + + is $leaderboard_rs->find_by_type( 'daily_total' )->sets->count, 4, 'correct leaderboard count'; +}; + +done_testing;