From ba254a8b962a42bd302a9fa75a232c8750be2e7b Mon Sep 17 00:00:00 2001
From: Adam Wright <adam.austin.wright@gmail.com>
Date: Fri, 8 Jan 2016 03:12:43 -0800
Subject: [PATCH] Moved main into a test in lib.rs

---
 src/macros/src/lib.rs  | 44 ++++++++++++++++++++++++++++++++++++++++++
 src/macros/src/main.rs | 42 ----------------------------------------
 2 files changed, 44 insertions(+), 42 deletions(-)
 delete mode 100644 src/macros/src/main.rs

diff --git a/src/macros/src/lib.rs b/src/macros/src/lib.rs
index 8b6a356..43a7c45 100644
--- a/src/macros/src/lib.rs
+++ b/src/macros/src/lib.rs
@@ -91,3 +91,47 @@ macro_rules! rpc_service {
         }
     }
 }
+
+#[cfg(test)]
+mod test {
+    use std::net::{TcpListener, TcpStream};
+    use std::thread;
+
+    #[derive(PartialEq, Debug, RustcEncodable, RustcDecodable)]
+    pub struct Foo {
+        message: String
+    }
+
+    rpc_service!(my_server:
+        hello(super::Foo) -> super::Foo;
+        add((i32, i32)) -> i32;
+    );
+
+    //use my_server::*;
+
+    impl my_server::Service for () {
+        fn hello(&self, s: Foo) -> Foo {
+            Foo{message: format!("Hello, {}", &s.message)}
+        }
+
+        fn add(&self, (x, y): (i32, i32)) -> i32 {
+            x + y
+        }
+    }
+
+    #[test]
+    fn simple_test() {
+        println!("Starting");
+        let listener = TcpListener::bind("127.0.0.1:9000").unwrap();
+        thread::spawn(|| {
+            let server = my_server::Server::new(());
+            println!("Server running");
+            server.serve(listener);
+        });
+        let mut client = my_server::Client(TcpStream::connect("127.0.0.1:9000").unwrap());
+        assert_eq!(3, client.add((1, 2)).unwrap());
+        let foo = Foo{message: "Adam".into()};
+        let want = Foo{message: format!("Hello, {}", &foo.message)};
+        assert_eq!(want, client.hello(Foo{message: "Adam".into()}).unwrap());
+    }
+}
diff --git a/src/macros/src/main.rs b/src/macros/src/main.rs
deleted file mode 100644
index f2b884d..0000000
--- a/src/macros/src/main.rs
+++ /dev/null
@@ -1,42 +0,0 @@
-#[macro_use]
-extern crate macros;
-extern crate rustc_serialize;
-extern crate byteorder;
-
-use std::net::{TcpListener, TcpStream};
-
-#[derive(Debug, RustcEncodable, RustcDecodable)]
-pub struct Foo {
-    message: String
-}
-
-rpc_service!(my_server:
-    hello(::Foo) -> ::Foo;
-    add((i32, i32)) -> i32;
-);
-
-use my_server::*;
-
-impl Service for () {
-    fn hello(&self, s: Foo) -> Foo {
-        Foo{message: format!("Hello, {}", &s.message)}
-    }
-    
-    fn add(&self, (x, y): (i32, i32)) -> i32 {
-        x + y
-    }
-}
-
-fn main() {
-    println!("Starting");
-    let listener = TcpListener::bind("127.0.0.1:9000").unwrap();
-    std::thread::spawn(|| {
-        let server = Server::new(());
-        println!("Server running");
-        server.serve(listener);
-    });
-    let mut client = Client(TcpStream::connect("127.0.0.1:9000").unwrap());
-    println!("Client running");
-    println!("add((1, 2)) => {}", client.add((1, 2)).unwrap());
-    println!("hello(\"adam\") => {:?}", client.hello(Foo{message: "Adam".into()}));
-}