use hdk::prelude::*;
pub fn mock_create(
mock_hdk: &mut MockHdkT,
expected_input: CreateInput,
expected_output: ExternResult<HeaderHash>,
) {
mock_hdk
.expect_create()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_create_link(
mock_hdk: &mut MockHdkT,
expected_input: CreateLinkInput,
expected_output: ExternResult<HeaderHash>,
) {
mock_hdk
.expect_create_link()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_agent_info(mock_hdk: &mut MockHdkT, expected_output: ExternResult<AgentInfo>) {
mock_hdk
.expect_agent_info()
.times(1)
.return_const(expected_output);
}
pub fn mock_remote_signal(
mock_hdk: &mut MockHdkT,
expected_input: RemoteSignal,
expected_output: ExternResult<()>,
) {
mock_hdk
.expect_remote_signal()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_update(
mock_hdk: &mut MockHdkT,
expected_input: UpdateInput,
expected_output: ExternResult<HeaderHash>,
) {
mock_hdk
.expect_update()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_get_links(
mock_hdk: &mut MockHdkT,
expected_input: Vec<GetLinksInput>,
expected_output: ExternResult<Vec<Vec<Link>>>,
) {
mock_hdk
.expect_get_links()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_sys_time(mock_hdk: &mut MockHdkT, expected_output: ExternResult<Timestamp>) {
mock_hdk
.expect_sys_time()
.times(1)
.return_const(expected_output);
}
pub fn mock_must_get_header(
mock_hdk: &mut MockHdkT,
expected_input: MustGetHeaderInput,
expected_output: ExternResult<SignedHeaderHashed>,
) {
mock_hdk
.expect_must_get_header()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_must_get_entry(
mock_hdk: &mut MockHdkT,
expected_input: MustGetEntryInput,
expected_output: ExternResult<EntryHashed>,
) {
mock_hdk
.expect_must_get_entry()
.with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}
pub fn mock_zome_info(mock_hdk: &mut MockHdkT, expected_output: ExternResult<ZomeInfo>) {
mock_hdk
.expect_zome_info()
.times(1)
.return_const(expected_output);
}
pub fn mock_get(
mock_hdk: &mut MockHdkT,
expected_input: Vec<GetInput>,
expected_output: ExternResult<Vec<Option<Element>>>,
) {
mock_hdk
.expect_get() .with(mockall::predicate::eq(expected_input))
.times(1)
.return_const(expected_output);
}