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 分钟')
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'])
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)
