Commit ff4d392

Lorak-mmkdkropachev

authored and

committed

execute_count: Use plain assert

1 parent 4a0b793 commit ff4d392Copy full SHA for ff4d392

File tree

-8

lines changed

-8

lines changed

Lines changed: 1 addition & 8 deletions

Original file line number	Diff line number	Diff line change
`@@ -77,12 +77,7 @@ def wrapped_function(args, *kwargs):`
`77`	`77`	`# DeMonkey Patch our code`
`78`	`78`	`cassandra.cqlengine.connection.execute = original_function`
`79`	`79`	`# Check to see if we have a pre-existing test case to work from.`
`80`		`- if args:`
`81`		`- test_case = args[0]`
`82`		`- else:`
`83`		`- test_case = unittest.TestCase("__init__")`
`84`		`- # Check to see if the count is what you expect`
`85`		`- test_case.assertEqual(count.get_counter(), expected, msg="Expected number of cassandra.cqlengine.connection.execute calls ({0}) doesn't match actual number invoked ({1})".format(expected, count.get_counter()))`
	`80`	`+ assert count.get_counter() == expected, "Expected number of cassandra.cqlengine.connection.execute calls ({0}) doesn't match actual number invoked ({1})".format(expected, count.get_counter())`
`86`	`81`	`return to_return`
`87`	`82`	`# Name of the wrapped function must match the original or unittest will error out.`
`88`	`83`	`wrapped_function.__name__ = fn.__name__`
`@@ -94,5 +89,3 @@ def wrapped_function(args, *kwargs):`
`94`	`89`	`return wrapped_function`
`95`	`90`
`96`	`91`	`return innerCounter`
`97`		`-`
`98`		`-`

Comments

(0)