python-patterns/test_borg.py at master · sd2017/python-patterns · GitHub

32 lines (25 loc) · 857 Bytes

#!/usr/bin/env python

from borg import Borg, YourBorg

import sys

if sys.version_info < (2, 7):

import unittest2 as unittest

else:

import unittest

class BorgTest(unittest.TestCase):

@classmethod

def setUpClass(self):

self.b1 = Borg()

self.b2 = Borg()

self.ib1 = YourBorg()

def test_initial_borg_state_shall_be_init(self):

b = Borg()

self.assertEqual(b.state, 'Init')

def test_changing_instance_attribute_shall_change_borg_state(self):

self.b1.state = 'Running'

self.assertEqual(self.b1.state, 'Running')

self.assertEqual(self.b2.state, 'Running')

self.assertEqual(self.ib1.state, 'Running')

def test_instances_shall_have_own_ids(self):

self.assertNotEqual(id(self.b1), id(self.b2), id(self.ib1))

if __name__ == "__main__":

unittest.main()

Expand file tree