team join fixed + test

2 files changed, 55 insertions, 1 deletions

diff --git a/_pytest/test_everything.py b/_pytest/test_everything.py
index 1abfa42..a121541 100644
--- a/_pytest/test_everything.py
+++ b/_pytest/test_everything.py
@@ -44,7 +44,7 @@ def test_process_message(monkeypatch, realish_eventrouter, mock_websocket):
         print "####################"
 
     print len(eventrouter.queue)
-#    assert False
+    #assert False
 
 
 
diff --git a/_pytest/test_processteamjoin.py b/_pytest/test_processteamjoin.py
new file mode 100644
index 0000000..00a8b4c
--- /dev/null
+++ b/_pytest/test_processteamjoin.py
@@ -0,0 +1,54 @@
+import glob
+import json
+
+from wee_slack import ProcessNotImplemented
+
+def test_process_reply(monkeypatch, mock_websocket, realish_eventrouter):
+
+    eventrouter = realish_eventrouter
+
+    t = eventrouter.teams.keys()[0]
+    #u = eventrouter.teams[t].users.keys()[0]
+
+    #user = eventrouter.teams[t].users[u]
+    #print user
+
+    #delete charles so we can add him
+    del eventrouter.teams[t].users['U4096CBHC']
+
+    assert len(eventrouter.teams[t].users) == 3
+
+    socket = mock_websocket
+    eventrouter.teams[t].ws = socket
+
+    datafiles = glob.glob("_pytest/data/websocket/1485975606.59-team_join.json")
+
+    print datafiles
+    #assert False
+
+    notimplemented = set()
+
+    for fname in datafiles:
+        try:
+            print "####################"
+            data = json.loads(open(fname, 'r').read())
+            socket.add(data)
+            print data
+            eventrouter.receive_ws_callback(t)
+            eventrouter.handle_next()
+        except ProcessNotImplemented as e:
+            notimplemented.add(str(e))
+        #this handles some message data not existing - need to fix
+        except KeyError:
+            pass
+
+    if len(notimplemented) > 0:
+        print "####################"
+        print sorted(notimplemented)
+        print "####################"
+
+    #print len(eventrouter.queue)
+    assert len(eventrouter.teams[t].users) == 4
+
+
+