From 2fd7a079c19ce42c7705e9e87bc5e807d45891b2 Mon Sep 17 00:00:00 2001
From: Soren Hansen <soren@canonical.com>
Date: Fri, 4 Sep 2009 11:16:09 -0500
Subject: Add a unittest for veriyfing that shell scripts passed in user-data
 get called.

---
 ec2-run-user-data.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'ec2-run-user-data.py')

diff --git a/ec2-run-user-data.py b/ec2-run-user-data.py
index d7e8e632..2108e0a0 100755
--- a/ec2-run-user-data.py
+++ b/ec2-run-user-data.py
@@ -154,10 +154,13 @@ class ApplianceConfig(object):
 def main():
     ec2 = ec2init.EC2Init()
 
-    user_data = ec2.get_user_data()
+    user_data = get_user_data()
     msg = parse_user_data(user_data)
     handle_part(msg)
 
+def get_user_data():
+    return ec2.get_user_data()
+
 def parse_user_data(user_data):
     return email.message_from_string(user_data)
 
-- 
cgit v1.2.3