36 lines
793 B
Python
36 lines
793 B
Python
|
import unittest
|
||
|
|
|
||
|
|
from ebermergen.models.auto_id import AutoID
|
||
|
|
|
||
|
|
|
||
|
|
class TestAutoID(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
AutoID.reset_id()
|
||
|
|
|
||
|
|
def test_first_instance(self):
|
||
|
|
a = AutoID()
|
||
|
|
self.assertEqual(a.id, 1)
|
||
|
|
|
||
|
|
def test_increment(self):
|
||
|
|
a, b = AutoID(), AutoID()
|
||
|
|
self.assertEqual(a.id, 1)
|
||
|
|
self.assertEqual(b.id, 2)
|
||
|
|
|
||
|
|
def test_inheritance(self):
|
||
|
|
class ClassA(AutoID):
|
||
|
|
pass
|
||
|
|
|
||
|
|
class ClassB(AutoID):
|
||
|
|
pass
|
||
|
|
|
||
|
|
a = ClassA()
|
||
|
|
b = ClassB()
|
||
|
|
self.assertEqual(a.id, 1)
|
||
|
|
self.assertEqual(b.id, 1)
|
||
|
|
a2 = ClassA()
|
||
|
|
b2 = ClassB()
|
||
|
|
self.assertEqual(a.id, 1)
|
||
|
|
self.assertEqual(b.id, 1)
|
||
|
|
self.assertEqual(a2.id, 2)
|
||
|
|
self.assertEqual(b2.id, 2)
|