def test_secret_created_stored(self, _uuid4, _path): _path.exists.return_value = False _uuid4.return_value = "secret_thing" with patch_open() as (_open, _file): self.assertEquals(quantum_contexts.get_shared_secret(), "secret_thing") _open.assert_called_with(quantum_contexts.SHARED_SECRET.format("quantum"), "w") _file.write.assert_called_with("secret_thing")
def test_secret_retrieved(self, _path): _path.exists.return_value = True with patch_open() as (_open, _file): _file.read.return_value = 'secret_thing\n' self.assertEquals(quantum_contexts.get_shared_secret(), 'secret_thing') _open.assert_called_with( quantum_contexts.SHARED_SECRET.format('quantum'), 'r')