Add test for dialog

Add test for dialog:

- [ ] test coverage
- [ ] finish the code
- [ ] gather feedback for my changes


| Q             | A
| ------------- | ---
| Bug fix?      | no
| New feature?  | yes
| Tests pass?   | yes
| Fixed tickets | #22 
| Doc           | none