Update tests

[johnfercher]

Description

Related Issue

Checklist

check with "x", ONLY IF APPLIED to your change

• All methods associated with structs has func (<first letter of struct> *struct) method() {} name style.
• Wrote unit tests for new/changed features.
• Followed the unit test when,should naming pattern.
• All mocks created with m := mocks.NewConstructor(t).
• All mocks using m.EXPECT().MethodName() method to mock methods.
• Updated docs/doc.go and docs/*
• Updated example_test.go
• Updated README.md
• Executed make examples to update all examples inside docs/examples.
• New public methods/structs/interfaces has comments upside them explaining they responsibilities
• Executed make dod with none issues pointed out by golangci-lint

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 82.12%. Comparing base (a65b707) to head (7afa947).
Report is 8 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #416      +/-   ##
==========================================
+ Coverage   82.08%   82.12%   +0.04%     
==========================================
  Files          61       61              
  Lines        2059     2063       +4     
==========================================
+ Hits         1690     1694       +4     
  Misses        341      341              
  Partials       28       28              

Flag	Coverage Δ
unittests	82.12% <ø> (+0.04%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.