Adding unit test

[#79483828]