mygpo/users/tests.py

   1 #
   2 # This file is part of my.gpodder.org.
   3 #
   4 # my.gpodder.org is free software: you can redistribute it and/or modify it
   5 # under the terms of the GNU Affero General Public License as published by
   6 # the Free Software Foundation, either version 3 of the License, or (at your
   7 # option) any later version.
   8 #
   9 # my.gpodder.org is distributed in the hope that it will be useful, but
  10 # WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  11 # or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public
  12 # License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with my.gpodder.org. If not, see <http://www.gnu.org/licenses/>.
  16 #
  17
  18 import unittest
  19 import doctest
  20
  21 from django.test import TestCase
  22
  23 import mygpo.utils
  24 from mygpo.users.models import User, Device
  25
  26
  27 class DeviceSyncTests(unittest.TestCase):
  28
  29     def setUp(self):
  30         self.user = User(username='test')
  31         self.user.email = 'test@invalid.com'
  32         self.user.set_password('secret!')
  33         self.user.save()
  34
  35
  36     def test_group(self):
  37         dev1 = Device(uid='d1')
  38         self.user.devices.append(dev1)
  39
  40         dev2 = Device(uid='d2')
  41         self.user.devices.append(dev2)
  42
  43         group = self.user.get_grouped_devices().next()
  44         self.assertEquals(group.is_synced, False)
  45         self.assertIn(dev1, group.devices)
  46         self.assertIn(dev2, group.devices)
  47
  48
  49         dev3 = Device(uid='d3')
  50         self.user.devices.append(dev3)
  51
  52         self.user.sync_devices(dev1, dev3)
  53
  54         groups = self.user.get_grouped_devices()
  55         g1 = groups.next()
  56
  57         self.assertEquals(g1.is_synced, True)
  58         self.assertIn(dev1, g1.devices)
  59         self.assertIn(dev3, g1.devices)
  60
  61         g2 = groups.next()
  62         self.assertEquals(g2.is_synced, False)
  63         self.assertIn(dev2, g2.devices)
  64
  65
  66         targets = self.user.get_sync_targets(dev1)
  67         target = targets.next()
  68         self.assertEquals(target, dev2)
  69
  70
  71 def suite():
  72     suite = unittest.TestSuite()
  73     suite.addTest(unittest.TestLoader().loadTestsFromTestCase(DeviceSyncTests))
  74     return suite