1 | <?php |
||
22 | class AffiliateTrackingTest extends TestCase |
||
23 | { |
||
24 | public function testAffiliateLink() |
||
25 | { |
||
26 | $user = UserStub::find(1); |
||
27 | $url = config('app.url'); |
||
28 | |||
29 | $this->assertEquals($url.'/?ref='.$user->affiliate_id, $user->getAffiliateLink()); |
||
30 | $this->assertEquals($url.'/register/?ref='.$user->affiliate_id, $user->getAffiliateLink('/register')); |
||
31 | } |
||
32 | |||
33 | public function testGetAffiliateIdFromCookieAndWithoutCookie() |
||
34 | { |
||
35 | $user = UserStub::find(1); |
||
36 | $request = m::mock(Request::class); |
||
37 | $cookieName = config('referral.ref_cookie'); |
||
38 | |||
39 | $request->shouldReceive('query')->once()->andReturn($user->affiliate_id); |
||
40 | $this->mockingCookieFacade($request, $cookieName, null); |
||
41 | |||
42 | $middleware = new AffiliateTracking(); |
||
43 | $response = $middleware->handle($request, function ($request) { |
||
44 | return response(''); |
||
45 | }); |
||
46 | |||
47 | $cookie = $response->headers->getCookies()[0]; |
||
48 | |||
49 | $this->assertEquals(200, $response->getStatusCode()); |
||
50 | $this->assertEquals($cookieName, $cookie->getName()); |
||
51 | $this->assertEquals($user->affiliate_id, $cookie->getValue()); |
||
52 | } |
||
53 | |||
54 | public function testGetAffiliateIdFromCookieAndWithCookie() |
||
55 | { |
||
56 | $user = UserStub::find(1); |
||
57 | $request = m::mock(Request::class); |
||
58 | $cookieName = config('referral.ref_cookie'); |
||
59 | |||
60 | $request->shouldReceive('query')->once()->andReturn($user->affiliate_id); |
||
61 | $this->mockingCookieFacade($request, $cookieName, $user->affiliate_id); |
||
62 | |||
63 | $middleware = new AffiliateTracking(); |
||
64 | $response = $middleware->handle($request, function ($request) { |
||
65 | return response(''); |
||
66 | }); |
||
67 | |||
68 | $cookie = $response->headers->getCookies(); |
||
69 | |||
70 | $this->assertEquals(200, $response->getStatusCode()); |
||
71 | $this->assertEmpty($cookie); |
||
72 | } |
||
73 | |||
74 | public function testCreateReferral() |
||
75 | { |
||
76 | $user = UserStub::find(1); |
||
77 | $cookieName = config('referral.ref_cookie'); |
||
78 | $request = m::mock(Request::class); |
||
79 | $this->mockingCookieFacade($request, $cookieName, $user->affiliate_id); |
||
80 | |||
81 | $referral = UserStub::create([ |
||
82 | 'name' => 'cat', |
||
83 | 'email' => '[email protected]', |
||
84 | 'password' => bcrypt('secret'), |
||
85 | ]); |
||
86 | |||
87 | $result = Referral::whereReferrerId($user->id)->first(); |
||
88 | |||
89 | $this->assertEquals($referral->id, $result->referral_id); |
||
90 | $this->assertEquals($user->id, $result->referrer_id); |
||
91 | } |
||
92 | |||
93 | /** |
||
94 | * @param Request $request |
||
95 | * @param $cookieName |
||
96 | * @param $value |
||
97 | * @param null $default |
||
98 | */ |
||
99 | protected function mockingCookieFacade(Request $request, $cookieName, $value, $default = null) |
||
100 | { |
||
101 | $request->shouldReceive('cookie') |
||
102 | ->times(1) |
||
103 | ->withArgs([$cookieName, $default]) |
||
104 | ->andReturn($value); |
||
105 | $app = new ApplicationStub(); |
||
106 | $app->setAttributes(['request' => $request]); |
||
107 | Cookie::setFacadeApplication($app); |
||
108 | } |
||
109 | } |
||
110 |