class RemindViewTestCase(TestCase): def setUp(self): self.r = Remind(time=timezone.now(), owner_id='miao', event='吃饭', desc='吃饭饭') self.r.save() def test_anonymous_delete(self): resp = self.client.delete(reverse('remind-detail', args=(self.r.pk.hex,)), HTTP_X_REFERER='abc.com/#def') # self.assertEqual(resp.status_code, 302) self.assertEqual(resp.status_code, 401) self.assertIn('//open.weixin.qq.com/connect/oauth2/authorize', resp.get('WWW-Authenticate', '')) self.assertIn('state=abc.com%2F%23def', resp.get('WWW-Authenticate', '')) def test_unauthorized_delete(self): u = WechatUser(openid='123', nickname='456') u.save() self.client.force_login(u) resp = self.client.delete(reverse('remind-detail', args=(self.r.pk.hex,))) self.assertEqual(resp.status_code, 403) def test_participant_delete(self): WechatUser(openid='miao', nickname='miaomiaomiao').save() self.r.add_participant('123') u = WechatUser(openid='123', nickname='456') u.save() self.client.force_login(u) resp = self.client.delete(reverse('remind-detail', args=(self.r.pk.hex,))) self.assertEqual(resp.status_code, 204) # self.assertEqual(resp.url, reverse('remind_list')) def test_delete(self): u = WechatUser(openid='miao', nickname='456') u.save() self.client.force_login(u) resp = self.client.delete(reverse('remind-detail', args=(self.r.pk.hex,))) self.assertEqual(resp.status_code, 204)
def test_add_add_participant(self): r = Remind(time=timezone.now(), owner_id='miao', event='吃饭', desc='吃饭饭') r.save() r.add_participant('abc') r.add_participant('abc') self.assertEqual(r.participants, ['abc'])
def test_default_title(self): r = Remind(time=timezone.now(), owner_id='miao', desc='吃饭饭', done=True) r.save() serializer = RemindSerializer(instance=r, context={'request': self.request}) with HTTMock(access_token_mock, create_qrcode_mock): self.assertEqual(serializer.data['title'], Remind.default_title)
def test_click_event_for_remind_today(self): req_text = """ <xml> <ToUserName><![CDATA[toUser]]></ToUserName> <FromUserName><![CDATA[FromUser]]></FromUserName> <CreateTime>123456789</CreateTime> <MsgType><![CDATA[event]]></MsgType> <Event><![CDATA[CLICK]]></Event> <EventKey><![CDATA[time_remind_today]]></EventKey> </xml> """ wechat_msg = self.build_wechat_msg(req_text) resp_xml = handle_message(wechat_msg) self.assertIn('今天没有提醒', resp_xml) WechatUser(openid='abc', nickname='abc').save() r = Remind(time=timezone.now(), owner_id=self.user.pk, event='睡觉') r.save() resp_xml = handle_message(wechat_msg) self.assertIn(r.title(), resp_xml) self.assertIn(r.local_time_string('G:i'), resp_xml) r = Remind(time=timezone.now(), owner_id=self.user.pk, event='吃饭', participants=['abc']) r.save() self.assertEqual( WechatUser.objects.get(pk='abc').get_time_reminds().first(), r)
def test_notify_time_update(self): n = timezone.now() r = Remind(time=n, owner_id='miao', event='吃饭', desc='吃饭饭') r.save() self.assertEqual(r.notify_time, n) r.defer = -10 r.save() self.assertEqual(r.notify_time, n - timedelta(minutes=10)) self.assertEqual(r.nature_time_defer(), '提前 10 分钟')
def test_notify_time_update(self): n = timezone.now() r = Remind(time=n, owner_id='miao', event='吃饭', desc='吃饭饭') r.save() self.assertEqual(r.notify_time, n) r.defer = -10 r.save() self.assertEqual(r.notify_time, n-timedelta(minutes=10)) self.assertEqual(r.humanize_defer(), '提前 10 分钟')
class RemindSerializerTestCase(TestCase): def setUp(self): self.r = Remind(time=timezone.now(), owner_id='miao', event='吃饭', desc='吃饭饭', done=True) self.r.save() def test_change_defer(self): update_data = { 'title': self.r.event, 'time': TimestampField().to_representation(self.r.time), 'defer': -2 * 60 } self.assertTrue(self.r.done) serializer = RemindSerializer(data=update_data, instance=self.r) self.assertTrue(serializer.is_valid()) serializer.save() self.assertFalse(self.r.done) self.assertEqual(self.r.defer, -2 * 60) self.assertEqual(self.r.notify_time, self.r.time + timedelta(minutes=self.r.defer)) update_data['defer'] = 2 * 24 * 60 serializer = RemindSerializer(data=update_data, instance=self.r) self.assertTrue(serializer.is_valid()) serializer.save() self.assertEqual(self.r.defer, 2 * 24 * 60) def test_uuid_format(self): serializer = RemindSerializer(instance=self.r) self.assertRegexpMatches(serializer.data['id'], r'\w{32}') def test_default_title(self): r = Remind(time=timezone.now(), owner_id='miao', desc='吃饭饭', done=True) r.save() serializer = RemindSerializer(instance=r) self.assertEqual(serializer.data['title'], Remind.default_title) def test_read_only_fields(self): update_data = { 'id': '123', 'owner': { 'id': '123' }, 'title': self.r.event, 'time': TimestampField().to_representation(self.r.time), 'aaa': 1 } serializer = RemindSerializer(data=update_data, initial=self.r) self.assertTrue(serializer.is_valid()) self.assertNotIn('id', serializer.validated_data) self.assertNotIn('owner', serializer.validated_data) self.assertNotIn('aaa', serializer.validated_data)
def test_reschedule(self): n = timezone.now() - timedelta(minutes=10) r = Remind(time=n, owner_id='miao', event='吃饭', desc='吃饭饭', repeat={REPEAT_KEY_DAY: 1}, defer=-60) r.save() self.assertEqual(r.notify_time, n - timedelta(minutes=60) + timedelta(days=1))
def test_add_add_participant(self): WechatUser(openid='abc', nickname='abcabc', subscribe=True).save() r = Remind(time=timezone.now(), owner_id='miao', event='吃饭', desc='吃饭饭') r.save() r.add_participant('abc') r.add_participant('abc') self.assertEqual(r.participants, ['abc'])
class RemindFormTestCase(TestCase): def setUp(self): self.r = Remind(time=timezone.now(), owner_id="miao", event="吃饭", desc="吃饭饭", done=True) self.r.save() def test_change_defer(self): form_data = { "event": self.r.event, "time": self.r.time.strftime("%Y-%m-%dT%H:%M"), "defer": "提前 2 小时", "desc": self.r.desc, } self.assertTrue(self.r.done) form = RemindForm(data=form_data, instance=self.r) form.save() self.assertTrue(form.is_valid()) self.assertFalse(self.r.done) self.assertEqual(self.r.defer, -2 * 60) self.assertEqual(self.r.notify_time, self.r.time + timedelta(minutes=self.r.defer)) form_data["defer"] = "延后 2 天" form = RemindForm(data=form_data, instance=self.r) form.save() self.assertEqual(self.r.defer, 2 * 24 * 60)
def test_click_event_for_remind_today(self): req_text = """ <xml> <ToUserName><![CDATA[toUser]]></ToUserName> <FromUserName><![CDATA[FromUser]]></FromUserName> <CreateTime>123456789</CreateTime> <MsgType><![CDATA[event]]></MsgType> <Event><![CDATA[CLICK]]></Event> <EventKey><![CDATA[time_remind_today]]></EventKey> </xml> """ wechat_msg = self.build_wechat_msg(req_text) resp_xml = handle_message(wechat_msg) self.assertIn('今天没有提醒', resp_xml) WechatUser(openid='abc', nickname='abc').save() r = Remind(time=timezone.now(), owner_id=self.user.pk, event='睡觉') r.save() resp_xml = handle_message(wechat_msg) self.assertIn(r.title(), resp_xml) self.assertIn(r.local_time_string('G:i'), resp_xml) r = Remind(time=timezone.now(), owner_id=self.user.pk, event='吃饭', participants=['abc']) r.save() self.assertEqual(WechatUser.objects.get(pk='abc').get_time_reminds().first(), r)
class RemindSerializerTestCase(TestCase): def setUp(self): self.request = RequestFactory().get('/') user = get_user_model()(openid='miao', nickname='abc') user.save() self.request.user = user self.r = Remind(time=timezone.now(), owner_id='miao', event='吃饭', desc='吃饭饭', done=True) self.r.save() def test_change_defer(self): update_data = { 'title': self.r.event, 'time': TimestampField().to_representation(self.r.time), 'defer': -2 * 60 } self.assertTrue(self.r.done) serializer = RemindSerializer(data=update_data, instance=self.r, partial=True) self.assertTrue(serializer.is_valid()) serializer.save() self.assertFalse(self.r.done) self.assertEqual(self.r.defer, -2 * 60) self.assertEqual(self.r.notify_time, self.r.time + timedelta(minutes=self.r.defer)) update_data['defer'] = 2 * 24 * 60 serializer = RemindSerializer(data=update_data, instance=self.r, partial=True) self.assertTrue(serializer.is_valid()) serializer.save() self.assertEqual(self.r.defer, 2 * 24 * 60) def test_uuid_format(self): serializer = RemindSerializer(instance=self.r, context={'request': self.request}) with HTTMock(access_token_mock, create_qrcode_mock): self.assertRegexpMatches(serializer.data['id'], r'\w{32}') def test_default_title(self): r = Remind(time=timezone.now(), owner_id='miao', desc='吃饭饭', done=True) r.save() serializer = RemindSerializer(instance=r, context={'request': self.request}) with HTTMock(access_token_mock, create_qrcode_mock): self.assertEqual(serializer.data['title'], Remind.default_title) def test_read_only_fields(self): update_data = { 'id': '123', 'owner': { 'id': 'miao' }, 'title': self.r.event, 'time': TimestampField().to_representation(self.r.time), 'aaa': 1 } serializer = RemindSerializer(data=update_data, initial=self.r, partial=True) self.assertTrue(serializer.is_valid()) self.assertNotIn('id', serializer.validated_data) self.assertNotIn('owner', serializer.validated_data) self.assertNotIn('aaa', serializer.validated_data)
def test_default_title(self): r = Remind(time=timezone.now(), owner_id='miao', desc='吃饭饭', done=True) r.save() serializer = RemindSerializer(instance=r) self.assertEqual(serializer.data['title'], Remind.default_title)