FilesExpand file tree

 master

/

test_pool.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (45 loc) · 1.86 KB

 master

/

test_pool.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (45 loc) · 1.86 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

#!/usr/bin/env python

# -*- coding: utf-8 -*-

import unittest

try:

import queue

except ImportError: # python 2.x compatibility

import Queue as queue

from creational.pool import ObjectPool

class TestPool(unittest.TestCase):

def setUp(self):

self.sample_queue = queue.Queue()

self.sample_queue.put('first')

self.sample_queue.put('second')

def test_items_recoil(self):

with ObjectPool(self.sample_queue, True) as pool:

self.assertEqual(pool, 'first')

self.assertTrue(self.sample_queue.get() == 'second')

self.assertFalse(self.sample_queue.empty())

self.assertTrue(self.sample_queue.get() == 'first')

self.assertTrue(self.sample_queue.empty())

def test_frozen_pool(self):

with ObjectPool(self.sample_queue) as pool:

self.assertEqual(pool, 'first')

self.assertEqual(pool, 'first')

self.assertTrue(self.sample_queue.get() == 'second')

self.assertFalse(self.sample_queue.empty())

self.assertTrue(self.sample_queue.get() == 'first')

self.assertTrue(self.sample_queue.empty())

class TestNaitivePool(unittest.TestCase):

"""def test_object(queue):

queue_object = QueueObject(queue, True)

print('Inside func: {}'.format(queue_object.object))"""

def test_pool_behavior_with_single_object_inside(self):

sample_queue = queue.Queue()

sample_queue.put('yam')

with ObjectPool(sample_queue) as obj:

# print('Inside with: {}'.format(obj))

self.assertEqual(obj, 'yam')

self.assertFalse(sample_queue.empty())

self.assertTrue(sample_queue.get() == 'yam')

self.assertTrue(sample_queue.empty())

# sample_queue.put('sam')

# test_object(sample_queue)

# print('Outside func: {}'.format(sample_queue.get()))

# if not sample_queue.empty():