I have made some improvements in the patch. The motive was to write tests for...

I have made some improvements in the patch. The motive was to write tests for the changes I have made.

I am using this module in my project, and we are using drupal-test-traits framework for tests. I haven't written tests using Drupal testing framework for a long time. Due to time constraints I was not able to convert the tests.