| 1 |
<?php |
| 2 |
|
| 3 |
class Test_WPMUDEV_Analytics extends WP_UnitTestCase { |
| 4 |
private $time; |
| 5 |
|
| 6 |
public function set_up() { |
| 7 |
parent::set_up(); |
| 8 |
|
| 9 |
Testable_Mixpanel::$track_count = array(); |
| 10 |
} |
| 11 |
|
| 12 |
/** |
| 13 |
* @dataProvider mysql_version |
| 14 |
*/ |
| 15 |
public function test_event_limit_enforced( $mysql_version ) { |
| 16 |
$time = time(); |
| 17 |
$this->set_time( $time ); |
| 18 |
|
| 19 |
$event_limit = 3; |
| 20 |
$analytics = $this->given_analytics( $event_limit ); |
| 21 |
$analytics->set_mysql_version( $mysql_version ); |
| 22 |
$analytics->track( 'event' ); |
| 23 |
$analytics->track( 'event' ); |
| 24 |
$analytics->track( 'event' ); |
| 25 |
// 3 events tracked |
| 26 |
$this->then_event_tracked_times( 'event', 3 ); |
| 27 |
|
| 28 |
$analytics->track( 'event' ); |
| 29 |
// Event not tracked but exceeded limit event is generated |
| 30 |
$this->then_event_tracked_times( 'event', 3 ); |
| 31 |
$this->then_event_tracked_times( 'exceeded_daily_limit', 1 ); |
| 32 |
|
| 33 |
$analytics->track( 'event' ); |
| 34 |
// Nothing happened |
| 35 |
$this->then_event_tracked_times( 'event', 3 ); |
| 36 |
$this->then_event_tracked_times( 'exceeded_daily_limit', 1 ); |
| 37 |
|
| 38 |
$time_window = HOUR_IN_SECONDS * 24; |
| 39 |
$this->set_time( $time + $time_window ); |
| 40 |
$analytics->track( 'event' ); |
| 41 |
$this->then_event_tracked_times( 'event', 3 ); |
| 42 |
$this->then_event_tracked_times( 'exceeded_daily_limit', 1 ); |
| 43 |
|
| 44 |
$this->set_time( $time + $time_window + 1 ); |
| 45 |
$analytics->track( 'event' ); |
| 46 |
$analytics->track( 'event' ); |
| 47 |
$analytics->track( 'event' ); |
| 48 |
// Time window has passed, 3 more events generated |
| 49 |
$this->then_event_tracked_times( 'event', 6 ); |
| 50 |
$this->then_event_tracked_times( 'exceeded_daily_limit', 1 ); |
| 51 |
|
| 52 |
$analytics->track( 'event' ); |
| 53 |
// Another exceeded event |
| 54 |
$this->then_event_tracked_times( 'event', 6 ); |
| 55 |
$this->then_event_tracked_times( 'exceeded_daily_limit', 2 ); |
| 56 |
} |
| 57 |
|
| 58 |
private function then_event_tracked_times( $event_name, $expected_count ) { |
| 59 |
$this->assertEquals( $expected_count, Testable_Mixpanel::$track_count[ $event_name ] ); |
| 60 |
} |
| 61 |
|
| 62 |
private function set_time( $time ) { |
| 63 |
$this->time = $time; |
| 64 |
} |
| 65 |
|
| 66 |
public function time_function() { |
| 67 |
return $this->time; |
| 68 |
} |
| 69 |
|
| 70 |
private function given_analytics( $event_limit ) { |
| 71 |
$options = array( |
| 72 |
'time_function' => array( $this, 'time_function' ), |
| 73 |
); |
| 74 |
$analytics = new WPMUDEV_Analytics_V4( 'plugin_slug', 'plugin_name', $event_limit, 'project_token', $options ); |
| 75 |
$analytics->set_mixpanel( new Testable_Mixpanel() ); |
| 76 |
return $analytics; |
| 77 |
} |
| 78 |
|
| 79 |
public function mysql_version() { |
| 80 |
return [ |
| 81 |
[ '5.6' ], |
| 82 |
[ '5.7' ], |
| 83 |
]; |
| 84 |
} |
| 85 |
} |
| 86 |
|
| 87 |
class Testable_Mixpanel { |
| 88 |
static $track_count = array(); |
| 89 |
|
| 90 |
public function track( $event_name ) { |
| 91 |
self::$track_count[ $event_name ] ++; |
| 92 |
} |
| 93 |
} |
| 94 |
|
