from secrets import token_urlsafe from monero.seed import Seed from django.test import TestCase from django.contrib.auth.models import User from django.core.paginator import Page from django.shortcuts import reverse from django.test.client import Client from items.models import Item from bids.models import ItemBid class ItemBidViewsTestCase(TestCase): def setUp(self): self.client = Client() self.seller_password = token_urlsafe(32) self.buyer_password = token_urlsafe(32) self.seller = User.objects.create_user( 'seller', password=self.seller_password ) self.buyer = User.objects.create_user( 'buyer', password=self.buyer_password ) self.payout_address = Seed().public_address() self.return_address = Seed().public_address() self.whereabouts = 'Los Angeles, CA' self.test_item = Item.objects.create( owner=self.seller, name='Test Item', description='Test item', ask_price_xmr=0.3, payout_address=self.payout_address, whereabouts=self.whereabouts ) def test_list_bids_requires_auth(self): response = self.client.get(reverse('list_bids')) self.assertTrue(response.url.startswith(reverse('login'))) self.assertEqual(response.status_code, 302) def test_list_bids_returns_pagination(self): ItemBid.objects.create( item=self.test_item, bidder=self.buyer, bid_price_xmr=0.1, return_address=self.return_address ) self.client.login(username=self.buyer.username, password=self.buyer_password) response = self.client.get(reverse('list_bids')) response_w_str_arg = self.client.get(reverse('list_bids') + "?page=bar") response_w_empty_pg = self.client.get(reverse('list_bids') + "?page=9001") self.client.logout() self.assertTrue(isinstance(response.context['bids'], Page), 'Paginated object not returned') self.assertTrue(isinstance(response_w_str_arg.context['bids'], Page), 'Paginated object not returned') self.assertTrue(isinstance(response_w_empty_pg.context['bids'], Page), 'Paginated object not returned') def test_list_bids_returns_only_user_bids(self): for i in range(1, 20): u = User.objects.create_user(f'list_bids_test{i}', password=token_urlsafe(16)) ItemBid.objects.create( item=self.test_item, bidder=u, bid_price_xmr=0.2, return_address=self.return_address ) ItemBid.objects.create( item=self.test_item, bidder=self.buyer, bid_price_xmr=0.1, return_address=self.return_address ) self.client.login(username=self.buyer.username, password=self.buyer_password) response = self.client.get(reverse('list_bids')) self.client.logout() # Test that buyer's bids are the only bids returned from view all_bids = ItemBid.objects.all() user_bids = all_bids.filter(bidder=self.buyer) self.assertEqual(len(user_bids), len(response.context['bids'])) self.assertLess(len(user_bids), len(all_bids)) # Test that each bid belongs to the buyer for bid in response.context['bids']: self.assertEqual(bid.bidder, self.buyer) def test_create_bid_requires_auth(self): response = self.client.get(reverse('create_bid', args=[self.test_item.id])) self.assertTrue(response.url.startswith(reverse('login'))) self.assertEqual(response.status_code, 302) def test_create_bid_redirects_home_if_item_id_missing(self): self.client.login(username=self.buyer.username, password=self.buyer_password) response = self.client.get(reverse('create_bid', args=[9999])) self.client.logout() self.assertTrue(response.url.startswith(reverse('home'))) self.assertEqual(response.status_code, 302) def test_create_bid_redirects_to_edit_if_bid_already_posted(self): existing_bid = ItemBid.objects.create( item=self.test_item, bidder=self.buyer, bid_price_xmr=0.2, return_address=self.return_address ) self.client.login(username=self.buyer.username, password=self.buyer_password) response = self.client.get(reverse('create_bid', args=[self.test_item.id])) self.client.logout() self.assertTrue(response.url.startswith(reverse('edit_bid', args=[existing_bid.id]))) self.assertEqual(response.status_code, 302)