def test_single_block(self):
pg_progress = PgProgress(10, 20, 10)
self.assertTrue(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 20)
with self.assertRaises(AssertionError):
pg_progress.advance_goal()
def test_single_block(self):
pg_progress = PgProgress(10, 20, 10)
self.assertTrue(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 20)
with self.assertRaises(AssertionError):
pg_progress.advance_goal()
def test_multiple_blocks(self):
pg_progress = PgProgress(10, 30, 10)
self.assertFalse(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 20)
self.assertEqual(pg_progress.get_status(), "Waiting for PG creation (0/20), currently creating PGs 10-20")
pg_progress.set_created_pg_count(20)
pg_progress.advance_goal()
self.assertTrue(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 30)
self.assertEqual(pg_progress.get_status(), "Waiting for PG creation (10/20)")
with self.assertRaises(AssertionError):
pg_progress.advance_goal()
def test_multiple_blocks(self):
pg_progress = PgProgress(10, 30, 10)
self.assertFalse(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 20)
self.assertEqual(
pg_progress.get_status(),
"Waiting for PG creation (0/20), currently creating PGs 10-20")
pg_progress.set_created_pg_count(20)
pg_progress.advance_goal()
self.assertTrue(pg_progress.is_final_block())
self.assertEqual(pg_progress.goal, 30)
self.assertEqual(pg_progress.get_status(),
"Waiting for PG creation (10/20)")
with self.assertRaises(AssertionError):
pg_progress.advance_goal()
