def setUp(self): self.test1 = TestModel() self.test1.save() self.logger = DBLogHandler() (self.user, is_new) = User.objects.get_or_create(username="******") self.user.save() self.user_request = HttpRequest() self.user_request.user = self.user self.user_request.META = {'header1': 'test1', 'header2': 'test2'} self.anon = AnonymousUser() self.anon_request = HttpRequest() self.anon_request.user = self.anon self.anon_request.META = {'header1': 'test1', 'header2': 'test2'}
class DBLogHandlerTest(unittest.TestCase): def setUp(self): self.test1 = TestModel() self.test1.save() self.logger = DBLogHandler() (self.user, is_new) = User.objects.get_or_create(username="******") self.user.save() self.user_request = HttpRequest() self.user_request.user = self.user self.user_request.META = {'header1': 'test1', 'header2': 'test2'} self.anon = AnonymousUser() self.anon_request = HttpRequest() self.anon_request.user = self.anon self.anon_request.META = {'header1': 'test1', 'header2': 'test2'} def test_log_object(self): self.assertEqual(LogObject.objects.count(), 0) self.assertEqual(LogUrl.objects.count(), 0) self.logger.log_object(self.test1, True, self.user_request, data1=1, data2=2, data3=3) self.assertEqual(LogObject.objects.count(), 1) self.logger.log_object(self.test1, True, self.anon_request, data1=1, data2=2, data3=3) self.assertEqual(LogObject.objects.count(), 2) self.assertEqual(LogObject.objects.count(), 2) self.assertEqual(LogUrl.objects.count(), 0) def test_log_url(self): self.assertEqual(LogObject.objects.count(), 0) self.assertEqual(LogUrl.objects.count(), 0) self.logger.log_url('/visit/%d/' % (self.test1.pk), True, self.user_request, data1=1, data2=2, data3=3) self.assertEqual(LogUrl.objects.count(), 1) self.logger.log_url('/visit/%d/' % (self.test1.pk), True, self.anon_request, data1=1, data2=2, data3=3) self.assertEqual(LogUrl.objects.count(), 2) self.assertEqual(LogObject.objects.count(), 0) self.assertEqual(LogUrl.objects.count(), 2) def tearDown(self): LogObject.objects.all().delete() LogUrl.objects.all().delete()