1 files changed, 14 insertions, 5 deletions
diff --git a/src/lib.rs b/src/lib.rs
index 9d7204a..27e26f0 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -2,18 +2,27 @@
 extern crate rand;
 extern crate curve25519_dalek;
 extern crate sha2;
+extern crate core;
 
-pub mod spake2;
-//use spake2::*;
+mod spake2;
+pub use spake2::*;
 
 #[cfg(test)]
 mod tests {
-    use spake2;
+    use spake2::{SPAKE2, Ed25519Group};
+
     #[test]
-    fn test_foo() {
-        assert_eq!(spake2::foo(), 1);
+    fn test_one() {
+        let (s1, msg1) = SPAKE2::<Ed25519Group>::new(b"password",
+                                                     b"idA", b"idB");
+        let (s2, msg2) = SPAKE2::<Ed25519Group>::new(b"password",
+                                                     b"idA", b"idB");
+        let key1 = s1.finish(msg2.as_slice()).unwrap();
+        let key2 = s2.finish(msg1.as_slice()).unwrap();
+        assert_eq!(key1, key2);
     }
 
+
     #[test]
     fn it_works() {
     }